More reverting formatting changes and removign unused import

VectorInstitute · Apr 10, 2024 · d6f1261 · d6f1261
1 parent 4b9f25f
commit d6f1261
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/florist/tests/integration/api/test_train.py b/florist/tests/integration/api/test_train.py
@@ -64,14 +64,12 @@ def test_train():
                 client_uuid = response.json()["client_uuids"][0]
 
                 # Wait for training to finish
-                wait_for_metric(server_uuid, "fit_end", test_redis_host,
-                                test_redis_port, LOGGER, max_retries=80)
+                wait_for_metric(server_uuid, "fit_end", test_redis_host, test_redis_port, LOGGER, max_retries=80)
 
                 # Check server metrics
                 server_metrics_result = redis_conn.get(server_uuid)
                 assert server_metrics_result is not None and isinstance(server_metrics_result, bytes)
-                server_metrics = json.loads(
-                    server_metrics_result.decode("utf8"))
+                server_metrics = json.loads(server_metrics_result.decode("utf8"))
                 assert server_metrics["type"] == "server"
                 assert "fit_start" in server_metrics
                 assert "fit_end" in server_metrics

diff --git a/florist/tests/unit/api/test_client.py b/florist/tests/unit/api/test_client.py
@@ -2,7 +2,6 @@
 import json
 from unittest.mock import ANY, Mock, patch
 
-from fastapi.responses import JSONResponse
 from florist.api import client
 from florist.api.clients.mnist import MnistClient
 from florist.api.monitoring.logs import get_client_log_file_path
@@ -89,4 +88,4 @@ def test_check_status(mock_redis: Mock) -> None:
 
     response = client.check_status(test_uuid, test_redis_host, test_redis_port)
 
-    assert json.loads(response.body.decode()) == {"info": "test"}
+    assert json.loads(response.body.decode()) == {"info": "test"}