[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

Author: pre-commit-ci[bot]

lumigator/backend/backend/alembic/env.py

@@ -2,12 +2,11 @@
 from logging.config import fileConfig
 
 from alembic import context
-from sqlalchemy import engine_from_config, pool
-
 from backend.records.base import BaseRecord
 from backend.records.datasets import *  # noqa: F403
 from backend.records.experiments import *  # noqa: F403
 from backend.records.jobs import *  # noqa: F403
+from sqlalchemy import engine_from_config, pool
 
 """
 NOTE: Do NOT remove imports for the data models: backend.records.{package}

lumigator/backend/backend/tests/e2e/test_api_workflows.py

@@ -6,6 +6,15 @@
 
 import pytest
 import requests
+from backend.main import app
+from backend.tests.conftest import (
+    DEFAULT_MAX_RETRIES,
+    DEFAULT_RETRY_INTERVAL_SECONDS,
+    MAX_JOB_TIMEOUT_SECS,
+    TEST_CAUSAL_MODEL,
+    TEST_SEQ2SEQ_MODEL,
+    wait_for_job,
+)
 from fastapi.testclient import TestClient
 from httpx import HTTPStatusError, RequestError
 from inference.schemas import GenerationConfig, InferenceJobConfig, InferenceServerConfig
@@ -28,16 +37,6 @@
 from lumigator_schemas.workflows import WorkflowDetailsResponse, WorkflowResponse, WorkflowStatus
 from pydantic import PositiveInt, ValidationError
 
-from backend.main import app
-from backend.tests.conftest import (
-    DEFAULT_MAX_RETRIES,
-    DEFAULT_RETRY_INTERVAL_SECONDS,
-    MAX_JOB_TIMEOUT_SECS,
-    TEST_CAUSAL_MODEL,
-    TEST_SEQ2SEQ_MODEL,
-    wait_for_job,
-)
-
 POST_HEADER = {
     "accept": "application/json",
     "Content-Type": "application/json",

lumigator/backend/backend/tests/unit/tracking/test_mlflow.py

@@ -3,14 +3,13 @@
 from unittest.mock import AsyncMock, MagicMock
 
 import pytest
+from backend.services.exceptions.tracking_exceptions import RunNotFoundError
+from backend.tracking.mlflow import MLflowTrackingClient
 from lumigator_schemas.jobs import JobResults
 from lumigator_schemas.workflows import WorkflowStatus
 from mlflow.entities import Metric, Param, Run, RunData, RunInfo, RunStatus, RunTag
 from mlflow.exceptions import MlflowException
 
-from backend.services.exceptions.tracking_exceptions import RunNotFoundError
-from backend.tracking.mlflow import MLflowTrackingClient
-
 
 @pytest.mark.asyncio
 @pytest.mark.parametrize(

lumigator/jobs/evaluator/definition.py

@@ -1,10 +1,9 @@
 from uuid import UUID
 
+from backend.services.job_interface import JobDefinition
 from evaluator.schemas import DatasetConfig, EvalJobConfig, EvaluationConfig
 from lumigator_schemas.jobs import JobCreate, JobType
 
-from backend.services.job_interface import JobDefinition
-
 
 class JobDefinitionEvaluation(JobDefinition):
     def generate_config(self, request: JobCreate, record_id: UUID, dataset_path: str, storage_path: str):

lumigator/jobs/inference/definition.py

@@ -1,6 +1,7 @@
 import os
 from uuid import UUID
 
+from backend.services.job_interface import JobDefinition
 from inference.schemas import (
     DatasetConfig,
     HuggingFacePipelineConfig,
@@ -10,8 +11,6 @@
 )
 from lumigator_schemas.jobs import JobCreate, JobType
 
-from backend.services.job_interface import JobDefinition
-
 # Served models
 OAI_API_URL: str = "https://api.openai.com/v1"
 MISTRAL_API_URL: str = "https://api.mistral.ai/v1"