slalom/qai-cli
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Mlflow implementation #2

Merged
slalom merged 10 commits from ml-flow into main 2026-06-02 19:04:23 +00:00
Conversation 0 Commits 10 Files Changed 13 +1 -6
1 changed files with 1 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 416e51901d - Show all commits

7
src/tracking/mlflow.py
View File

@@ -12,12 +12,7 @@ from src.config import Config, MlflowMode
class Tracker(Protocol): class Tracker(Protocol):
def start_training_run(self, training_job: Any, *, region: str, profile: str, role_arn: str) -> str | None: ... def start_training_run(self, training_job: Any, *, region: str, profile: str, role_arn: str) -> str | None: ...
def finalize_training_run( def finalize_training_run(self, *, run_id: str | None, training_job_status: Any) -> str | None: ...
self,
*,
run_id: str | None,
training_job_status: Any,
) -> str | None: ...
@dataclass(frozen=True) @dataclass(frozen=True)