phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-01 16:24:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

remove prints

Browse source
This commit is contained in:
Xi Yan 2024-11-14 00:05:24 -05:00
parent 0c9998ead5
commit 2080cc89fa
1 changed files with 1 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/providers/inline/eval/meta_reference/eval.py
View file

@ -53,7 +53,7 @@ class MetaReferenceEvalImpl(Eval, EvalTasksProtocolPrivate):
async def initialize(self) -> None: async def initialize(self) -> None:
self.kvstore = await kvstore_impl(self.config.kvstore) self.kvstore = await kvstore_impl(self.config.kvstore)
# Load existing datasets from kvstore # Load existing eval_tasks from kvstore
start_key = EVAL_TASKS_PREFIX start_key = EVAL_TASKS_PREFIX
end_key = f"{EVAL_TASKS_PREFIX}\xff" end_key = f"{EVAL_TASKS_PREFIX}\xff"
stored_eval_tasks = await self.kvstore.range(start_key, end_key) stored_eval_tasks = await self.kvstore.range(start_key, end_key)
@ -62,19 +62,15 @@ class MetaReferenceEvalImpl(Eval, EvalTasksProtocolPrivate):
eval_task = EvalTask.model_validate_json(eval_task) eval_task = EvalTask.model_validate_json(eval_task)
self.eval_tasks[eval_task.identifier] = eval_task self.eval_tasks[eval_task.identifier] = eval_task
print(f"Meta Reference Eval initialized with {self.eval_tasks}")
async def shutdown(self) -> None: ... async def shutdown(self) -> None: ...
async def register_eval_task(self, task_def: EvalTask) -> None: async def register_eval_task(self, task_def: EvalTask) -> None:
print("REGISTER EVAL TASK")
# Store in kvstore # Store in kvstore
key = f"{EVAL_TASKS_PREFIX}{task_def.identifier}" key = f"{EVAL_TASKS_PREFIX}{task_def.identifier}"
await self.kvstore.set( await self.kvstore.set(
key=key, key=key,
value=task_def.json(), value=task_def.json(),
) )
print(self.eval_tasks)
self.eval_tasks[task_def.identifier] = task_def self.eval_tasks[task_def.identifier] = task_def
async def validate_eval_input_dataset_schema(self, dataset_id: str) -> None: async def validate_eval_input_dataset_schema(self, dataset_id: str) -> None: