refactor(rf_model.py): rename model file

end-to-end-mlops-databricks · Oct 22, 2024 · 8796f7f · 8796f7f
1 parent 96ccf34
commit 8796f7f
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/power_consumption/model/__init__.py b/power_consumption/model/__init__.py
@@ -1,3 +1,3 @@
-from .main import ConsumptionModel
+from .rf_model import ConsumptionModel
 
 __all__ = ["ConsumptionModel"]
diff --git a/power_consumption/model/main.py → power_consumption/model/rf_model.py b/power_consumption/model/main.py → power_consumption/model/rf_model.py
diff --git a/tests/model/test_main.py b/tests/model/test_main.py
@@ -4,7 +4,7 @@
 from sklearn.preprocessing import StandardScaler
 from sklearn.pipeline import Pipeline
 
-from power_consumption.model.main import ConsumptionModel
+from power_consumption.model import ConsumptionModel
 
 @pytest.fixture
 def sample_data():