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)