From 278e14ed858d4d862ad78d68344fc923cd040fca Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Thu, 12 Dec 2024 09:35:42 -0800 Subject: [PATCH] typo --- .../providers/remote/datasetio/huggingface/huggingface.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/llama_stack/providers/remote/datasetio/huggingface/huggingface.py b/llama_stack/providers/remote/datasetio/huggingface/huggingface.py index 0432be490..2fde7c3d0 100644 --- a/llama_stack/providers/remote/datasetio/huggingface/huggingface.py +++ b/llama_stack/providers/remote/datasetio/huggingface/huggingface.py @@ -30,7 +30,7 @@ def load_hf_dataset(dataset_def: Dataset): dataset = hf_datasets.Dataset.from_pandas(df) - # drop rows not specified by schema + # drop columns not specified by schema if dataset_def.dataset_schema: dataset = dataset.select_columns(list(dataset_def.dataset_schema.keys()))