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.