From c5db0253207de310210bc69e38b8a5d0fe12ff1d Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Wed, 23 Oct 2024 13:17:47 -0700 Subject: [PATCH] error checking --- .../providers/impls/meta_reference/datasetio/datasetio.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/llama_stack/providers/impls/meta_reference/datasetio/datasetio.py b/llama_stack/providers/impls/meta_reference/datasetio/datasetio.py index adda1e3bd..938cc23e3 100644 --- a/llama_stack/providers/impls/meta_reference/datasetio/datasetio.py +++ b/llama_stack/providers/impls/meta_reference/datasetio/datasetio.py @@ -10,8 +10,10 @@ import pandas from llama_models.llama3.api.datatypes import * # noqa: F403 from llama_stack.apis.datasetio import * # noqa: F403 +import base64 from abc import ABC, abstractmethod from dataclasses import dataclass +from urllib.parse import unquote from llama_stack.providers.datatypes import DatasetsProtocolPrivate from llama_stack.providers.utils.memory.vector_store import parse_data_url @@ -53,6 +55,7 @@ class PandasDataframeDataset(BaseDataset): return len(self.df) def __getitem__(self, idx): + assert self.df is not None, "Dataset not loaded. Please call .load() first" if isinstance(idx, slice): return self.df.iloc[idx].to_dict(orient="records") else: