Import genai in mlflow.metrics (mlflow#11336)

Signed-off-by: harupy <[email protected]> Signed-off-by: Daniel Lok <[email protected]>
daniellok-db · Mar 6, 2024 · 509cacd · 509cacd
1 parent 22f06f5
commit 509cacd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/mlflow/metrics/__init__.py b/mlflow/metrics/__init__.py
@@ -1,3 +1,4 @@
+from mlflow.metrics import genai
 from mlflow.metrics.base import (
     MetricValue,
 )
@@ -457,4 +458,5 @@ def f1_score() -> EvaluationMetric:
     "binary_f1_score",
     "token_count",
     "latency",
+    "genai",
 ]