aip_sdk.DatasetHandle

aip_sdk.DatasetHandle(dataset_id: str, version: int, version_id: str, client: APIClient)

Handle for accessing golden dataset data.

aip_sdk.DatasetHandle.dataset_id

aip_sdk.DatasetHandle.dataset_id = dataset_id

No docstring is defined in the source.

aip_sdk.DatasetHandle.pull

aip_sdk.DatasetHandle.pull(max_retries: int = 3, backoff_factor: float = 1.0) -> pd.DataFrame

Download the golden dataset as a DataFrame.

Fetches a presigned URL from GET /datasets/{id}/versions/{version_id}/download and downloads the Parquet file directly from MinIO, with automatic retry on transient network errors and 5xx responses.

Parameters

• max_retries int: Maximum download attempts (default 3).
• backoff_factor float: Multiplier for exponential wait between retries (default 1.0).

Returns

• pd.DataFrame: DataFrame with golden dataset

Raises

• NotFoundError: If dataset version doesn't exist.
• DownloadError: If the download fails after all retries, or immediately on 4xx.

aip_sdk.DatasetHandle.version

aip_sdk.DatasetHandle.version = version

No docstring is defined in the source.

aip_sdk.DatasetHandle.version_id

aip_sdk.DatasetHandle.version_id = version_id

No docstring is defined in the source.