Fix test class signature. (#75)

Quantco · Aug 7, 2024 · 1c4f019 · 1c4f019
1 parent 7ddf544
commit 1c4f019
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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):