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

fix all occurrence

Browse source
This commit is contained in:
Xi Yan 2025-03-15 14:20:17 -07:00
parent 917679cc2f
commit a9c662d68b
9 changed files with 177 additions and 70 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/distribution/routers/routers.py
View file

@ -537,7 +537,7 @@ class DatasetIORouter(DatasetIO):
logger.debug("DatasetIORouter.shutdown")
pass
async def get_rows_paginated(
async def iterrows(
self,
dataset_id: str,
rows_in_page: int,
@ -545,11 +545,9 @@ class DatasetIORouter(DatasetIO):
filter_condition: Optional[str] = None,
) -> IterrowsResponse:
logger.debug(
f"DatasetIORouter.get_rows_paginated: {dataset_id}, rows_in_page={rows_in_page}",
f"DatasetIORouter.iterrows: {dataset_id}, rows_in_page={rows_in_page}",
)
return await self.routing_table.get_provider_impl(
dataset_id
).get_rows_paginated(
return await self.routing_table.get_provider_impl(dataset_id).iterrows(
dataset_id=dataset_id,
rows_in_page=rows_in_page,
page_token=page_token,

6
llama_stack/distribution/ui/page/evaluations/native_eval.py
View file

@ -166,7 +166,7 @@ def run_evaluation_3():
eval_candidate = st.session_state["eval_candidate"]
dataset_id = benchmarks[selected_benchmark].dataset_id
rows = llama_stack_api.client.datasetio.get_rows_paginated(
rows = llama_stack_api.client.datasetio.iterrows(
dataset_id=dataset_id,
rows_in_page=-1,
)
@ -230,7 +230,9 @@ def run_evaluation_3():
output_res[scoring_fn] = []
output_res[scoring_fn].append(eval_res.scores[scoring_fn].score_rows[0])
progress_text_container.write(f"Expand to see current processed result ({i + 1} / {len(rows)})")
progress_text_container.write(
f"Expand to see current processed result ({i + 1} / {len(rows)})"
)
results_container.json(eval_res, expanded=2)
progress_bar.progress(1.0, text="Evaluation complete!")