update module name

ccomkhj · Nov 21, 2023 · 185e463 · 185e463
1 parent d497134
commit 185e463
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/tests/test_shape_check.py b/tests/test_shape_check.py
@@ -1,9 +1,12 @@
 import numpy as np
-from constrained_linear_regression.constrained_linear_regression import ConstrainedLinearRegression
-from constrained_linear_regression.multi_constrained_linear_regression import MultiConstrainedLinearRegression
+from constrainedML.constrained_linear_regression import ConstrainedLinearRegression
+from constrainedML.multi_constrained_linear_regression import (
+    MultiConstrainedLinearRegression,
+)
 from sklearn.datasets import load_linnerud
 import pytest
 
+
 @pytest.mark.xfail(raises=AssertionError)
 def test_constraint_min_coef():
     X, Y = load_linnerud(return_X_y=True)
@@ -17,6 +20,7 @@ def test_constraint_min_coef():
     model.reset()
     assert model.global_horizon_count == 0  # Check reset function
 
+
 @pytest.mark.xfail(raises=AssertionError)
 def test_constraint_max_coef():
     X, Y = load_linnerud(return_X_y=True)
@@ -30,6 +34,7 @@ def test_constraint_max_coef():
     model.reset()
     assert model.global_horizon_count == 0  # Check reset function
 
+
 @pytest.mark.xfail(raises=AssertionError)
 def test_multi_constraint_min_coef():
     X, Y = load_linnerud(return_X_y=True)
@@ -43,6 +48,7 @@ def test_multi_constraint_min_coef():
     model.reset()
     assert model.global_horizon_count == 0  # Check reset function
 
+
 @pytest.mark.xfail(raises=AssertionError)
 def test_multi_constraint_max_coef():
     X, Y = load_linnerud(return_X_y=True)