fix formatting

Author: saichandrapandraju

src/main.py

@@ -1,5 +1,5 @@
-import logging
 import asyncio
+import logging
 from contextlib import asynccontextmanager
 
 import uvicorn
@@ -50,14 +50,16 @@
 
 prometheus_scheduler = PrometheusScheduler()
 
+
 @repeat_every(
     seconds=prometheus_scheduler.service_config.get("metrics_schedule", 30),
     logger=logger,
-    raise_exceptions=False
+    raise_exceptions=False,
 )
 async def schedule_metrics_calculation():
     await prometheus_scheduler.calculate()
 
+
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     task = asyncio.create_task(schedule_metrics_calculation())

src/service/data/datasources/data_source.py

@@ -1,4 +1,3 @@
-import asyncio
 import logging
 import os
 from concurrent.futures import ThreadPoolExecutor
@@ -90,11 +89,12 @@ async def get_dataframe_with_batch_size(
             start_row = max(0, available_rows - batch_size)
             n_rows = min(batch_size, available_rows)
 
-            input_data, output_data, metadata = await model_data.data(start_row=start_row, n_rows=n_rows)
+            input_data, output_data, metadata = await model_data.data(
+                start_row=start_row, n_rows=n_rows
+            )
 
             input_names, output_names, metadata_names = await model_data.column_names()
 
-
             # Combine the data into a single dataframe
             df_data = {}
 
@@ -121,7 +121,9 @@ async def get_dataframe_with_batch_size(
                 f"Error creating dataframe for model={model_id}: {str(e)}"
             )
 
-    async def get_organic_dataframe(self, model_id: str, batch_size: int) -> pd.DataFrame:
+    async def get_organic_dataframe(
+        self, model_id: str, batch_size: int
+    ) -> pd.DataFrame:
         """
         Get a dataframe with only organic data (not synthetic).
 
@@ -208,7 +210,9 @@ async def has_metadata(self, model_id: str) -> bool:
         try:
             return await self.get_metadata(model_id) is not None
         except Exception as e:
-            logger.error(f"Error checking if metadata exists for model={model_id}: {str(e)}")
+            logger.error(
+                f"Error checking if metadata exists for model={model_id}: {str(e)}"
+            )
             return False
 
     # DATAFRAME QUERIES

src/service/payloads/metrics/base_metric_request.py

@@ -1,14 +1,16 @@
 from abc import abstractmethod
 from typing import Dict, Optional
+
 from pydantic import BaseModel, ConfigDict
 
 
 class BaseMetricRequest(BaseModel):
     """
     Abstract base class for metric requests.
     """
+
     # To allow extra fields to be set on instances
-    model_config = ConfigDict(extra='allow')
+    model_config = ConfigDict(extra="allow")
 
     model_id: str
     metric_name: str

src/service/payloads/metrics/request_reconciler.py

@@ -24,7 +24,9 @@ async def reconcile(request: BaseMetricRequest, data_source: DataSource) -> None
             request: The metric request to reconcile
             data_source: The data source to use for reconciliation
         """
-        storage_metadata: StorageMetadata = await data_source.get_metadata(request.model_id)
+        storage_metadata: StorageMetadata = await data_source.get_metadata(
+            request.model_id
+        )
         RequestReconciler.reconcile_with_metadata(request, storage_metadata)
 
     @staticmethod

src/service/payloads/service/schema_item.py

@@ -1,6 +1,8 @@
-from src.service.payloads.values.data_type import DataType
 from dataclasses import dataclass
 
+from src.service.payloads.values.data_type import DataType
+
+
 @dataclass
 class SchemaItem:
     type: DataType
@@ -24,4 +26,4 @@ def get_column_index(self) -> int:
         return self.column_index
 
     def set_column_index(self, column_index: int) -> None:
-        self.column_index = column_index
+        self.column_index = column_index

src/service/prometheus/prometheus_publisher.py

@@ -15,7 +15,8 @@
 # Must start with letter, underscore, or colon
 # Can contain letters, numbers, underscores, and colons
 # Lowercase only (shall we allow uppercase?)
-PROMETHEUS_METRIC_NAME_REGEX = re.compile(r'^[a-z_:][a-z0-9_:]*$')
+PROMETHEUS_METRIC_NAME_REGEX = re.compile(r"^[a-z_:][a-z0-9_:]*$")
+
 
 class PrometheusPublisher:
     def __init__(self, registry: CollectorRegistry = REGISTRY) -> None:
@@ -68,7 +69,7 @@ def remove_gauge(self, name: str, id: uuid.UUID) -> None:
 
             if full_name in self._gauges:
                 gauge = self._gauges[full_name]
-                
+
                 # IMPORTANT: Accessing private attributes of prometheus_client.Gauge
                 # This is necessary because the prometheus_client library does not provide
                 # public methods to:
@@ -184,7 +185,7 @@ def _get_full_metric_name(self, metric_name: str) -> str:
                 f"underscore, or colon, and contain only lowercase letters, "
                 f"numbers, underscores, and colons."
             )
-        
+
         return full_name
 
     @staticmethod