From 416e51901d94730cc3f2d886335cb6f82d32da50 Mon Sep 17 00:00:00 2001 From: slalom Date: Fri, 29 May 2026 14:33:17 -0400 Subject: [PATCH] space --- src/tracking/mlflow.py | 7 +------ 1 file changed, 1 insertion(+), 6 deletions(-) diff --git a/src/tracking/mlflow.py b/src/tracking/mlflow.py index 008155c..b38e8af 100644 --- 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)