Cleanup

superdosh · superdosh · commit 46fb27bfe9ab · 2025-05-30T09:18:59.000-04:00
diff --git a/src/modelplane/runways/responder.py b/src/modelplane/runways/responder.py
@@ -113,7 +113,4 @@ def respond(
                 local_path=pipeline_runner.output_dir()
                 / pipeline_runner.output_file_name,
             )
-            print(
-                f"Logged prompt responses to {pipeline_runner.output_dir() / pipeline_runner.output_file_name}"
-            )
         return mlflow.active_run().info.run_id  # type: ignore
diff --git a/tests/it/test_health.py b/tests/it/test_health.py
@@ -6,7 +6,4 @@
 
 def test_tracking_server_is_live():
     """Test if the MLflow tracking server is live."""
-    # TODO: remove this
-    print("Checking if MLflow tracking server is live...")
-    print("Current tracking URI:", mlflow.get_tracking_uri())
     assert tracking_server_is_live(), "MLflow tracking server should be live"

Original file line number	Diff line number	Diff line change
`@@ -113,7 +113,4 @@ def respond(`
`113`	`113`	`local_path=pipeline_runner.output_dir()`
`114`	`114`	`/ pipeline_runner.output_file_name,`
`115`	`115`	`)`
`116`		`- print(`
`117`		`- f"Logged prompt responses to {pipeline_runner.output_dir() / pipeline_runner.output_file_name}"`
`118`		`- )`
`119`	`116`	`return mlflow.active_run().info.run_id # type: ignore`