Refactor init

tryolabs · Jun 5, 2024 · 561c743 · 561c743
1 parent 44ec38d
commit 561c743
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ml_garden/__init__.py b/ml_garden/__init__.py
@@ -1,6 +1,6 @@
 from .core.pipeline import Pipeline
 
-Pipeline.step_registry.auto_register_steps_from_package("pipeline_lib.core.steps")
+Pipeline.step_registry.auto_register_steps_from_package("ml_garden.core.steps")
 Pipeline.model_registry.auto_register_models_from_package(
-    "pipeline_lib.implementation.tabular.xgboost"
+    "ml_garden.implementation.tabular.xgboost"
 )