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

comments

Browse source
This commit is contained in:
Xi Yan 2024-10-23 13:16:08 -07:00
parent 5e1323b5bf
commit d8bbce6f7c
2 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -64,6 +64,7 @@ class PandasDataframeDataset(BaseDataset):
assert self.df is not None, "Dataset loading failed. Please check logs."
# note that we will drop any columns in dataset that are not in the schema
self.df = self.df[self.dataset_def.dataset_schema.keys()]
# check all columns in dataset schema are present