fix bug

simplymathematics · Aug 13, 2024 · d674b97 · d674b97
1 parent a38ed17
commit d674b97
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/deckard/base/model/sklearn_pipeline.py b/deckard/base/model/sklearn_pipeline.py
@@ -51,7 +51,7 @@
 
 
 @dataclass
-class SklearnModelPipelineStage(Hashable):
+class SklearnModelPipelineStage:
     name: str
     stage_name: str = None
     kwargs: dict = field(default_factory=dict)
@@ -92,7 +92,7 @@ def __call__(self, model):
 
 
 @dataclass
-class SklearnModelPipeline(Hashable):
+class SklearnModelPipeline:
     pipeline: Dict[str, SklearnModelPipelineStage] = field(default_factory=dict)
 
     def __init__(self, **kwargs):
@@ -202,7 +202,7 @@ def __call__(self, model):
 
 
 @dataclass
-class SklearnModelInitializer(Hashable):
+class SklearnModelInitializer:
     model: object = field(default_factory=None)
     library: str = field(default_factory="sklearn")
     pipeline: SklearnModelPipeline = field(default_factory=None)