diff --git a/vertexai/_genai/types/common.py b/vertexai/_genai/types/common.py index 97ecc9c3eb..d01c803a81 100644 --- a/vertexai/_genai/types/common.py +++ b/vertexai/_genai/types/common.py @@ -13325,6 +13325,12 @@ def to_bigframes( raise ValueError("Multimodal dataset bigquery source uri is not set.") return bigframes.pandas.read_gbq_table(self.bigquery_uri.removeprefix("bq://")) + def to_batch_job_source(self) -> BatchJobSource: + """Converts the dataset to a BatchJobSource.""" + return genai_types.BatchJobSource( + vertex_dataset_name=self.name, + ) + class MultimodalDatasetDict(TypedDict, total=False): """Represents a multimodal dataset."""