From aaf11c3bfc1d041e42f1ca87aa35c3c1e75e175c Mon Sep 17 00:00:00 2001 From: A Vertex SDK engineer Date: Wed, 29 Jan 2025 15:40:59 -0800 Subject: [PATCH] fix: Use GA generative_models path in `tune_autorater` to make it consistent with `model` parameter in Rapid Eval SDK `evaluate` function PiperOrigin-RevId: 721132315 --- vertexai/preview/evaluation/autorater_utils.py | 2 +- vertexai/preview/evaluation/eval_task.py | 6 ------ 2 files changed, 1 insertion(+), 7 deletions(-) diff --git a/vertexai/preview/evaluation/autorater_utils.py b/vertexai/preview/evaluation/autorater_utils.py index d4cd0ca69c..8c8fc1ca31 100644 --- a/vertexai/preview/evaluation/autorater_utils.py +++ b/vertexai/preview/evaluation/autorater_utils.py @@ -20,7 +20,7 @@ import time from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union -from vertexai.preview import generative_models +from vertexai import generative_models from vertexai.preview.evaluation import _base as evaluation_base from vertexai.preview.evaluation import eval_task from vertexai.preview.evaluation.metrics import pairwise_metric diff --git a/vertexai/preview/evaluation/eval_task.py b/vertexai/preview/evaluation/eval_task.py index c795d84124..b137d16e8c 100644 --- a/vertexai/preview/evaluation/eval_task.py +++ b/vertexai/preview/evaluation/eval_task.py @@ -56,12 +56,6 @@ _RunnableType = Union[reasoning_engines.Queryable, Callable[[str], Dict[str, str]]] _ModelType = Union[generative_models.GenerativeModel, Callable[[str], str]] -_LOGGER = base.Logger(__name__) - - -EvalResult = eval_base.EvalResult -GenerativeModel = generative_models.GenerativeModel - class EvalTask: """A class representing an EvalTask.