From 813eec4c995df71c921e504eb0d4b8d1eaf21411 Mon Sep 17 00:00:00 2001 From: Apurva Koti Date: Tue, 7 May 2024 13:06:10 -0700 Subject: [PATCH] init Signed-off-by: Apurva Koti --- mlflow/metrics/genai/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/mlflow/metrics/genai/__init__.py b/mlflow/metrics/genai/__init__.py index 8e092810b1886..2b8c93f388588 100644 --- 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",