🩹 Fix error B904 of pre-commit

DashAI/back/api/api_v1/endpoints/.cph/.predict.py_29776e670b8e2e7b1a3540d61a89dddd.prob

@@ -1 +1 @@
-{"name":"Local: predict","url":"c:\\Users\\felip\\OneDrive\\Escritorio\\proyecto\\python-envs\\DashAI\\DashAI\\back\\api\\api_v1\\endpoints\\predict.py","tests":[{"id":1733963920211,"input":"","output":""}],"interactive":false,"memoryLimit":1024,"timeLimit":3000,"srcPath":"c:\\Users\\felip\\OneDrive\\Escritorio\\proyecto\\python-envs\\DashAI\\DashAI\\back\\api\\api_v1\\endpoints\\predict.py","group":"local","local":true}
+{"name":"Local: predict","url":"c:\\Users\\felip\\OneDrive\\Escritorio\\proyecto\\python-envs\\DashAI\\DashAI\\back\\api\\api_v1\\endpoints\\predict.py","tests":[{"id":1733963920211,"input":"","output":""}],"interactive":false,"memoryLimit":1024,"timeLimit":3000,"srcPath":"c:\\Users\\felip\\OneDrive\\Escritorio\\proyecto\\python-envs\\DashAI\\DashAI\\back\\api\\api_v1\\endpoints\\predict.py","group":"local","local":true}

DashAI/back/api/api_v1/endpoints/predict.py

@@ -5,13 +5,12 @@
 
 from fastapi import APIRouter, Depends, status
 from fastapi.exceptions import HTTPException
-from fastapi.responses import JSONResponse
 from kink import di, inject
 from sqlalchemy.orm import sessionmaker
 
 from DashAI.back.api.api_v1.schemas.predict_params import (
+    FilterDatasetParams,
     RenameRequest,
-    filterDatasetParams,
 )
 from DashAI.back.dataloaders.classes.dashai_dataset import get_columns_spec
 from DashAI.back.dependencies.database.models import Dataset, Experiment, Run
@@ -51,7 +50,7 @@ async def get_metadata_prediction_json(
         path.mkdir(parents=True, exist_ok=True)
         files = os.listdir(path)
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail=str(e)) from e
 
     json_files = [f for f in files if f.endswith(".json")]
     if not json_files:
@@ -193,8 +192,10 @@ async def get_predict_summary(
         with open(path, "r") as f:
             try:
                 data = json.load(f)["prediction"]
-            except json.JSONDecodeError:
-                raise HTTPException(status_code=400, detail="Invalid JSON format")
+            except json.JSONDecodeError as e:
+                raise HTTPException(
+                    status_code=400, detail="Invalid JSON format"
+                ) from e
             summary["total_data_points"] = len(data)
             class_set = set(data)
             classes = [str(item) for item in class_set]
@@ -204,10 +205,10 @@ async def get_predict_summary(
             for class_name in classes:
                 try:
                     occurrences = data.count(int(class_name))
-                except ValueError:
+                except ValueError as e:
                     raise HTTPException(
                         status_code=400, detail=f"Invalid class value: {class_name}"
-                    )
+                    ) from e
                 distribution = {
                     "id": id,
                     "Class": class_name,
@@ -221,16 +222,16 @@ async def get_predict_summary(
             {"id": idx, "value": value} for idx, value in enumerate(data[:50], 1)
         ]
         summary["sample_data"] = sample_data
-    except FileNotFoundError:
-        raise HTTPException(status_code=404, detail="Prediction not found")
+    except FileNotFoundError as e:
+        raise HTTPException(status_code=404, detail="Prediction not found") from e
     except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+        raise HTTPException(status_code=500, detail=str(e)) from e
     return summary
 
 
 @router.post("/filter_datasets")
 async def filter_datasets_endpoint(
-    params: filterDatasetParams,
+    params: FilterDatasetParams,
     session_factory: sessionmaker = Depends(lambda: di["session_factory"]),
 ):
     """
@@ -275,7 +276,7 @@ async def filter_datasets_endpoint(
         raise HTTPException(
             status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
             detail="An error occurred while filtering datasets",
-        )
+        ) from e
 
 
 @router.get("/download/{predict_name}")
@@ -314,7 +315,7 @@ async def download_prediction(
         raise HTTPException(
             status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
             detail="An error occurred while downloading the prediction file",
-        )
+        ) from e
 
 
 @router.delete("/{predict_name}")
@@ -354,7 +355,7 @@ async def delete_prediction(
         raise HTTPException(
             status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
             detail="An error occurred while deleting the prediction file",
-        )
+        ) from e
 
 
 @router.patch("/{predict_name}")
@@ -409,4 +410,4 @@ async def rename_prediction(
         raise HTTPException(
             status_code=status.HTTP_500_INTERNAL_SERVER_ERROR,
             detail="An error occurred while renaming the prediction file",
-        )
+        ) from e

DashAI/back/api/api_v1/schemas/predict_params.py

@@ -11,6 +11,6 @@ class RenameRequest(BaseModel):
     new_name: str
 
 
-class filterDatasetParams(BaseModel):
+class FilterDatasetParams(BaseModel):
     train_dataset_id: int
     datasets: List[str]

DashAI/back/dependencies/database/models.py

@@ -2,7 +2,7 @@
 from datetime import datetime
 from typing import List
 
-from sqlalchemy import JSON, Boolean, DateTime, Enum, ForeignKey, String
+from sqlalchemy import JSON, DateTime, Enum, ForeignKey, String
 from sqlalchemy.ext.declarative import declarative_base
 from sqlalchemy.orm import Mapped, mapped_column, relationship
 

DashAI/back/job/predict_job.py

@@ -52,7 +52,6 @@ def run(
         session_factory: sessionmaker = Depends(lambda: di["session_factory"]),
         config=lambda di: di["config"],
     ) -> List[Any]:
-
         run_id: int = self.kwargs["run_id"]
         id: int = self.kwargs["id"]
         db: Session = self.kwargs["db"]
@@ -112,7 +111,7 @@ def run(
             raise HTTPException(
                 status_code=400,
                 detail=f"Invalid columns selected: {str(ve)}",
-            )
+            ) from ve
         except Exception as e:
             log.error(e)
             raise JobError(

tests/back/api/test_predict_api.py

@@ -1,19 +1,11 @@
-import csv
-import datetime
 import json
 import os
 from pathlib import Path
 
 import joblib
-import numpy as np
 import pytest
 from fastapi.testclient import TestClient
 
-from DashAI.back.api.api_v1.schemas.job_params import JobParams
-from DashAI.back.dataloaders.classes.dashai_dataset import (
-    DashAIDataset,
-    get_dataset_info,
-)
 from DashAI.back.dataloaders.classes.json_dataloader import JSONDataLoader
 from DashAI.back.dependencies.database.models import Dataset, Experiment, Run
 from DashAI.back.dependencies.registry import ComponentRegistry