chore: formatting

Author: rokroskar

components/renku_data_services/metrics/db.py

@@ -46,18 +46,18 @@ async def delete_processed_metrics(self, metrics_ids: list[ULID]) -> None:
 
     async def wait_for_metrics(self, timeout: float = 5.0, poll_interval: float = 0.1) -> bool:
         """Wait for metrics to be processed.
-        
+
         Polls for metrics events and returns when at least one event is found or timeout is reached.
-        
+
         Args:
             timeout: Maximum time to wait in seconds
             poll_interval: Time between polls in seconds
-            
+
         Returns:
             True if metrics were found, False if timeout reached
         """
         import time
-        
+
         start_time = time.monotonic()
         while time.monotonic() - start_time < timeout:
             metrics = [m async for m in self.get_unprocessed_metrics()]

test/bases/renku_data_services/data_api/test_metrics.py

@@ -17,6 +17,7 @@ def _has_docker() -> bool:
     except (FileNotFoundError, subprocess.TimeoutExpired):
         return False
 
+
 from renku_data_services.base_models.metrics import ProjectCreationType
 from renku_data_services.metrics.core import StagingMetricsService
 
@@ -74,17 +75,19 @@ async def test_metrics_are_stored(sanic_metrics_client, app_manager, create_proj
 
 @pytest.mark.asyncio
 @pytest.mark.skipif(not _has_docker(), reason="docker is not available - kind cannot create clusters")
-async def test_session_metrics_are_stored(sanic_metrics_client, app_manager, create_project, create_session_launcher, create_resource_pool, user_headers) -> None:
+async def test_session_metrics_are_stored(
+    sanic_metrics_client, app_manager, create_project, create_session_launcher, create_resource_pool, user_headers
+) -> None:
     """Test that session lifecycle metrics with metadata are stored correctly.
-    
+
     Note: This test requires kind cluster to be available.
     """
 
     project = await create_project(name="Project", sanic_client=sanic_metrics_client)
     # Create a resource pool with a resource class to use in the session launcher
     resource_pool = await create_resource_pool(admin=True)
     resource_class_id = resource_pool["classes"][0]["id"]
-    
+
     # Create a session launcher with a resource_class_id
     session_launcher = await create_session_launcher(
         "Launcher 1", project_id=project["id"], resource_class_id=resource_class_id

test/components/renku_data_services/base_models/test_session_metrics.py

@@ -7,13 +7,15 @@
 """
 
 import pytest
+
 from renku_data_services.metrics.core import StagingMetricsService
 from renku_data_services.notebooks.core_sessions import _make_patch_spec_list
 
 
 @pytest.mark.asyncio
 async def test_session_metrics_metadata_structure():
     """Test that session metrics store metadata with correct structure."""
+
     # Create mock metrics repo
     async def mock_store_event(*args, **kwargs):
         pass
@@ -24,10 +26,14 @@ async def mock_store_event(*args, **kwargs):
     metrics = StagingMetricsService(enabled=True, metrics_repo=mock_metrics_repo)
 
     # Create mock user with required attributes
-    mock_user = type("APIUser", (), {
-        "id": "test-user-123",
-        "is_authenticated": True,
-    })()
+    mock_user = type(
+        "APIUser",
+        (),
+        {
+            "id": "test-user-123",
+            "is_authenticated": True,
+        },
+    )()
 
     # Test session_started metadata with all resource fields
     await metrics.session_started(
@@ -122,11 +128,12 @@ class MyResource:
 @pytest.mark.asyncio
 async def test_session_metrics_metadata_fields_match_implementation():
     """Verify that test metadata matches what's actually sent in the implementation.
-    
+
     This test validates the exact metadata fields that are sent from:
     - k8s watcher (session_started, session_resumed, session_hibernated, session_stopped)
     - notebooks.core_sessions (user_requested_session_resume)
     """
+
     # Create mock metrics repo
     async def mock_store_event(*args, **kwargs):
         pass
@@ -135,10 +142,14 @@ async def mock_store_event(*args, **kwargs):
     metrics = StagingMetricsService(enabled=True, metrics_repo=mock_metrics_repo)
 
     # Create mock user
-    mock_user = type("APIUser", (), {
-        "id": "test-user-123",
-        "is_authenticated": True,
-    })()
+    mock_user = type(
+        "APIUser",
+        (),
+        {
+            "id": "test-user-123",
+            "is_authenticated": True,
+        },
+    )()
 
     # Test ALL the metadata fields that are actually sent in the implementation
     # From: components/renku_data_services/k8s/watcher/core.py:208-219 (session_started/resumed)