Add linear regression recipe

ThomasMeissnerDS · Aug 19, 2024 · f55589e · f55589e
1 parent 2141823
commit f55589e
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/bluecast/tests/test_custom_model_recipes.py b/bluecast/tests/test_custom_model_recipes.py
@@ -114,7 +114,7 @@ def test_linear_autotune(regression_data, linear_model):
 
     # Check if the model has been fitted by inspecting the attributes
     assert hasattr(
-        linear_model.model.best_estimator_, "coef_"
+        linear_model.model, "coef_"
     ), "Model should have been fitted and have coefficients."
 
 
@@ -124,7 +124,7 @@ def test_linear_fit(regression_data, linear_model):
 
     # Again, check if the model has been fitted
     assert hasattr(
-        linear_model.model.best_estimator_, "coef_"
+        linear_model.model, "coef_"
     ), "Model should have been fitted after calling fit method."
 
 

diff --git a/dist/bluecast-1.6.0-py3-none-any.whl b/dist/bluecast-1.6.0-py3-none-any.whl
diff --git a/dist/bluecast-1.6.0.tar.gz b/dist/bluecast-1.6.0.tar.gz