Skip to content

Commit 6e7e45a

Browse files
Fixing MetricValue justification type (mlflow#10131)
Signed-off-by: Sunish Sheth <[email protected]>
1 parent c468b19 commit 6e7e45a

File tree

2 files changed

+3
-2
lines changed

2 files changed

+3
-2
lines changed

mlflow/metrics/base.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@ class MetricValue:
1616
"""
1717

1818
scores: List[float] = None
19-
justifications: List[float] = None
19+
justifications: List[str] = None
2020
aggregate_results: Dict[str, float] = None
2121

2222

mlflow/models/evaluation/default_evaluator.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1292,7 +1292,6 @@ def _generate_model_predictions(self, compute_latency=False):
12921292
"""
12931293
Helper method for generating model predictions
12941294
"""
1295-
_logger.info("Computing model predictions.")
12961295

12971296
def predict_with_latency(X_copy):
12981297
y_pred_list = []
@@ -1333,6 +1332,8 @@ def predict_with_latency(X_copy):
13331332

13341333
X_copy = self.X.copy_to_avoid_mutation()
13351334
if self.model is not None:
1335+
_logger.info("Computing model predictions.")
1336+
13361337
if compute_latency:
13371338
model_predictions = predict_with_latency(X_copy)
13381339
else:

0 commit comments

Comments
 (0)