phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-03 13:02:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix iterrows

Browse source
This commit is contained in:
Xi Yan 2025-03-15 14:24:46 -07:00
parent 82ec0d24f3
commit a197101635
2 changed files with 10 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/remote/datasetio/huggingface/huggingface.py
View file

@ -84,7 +84,7 @@ class HuggingfaceDatasetIOImpl(DatasetIO, DatasetsProtocolPrivate):
start_index = start_index or 0
if limit == -1:
if limit is None or limit == -1:
end = len(loaded_dataset)
else:
end = min(start_index + limit, len(loaded_dataset))
@ -93,7 +93,7 @@ class HuggingfaceDatasetIOImpl(DatasetIO, DatasetsProtocolPrivate):
return IterrowsResponse(
data=rows,
next_index=end,
next_index=end if end < len(loaded_dataset) else None,
)
async def append_rows(self, dataset_id: str, rows: List[Dict[str, Any]]) -> None: