update tests

Author: oaksharks

deeptables/models/deeptable.py

@@ -512,7 +512,7 @@ def fit_cross_validation(self, X, y, X_eval=None, X_test=None, num_folds=5, stra
         logger.info(f'fit_cross_validation taken {time.time() - start}s')
 
         if oof_metrics is not None:
-            return oof_proba_fixed, eval_proba_mean_fixed, test_proba_mean_fixed, oof_scores
+            return oof_proba_fixed, eval_proba_mean_fixed, test_proba_mean_fixed, oof_scores,
         else:
             return oof_proba_fixed, eval_proba_mean_fixed, test_proba_mean_fixed
 

deeptables/models/hyper_dt.py

@@ -176,7 +176,7 @@ def fit_cross_validation(self, X, y, eval_set=None, metrics=None, pos_label=None
 
         self.classes_ = getattr(self.model, 'classes_', None)
 
-        return scores, oof_proba, oof_scores
+        return scores, oof_proba, oof_scores, None, None  # TODO: add xvals, yvals
 
     def predict(self, X, **kwargs):
         return self.model.predict(X, **kwargs)