phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 23:29:43 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

error checking

Browse source
This commit is contained in:
Xi Yan 2024-10-23 13:17:47 -07:00
parent d8bbce6f7c
commit c5db025320
1 changed files with 3 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

3
llama_stack/providers/impls/meta_reference/datasetio/datasetio.py
View file

@ -10,8 +10,10 @@ import pandas
from llama_models.llama3.api.datatypes import * # noqa: F403 from llama_models.llama3.api.datatypes import * # noqa: F403
from llama_stack.apis.datasetio import * # noqa: F403 from llama_stack.apis.datasetio import * # noqa: F403
import base64
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
from dataclasses import dataclass from dataclasses import dataclass
from urllib.parse import unquote
from llama_stack.providers.datatypes import DatasetsProtocolPrivate from llama_stack.providers.datatypes import DatasetsProtocolPrivate
from llama_stack.providers.utils.memory.vector_store import parse_data_url from llama_stack.providers.utils.memory.vector_store import parse_data_url
@ -53,6 +55,7 @@ class PandasDataframeDataset(BaseDataset):
return len(self.df) return len(self.df)
def __getitem__(self, idx): def __getitem__(self, idx):
assert self.df is not None, "Dataset not loaded. Please call .load() first"
if isinstance(idx, slice): if isinstance(idx, slice):
return self.df.iloc[idx].to_dict(orient="records") return self.df.iloc[idx].to_dict(orient="records")
else: else: