delete unnecesary steps

tryolabs · Mar 25, 2024 · 38279c0 · 38279c0
1 parent ac8443c
commit 38279c0
Show file tree

Hide file tree

Showing 5 changed files with 1 addition and 220 deletions.
diff --git a/pipeline_lib/core/pipeline.py b/pipeline_lib/core/pipeline.py
@@ -4,8 +4,6 @@
 import logging
 from typing import Optional
 
-from joblib import load
-
 from pipeline_lib.core.data_container import DataContainer
 from pipeline_lib.core.model_registry import ModelRegistry
 from pipeline_lib.core.step_registry import StepRegistry

diff --git a/pipeline_lib/core/steps/target_scaling.py b/pipeline_lib/core/steps/target_scaling.py
@@ -1,5 +1,3 @@
-from typing import Optional
-
 from pipeline_lib.core import DataContainer
 from pipeline_lib.core.steps.base import PipelineStep
 

diff --git a/pipeline_lib/implementation/tabular/xgboost/__init__.py b/pipeline_lib/implementation/tabular/xgboost/__init__.py
@@ -1,2 +1 @@
-from .fit_model import XGBoostFitModelStep  # noqa: F401
-from .predict import XGBoostPredictStep  # noqa: F401
+from .model import XGBoostModel  # noqa: F401
diff --git a/pipeline_lib/implementation/tabular/xgboost/fit_model.py b/pipeline_lib/implementation/tabular/xgboost/fit_model.py
diff --git a/pipeline_lib/implementation/tabular/xgboost/predict.py b/pipeline_lib/implementation/tabular/xgboost/predict.py