Quick fix for formulaic changes

Quantco · Dec 4, 2024 · dda850b · dda850b
1 parent 96aab71
commit dda850b
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/tabmat/formula.py b/src/tabmat/formula.py
@@ -10,7 +10,6 @@
 from formulaic import ModelMatrix, ModelSpec
 from formulaic.errors import FactorEncodingError
 from formulaic.materializers import FormulaMaterializer
-from formulaic.materializers.base import EncodedTermStructure
 from formulaic.materializers.types import FactorValues, NAAction, ScopedTerm
 from formulaic.parser.types import Term
 from formulaic.transforms import stateful_transform
@@ -24,6 +23,11 @@
 from .sparse_matrix import SparseMatrix
 from .split_matrix import SplitMatrix
 
+try:
+    from formulaic.materializers.base import EncodedTermStructure
+except ImportError:
+    from formulaic.materializers.types.formula_materializer import EncodedTermStructure
+
 
 class TabmatMaterializer(FormulaMaterializer):
     """Materializer for pandas input and tabmat output."""