Skip to content

Commit ef596f5

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
fix: Use GA generative_models path in tune_autorater to make it consistent with model parameter in Rapid Eval SDK evaluate function
PiperOrigin-RevId: 721178327
1 parent be7dcf7 commit ef596f5

File tree

2 files changed

+1
-7
lines changed

2 files changed

+1
-7
lines changed

vertexai/preview/evaluation/autorater_utils.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@
2020
import time
2121
from typing import Any, Dict, List, Literal, Optional, TYPE_CHECKING, Union
2222

23-
from vertexai.preview import generative_models
23+
from vertexai import generative_models
2424
from vertexai.preview.evaluation import _base as evaluation_base
2525
from vertexai.preview.evaluation import eval_task
2626
from vertexai.preview.evaluation.metrics import pairwise_metric

vertexai/preview/evaluation/eval_task.py

-6
Original file line numberDiff line numberDiff line change
@@ -56,12 +56,6 @@
5656
_RunnableType = Union[reasoning_engines.Queryable, Callable[[str], Dict[str, str]]]
5757
_ModelType = Union[generative_models.GenerativeModel, Callable[[str], str]]
5858

59-
_LOGGER = base.Logger(__name__)
60-
61-
62-
EvalResult = eval_base.EvalResult
63-
GenerativeModel = generative_models.GenerativeModel
64-
6559

6660
class EvalTask:
6761
"""A class representing an EvalTask.

0 commit comments

Comments
 (0)