Skip to content

Commit

Permalink
Fix attribute reference.
Browse files Browse the repository at this point in the history
  • Loading branch information
kklein committed Aug 15, 2024
1 parent 413e5b0 commit 59554b1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion metalearners/xlearner.py
Original file line number Diff line number Diff line change
Expand Up @@ -172,7 +172,7 @@ def fit_all_treatment(
if not hasattr(self, "_treatment_cv_split_indices"):
raise ValueError(
"The nuisance models need to be fitted before fitting the treatment models. "
"In particular, the MetaLearner's attribute _cvs, "
"In particular, the MetaLearner's attribute _treatment_cv_split_indices, "
"typically set during nuisance fitting, does not exist."
)
qualified_fit_params = self._qualified_fit_params(fit_params)
Expand Down

0 comments on commit 59554b1

Please sign in to comment.