fix some test failures

uxlfoundation · Sep 2, 2024 · db660a6 · db660a6
1 parent e08f503
commit db660a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/daal4py/sklearn/ensemble/GBTDAAL.py b/daal4py/sklearn/ensemble/GBTDAAL.py
@@ -193,7 +193,7 @@ def fit(self, X, y):
         # Return the classifier
         return self
 
-    def _predict(self, X, resultsToEvaluate, pred_contribs, pred_interactions):
+    def _predict(self, X, resultsToEvaluate, pred_contrib=False, pred_interactions=False):
         # Input validation
         if not self.allow_nan_:
             X = check_array(X, dtype=[np.single, np.double])