diff --git a/tests/test_metalearner.py b/tests/test_metalearner.py index 5721f5c..d9ac1f6 100644 --- a/tests/test_metalearner.py +++ b/tests/test_metalearner.py @@ -107,7 +107,7 @@ def fit_all_treatment( def predict(self, X, is_oos, oos_method=None): return np.zeros((len(X), self.n_variants - 1, 1)) - def evaluate(self, X, y, w, is_oos, oos_method=None): + def evaluate(self, X, y, w, is_oos, oos_method=None, scoring=None): return {} def predict_conditional_average_outcomes(self, X, is_oos, oos_method=None):