Skip to content

Commit

Permalink
Merge remote-tracking branch 'origin/main' into fix-formulaic-reference
Browse files Browse the repository at this point in the history
  • Loading branch information
MarcAntoineSchmidtQC committed Dec 5, 2024
2 parents ea6704f + 0ec0935 commit d345a21
Showing 1 changed file with 5 additions and 1 deletion.
6 changes: 5 additions & 1 deletion src/tabmat/formula.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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."""
Expand Down

0 comments on commit d345a21

Please sign in to comment.