diff --git a/tests/test_predictive_variable.py b/tests/test_predictive_variable.py index 83a238ef..2a3ef478 100644 --- a/tests/test_predictive_variable.py +++ b/tests/test_predictive_variable.py @@ -4,7 +4,7 @@ from helper_functions import load_and_format from pandas.testing import assert_frame_equal -from src.predictive_variable import shift_by_strata_period +from mbs_results.predictive_variable import shift_by_strata_period @pytest.fixture(scope="class")