From 0d116f62c8dc86acd23894ff02899e379c2c276c Mon Sep 17 00:00:00 2001
From: Jesse Chan <jesse.chan@databricks.com>
Date: Mon, 6 May 2024 01:20:55 -0400
Subject: [PATCH] add prints

Signed-off-by: Jesse Chan <jesse.chan@databricks.com>
---
 mlflow/models/evaluation/default_evaluator.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/mlflow/models/evaluation/default_evaluator.py b/mlflow/models/evaluation/default_evaluator.py
index 33da62ec9ac2d..4512dd4af9ad0 100644
--- 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=(