add prints

Signed-off-by: Jesse Chan <jesse.chan@databricks.com>
mlflow · May 6, 2024 · 0d116f6 · 0d116f6
1 parent 4dbd76b
commit 0d116f6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mlflow/models/evaluation/default_evaluator.py b/mlflow/models/evaluation/default_evaluator.py
@@ -1866,6 +1866,7 @@ def _evaluate(
             else:
                 # model is constructed from a user specified function or not provided
                 self.model_loader_module, self.raw_model = None, None
+            print("EXTRACTING PREDICTION FUNCTION")
             self.predict_fn, self.predict_proba_fn = _extract_predict_fn(model, self.raw_model)
 
             self.artifacts = {}
@@ -1919,6 +1920,7 @@ def evaluate(
         predictions=None,
         **kwargs,
     ):
+        print("USING DEFAULT EVALUTOR")
         if model is None and predictions is None and dataset.predictions_data is None:
             raise MlflowException(
                 message=(