Skip to content

Commit e078430

Browse files
committed
refactor(quantreg): type the demeaned-data lookup as DemeanedData
The lookup has held array-native DemeanedData entries since the shared demean cache stopped converting to DataFrames; the annotation still said pd.DataFrame. Annotation only, no runtime change.
1 parent 6503c65 commit e078430

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

pyfixest/estimation/quantreg/QuantregMulti.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111

1212
from pyfixest.demeaners import AnyDemeaner
1313
from pyfixest.estimation.formula.parse import Formula as FixestFormula
14+
from pyfixest.estimation.internals.demean_ import DemeanedData
1415
from pyfixest.estimation.internals.literals import (
1516
QuantregMethodOptions,
1617
QuantregMultiOptions,
@@ -35,7 +36,7 @@ def __init__(
3536
weights: str | None,
3637
weights_type: str | None,
3738
collin_tol: float,
38-
lookup_demeaned_data: dict[frozenset[int], pd.DataFrame],
39+
lookup_demeaned_data: dict[frozenset[int], DemeanedData],
3940
solver: SolverOptions = "np.linalg.solve",
4041
demeaner: AnyDemeaner | None = None,
4142
store_data: bool = True,

0 commit comments

Comments
 (0)