delete unnecesary steps

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

Hide file tree

Showing 3 changed files with 1 addition and 216 deletions.
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
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