From 348d59dc02a4e95a3c5a069756c47f596bef352a Mon Sep 17 00:00:00 2001
From: edtechre <edtechre@gmail.com>
Date: Sat, 16 Mar 2024 18:51:02 -0700
Subject: [PATCH] Remove lazy_fixture usage.

---
 tests/test_model.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/tests/test_model.py b/tests/test_model.py
index 7c4ff04..0bb87a5 100644
--- 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,