Remove lazy_fixture usage.

edtechre · Mar 17, 2024 · 348d59d · 348d59d
1 parent aba7019
commit 348d59d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_model.py b/tests/test_model.py
@@ -10,6 +10,7 @@
 import pytest
 import re
 from .fixtures import *  # noqa: F401
+from .util import *  # noqa: F401
 from unittest.mock import Mock
 from pybroker.cache import CacheDateFields
 from pybroker.common import ModelSymbol, TrainedModel, to_datetime
@@ -179,7 +180,7 @@ def _assert_models(self, models, expected_model_syms):
         "param_test_data",
         [
             pd.DataFrame(columns=["symbol", "date"]),
-            pytest.lazy_fixture("test_data"),
+            "test_data",
         ],
     )
     def test_train_models(
@@ -189,7 +190,9 @@ def test_train_models(
         param_test_data,
         ind_data,
         cache_date_fields,
+        request,
     ):
+        param_test_data = get_fixture(request, param_test_data)
         mixin = ModelsMixin()
         models = mixin.train_models(
             model_syms,