fix: Use GA generative_models path in tune_autorater to make it con…

…sistent with `model` parameter in Rapid Eval SDK `evaluate` function PiperOrigin-RevId: 721132315
googleapis · Jan 30, 2025 · aaf11c3 · aaf11c3
1 parent be7dcf7
commit aaf11c3
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 7 deletions.
diff --git 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
@@ -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.