mypy happy?

Author: s3alfisc

pyfixest/estimation/collinearity.py

@@ -197,7 +197,12 @@ def drop_multicollinear_variables(
         collin_vars.extend(chol_vars)
         collin_index.extend(chol_idx)
 
-    if has_fixef and collin_tol_var > 0 and X_raw_sumsq is not None and X_demeaned.shape[1] > 0:
+    if (
+        has_fixef
+        and collin_tol_var > 0
+        and X_raw_sumsq is not None
+        and X_demeaned.shape[1] > 0
+    ):
         if chol_idx:
             X_raw_sumsq = np.delete(X_raw_sumsq, chol_idx)
         (X_demeaned, coefnames, var_vars, var_idx) = _drop_multicollinear_variables_var(

pyfixest/estimation/feols_compressed_.py

@@ -93,6 +93,7 @@ def __init__(
         fixef_maxiter: int,
         lookup_demeaned_data: dict[str, pd.DataFrame],
         solver: SolverOptions = "np.linalg.solve",
+        collin_tol_var: Optional[float] = None,
         demeaner_backend: DemeanerBackendOptions = "numba",
         store_data: bool = True,
         copy_data: bool = True,
@@ -116,6 +117,7 @@ def __init__(
             fixef_maxiter,
             lookup_demeaned_data,
             solver,
+            collin_tol_var,
             demeaner_backend,
             store_data,
             copy_data,