Fix

Quantco · Jul 5, 2024 · efc37ce · efc37ce
1 parent e9f5463
commit efc37ce
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/metalearners/slearner.py b/metalearners/slearner.py
@@ -155,7 +155,6 @@ def fit(
         self._validate_treatment(w)
         self._validate_outcome(y)
         self._fitted_treatments = convert_treatment(w)
-        self._n_features = X.shape[1]
 
         mock_model = self.nuisance_model_factory[_BASE_MODEL](
             **self.nuisance_model_params[_BASE_MODEL]