update Structured

Quantco · Dec 5, 2024 · ea6704f · ea6704f
1 parent 8a56828
commit ea6704f
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/test_formula.py b/tests/test_formula.py
@@ -11,6 +11,11 @@
 from formulaic.parser.types import Factor
 from scipy import sparse as sps
 
+try:
+    from formulaic.utils.structured import Structured
+except ImportError:
+    from formulaic.parser.types import Structured
+
 import tabmat as tm
 from tabmat.formula import (
     TabmatMaterializer,
@@ -1145,5 +1150,5 @@ def test_model_spec_pickleable(self, materializer):
         pickle.dump(ms.model_spec, o)
         o.seek(0)
         ms2 = pickle.load(o)
-        assert isinstance(ms, formulaic.parser.types.Structured)
+        assert isinstance(ms, Structured)
         assert ms2.lhs.formula.root == ["a"]