2026-06-02 19:04:23 +00:00
1 changed files with 1 additions and 6 deletions
--- a/src/tracking/mlflow.py
+++ b/src/tracking/mlflow.py
@@ -12,12 +12,7 @@ from src.config import Config, MlflowMode
 class Tracker(Protocol):
    def start_training_run(self, training_job: Any, *, region: str, profile: str, role_arn: str) -> str | None: ...

-    def finalize_training_run(
-        self,
-        *,
-        run_id: str | None,
-        training_job_status: Any,
-    ) -> str | None: ...
+    def finalize_training_run(self, *, run_id: str | None, training_job_status: Any) -> str | None: ...


@dataclass(frozen=True)