SIENTIAPDE-1579: Lint fixes and formatting

model_manager/activities/experiment_tracking.py

@@ -6,6 +6,8 @@ in the PostgreSQL database, extending the base Postgres activity with specialize
 methods for experiment management.
 """
 
+import enum
+
 from temporalio import activity, workflow
 
 with workflow.unsafe.imports_passed_through():
@@ -13,7 +15,6 @@ with workflow.unsafe.imports_passed_through():
     import traceback
     from collections.abc import Mapping
     from datetime import UTC, datetime
-    from enum import Enum
     from typing import Any
 
     from sientia_do.notifications.handlers import CoreNotificationHandler as NotificationHandler
@@ -24,7 +25,7 @@ with workflow.unsafe.imports_passed_through():
     from sqlalchemy import text
 
 
-class UpdateType(str, Enum):
+class UpdateType(enum.StrEnum):
     """Types of experiment run updates."""
 
     STATUS = 'status'

model_manager/sientia/models.py

@@ -463,7 +463,9 @@ class DataPreprocessor(BaseEstimator, TransformerMixin):
         if not date_str:
             return None
         try:
-            python_fmt = _frontend_date_format_to_strftime(self.date_format) if self.date_format else None
+            python_fmt = (
+                _frontend_date_format_to_strftime(self.date_format) if self.date_format else None
+            )
             if python_fmt:
                 return pd.to_datetime(date_str, format=python_fmt)
             return pd.to_datetime(date_str)

model_manager/utils/models/experiment_status.py

@@ -1,7 +1,7 @@
-from enum import Enum
+from enum import StrEnum
 
 
-class ExperimentStatus(str, Enum):
+class ExperimentStatus(StrEnum):
     """
     Status values for experiment run lifecycle.
 

model_manager/utils/repository/training_repository.py

@@ -481,9 +481,9 @@ class TrainingRepository:
             'datetime',
             'DateTime',
         ]
-        timestamp_columns = (
-            [params.date_column] if params.date_column else []
-        ) + [c for c in common_timestamp_columns if c != params.date_column]
+        timestamp_columns = ([params.date_column] if params.date_column else []) + [
+            c for c in common_timestamp_columns if c != params.date_column
+        ]
 
         for col in timestamp_columns:
             if col in data.columns: