phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-16 06:53:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix datasetio

Browse source
This commit is contained in:
Xi Yan 2024-11-11 15:29:29 -05:00
parent acd055d763
commit e27c6e3662
5 changed files with 15 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/providers/inline/datasetio/localfs/datasetio.py
View file

@ -15,7 +15,7 @@ from dataclasses import dataclass
from llama_stack.providers.datatypes import DatasetsProtocolPrivate
from llama_stack.providers.utils.datasetio.url_utils import get_dataframe_from_url
from .config import MetaReferenceDatasetIOConfig
from .config import LocalfsDatasetIOConfig
class BaseDataset(ABC):
@ -77,8 +77,8 @@ class PandasDataframeDataset(BaseDataset):
self.df = self._validate_dataset_schema(df)
class MetaReferenceDatasetIOImpl(DatasetIO, DatasetsProtocolPrivate):
def __init__(self, config: MetaReferenceDatasetIOConfig) -> None:
class LocalfsDatasetIOImpl(DatasetIO, DatasetsProtocolPrivate):
def __init__(self, config: LocalfsDatasetIOConfig) -> None:
self.config = config
# local registry for keeping track of datasets within the provider
self.dataset_infos = {}