Skip to content

Commit cee8d8b

Browse files
jsondaicopybara-github
authored andcommitted
fix: GenAI SDK client(evals) - Handle optional pandas dependency in type hints
PiperOrigin-RevId: 782072742
1 parent 0d63396 commit cee8d8b

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

vertexai/_genai/types.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -62,12 +62,16 @@ def __getattr__(name: str) -> typing.Any:
6262

6363
if typing.TYPE_CHECKING:
6464
import pandas as pd
65+
66+
PandasDataFrame = pd.DataFrame
6567
else:
66-
pd: typing.Type = Any
6768
try:
6869
import pandas as pd
70+
71+
PandasDataFrame = pd.DataFrame
6972
except ImportError:
7073
pd = None
74+
PandasDataFrame = Any
7175
if typing.TYPE_CHECKING:
7276
import yaml
7377
else:
@@ -6366,7 +6370,7 @@ class EvaluationDataset(_common.BaseModel):
63666370
eval_cases: Optional[list[EvalCase]] = Field(
63676371
default=None, description="""The evaluation cases to be evaluated."""
63686372
)
6369-
eval_dataset_df: Optional["pd.DataFrame"] = Field(
6373+
eval_dataset_df: Optional[PandasDataFrame] = Field(
63706374
default=None,
63716375
description="""The evaluation dataset in the form of a Pandas DataFrame.""",
63726376
)
@@ -6408,7 +6412,7 @@ class EvaluationDatasetDict(TypedDict, total=False):
64086412
eval_cases: Optional[list[EvalCaseDict]]
64096413
"""The evaluation cases to be evaluated."""
64106414

6411-
eval_dataset_df: Optional["pd.DataFrame"]
6415+
eval_dataset_df: Optional[PandasDataFrame]
64126416
"""The evaluation dataset in the form of a Pandas DataFrame."""
64136417

64146418
candidate_name: Optional[str]

0 commit comments

Comments
 (0)