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)