init

Signed-off-by: Apurva Koti <[email protected]>
apurva-koti · May 7, 2024 · 813eec4 · 813eec4
1 parent 9f5e0d2
commit 813eec4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mlflow/metrics/genai/__init__.py b/mlflow/metrics/genai/__init__.py
@@ -1,5 +1,5 @@
 from mlflow.metrics.genai.base import EvaluationExample
-from mlflow.metrics.genai.genai_metric import make_genai_metric
+from mlflow.metrics.genai.genai_metric import make_custom_genai_metric, make_genai_metric
 from mlflow.metrics.genai.metric_definitions import (
     answer_correctness,
     answer_relevance,
@@ -11,6 +11,7 @@
 __all__ = [
     "EvaluationExample",
     "make_genai_metric",
+    "make_custom_genai_metric",
     "answer_similarity",
     "answer_correctness",
     "faithfulness",