fix: Sort import and other pre-commit fix

PR_4.md

@@ -13,7 +13,7 @@ The goal of this PR is to:
 - [ ] Create a FastAPI API to make predictions on new data
 - [ ] Dockerize the API
 
-Files to be modified: 
+Files to be modified:
 - [ ] the `src/web_service` folder
 - [ ] the `bin/run_services.sh` file
 - [ ] `Dockerfile.app`

src/web_service/lib/inference.py

@@ -2,13 +2,12 @@
 
 import numpy as np
 import pandas as pd
-from loguru import logger
 from lib.models import InputData
-from sklearn.base import BaseEstimator
-from sklearn.compose import ColumnTransformer
-
 from lib.predicting import predict
 from lib.preprocessing import preprocess_data
+from loguru import logger
+from sklearn.base import BaseEstimator
+from sklearn.compose import ColumnTransformer
 
 
 def run_inference(

src/web_service/main.py

@@ -1,10 +1,15 @@
-from app_config import APP_DESCRIPTION, APP_TITLE, APP_VERSION, MODEL_VERSION
+from app_config import (
+    APP_DESCRIPTION,
+    APP_TITLE,
+    APP_VERSION,
+    MODEL_VERSION,
+    PATH_TO_MODEL,
+    PATH_TO_PREPROCESSOR,
+)
 from fastapi import FastAPI
 from lib.inference import run_inference
-from lib.utils import load_pickle
 from lib.models import InputData, PredictionOut
-from app_config import PATH_TO_MODEL, PATH_TO_PREPROCESSOR
-
+from lib.utils import load_pickle
 
 app = FastAPI(
     title=APP_TITLE, description=APP_DESCRIPTION, version=APP_VERSION
@@ -23,5 +28,5 @@ def home() -> dict:
 def predict(value: InputData) -> dict:
     model = load_pickle(PATH_TO_MODEL)
     preprocessor = load_pickle(PATH_TO_PREPROCESSOR)
-    y = run_inference([value], preprocessor, model) 
-    return {"abalone_age_prediction": y}
+    y = run_inference([value], preprocessor, model)
+    return {"abalone_age_prediction": y}