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

space

Browse Source
This commit is contained in:
slalom
2026-05-29 14:33:17 -04:00
parent 556797cf13
commit 416e51901d
1 changed files with 1 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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)