clean: Clean ComplianceModel

pass-culture · Aug 28, 2024 · 626eefd · 626eefd
1 parent 5005d5b
commit 626eefd
Showing 1 changed file with 0 additions and 13 deletions.
diff --git a/apps/fraud/compliance/api/src/pcpapillon/core/compliance_model.py b/apps/fraud/compliance/api/src/pcpapillon/core/compliance_model.py
@@ -1,25 +1,12 @@
-from dataclasses import dataclass
-from typing import Union
-
-import mlflow
 from main import custom_logger
 from pcpapillon.utils.constants import ModelName, ModelType
 from pcpapillon.utils.data_model import ComplianceInput, ComplianceOutput
 from pcpapillon.utils.model_handler import ModelHandler, ModelWithMetadata
-from sentence_transformers import SentenceTransformer
-
-
-@dataclass
-class ModelData:
-    classification_model: Union[mlflow.pyfunc.PythonModel, SentenceTransformer]
-    model_identifier: str
-    preprocessing_models: dict[str, SentenceTransformer]
 
 
 class ComplianceModel:
     MODEL_NAME = ModelName.COMPLIANCE
     MODEL_TYPE = ModelType.DEFAULT
-    PREPROC_MODEL_TYPE = MODEL_TYPE.PREPROCESSING
 
     def __init__(self):
         self.model_handler = ModelHandler()