Spaces:

DevastatingRPG
/

toolforge-env

Sleeping

toolforge-env / server /inputs /simulated /task_selector.py

Upload folder using huggingface_hub

750e08b verified 7 months ago

388 Bytes

from typing import Dict, Any, List

from server.inputs.simulated.tasks import TASKS

class TaskSelector:
    def __init__(self):
        pass

    def next_task_list(self, task_id: str) -> Dict:
        match = next((item for item in TASKS if item.get("task_id") == task_id), None)
        if match is None:
            raise ValueError(f"Unknown task_id: {task_id}")
        return match

	from typing import Dict, Any, List

	from server.inputs.simulated.tasks import TASKS

	class TaskSelector:
	def __init__(self):
	pass

	def next_task_list(self, task_id: str) -> Dict:
	match = next((item for item in TASKS if item.get("task_id") == task_id), None)
	if match is None:
	raise ValueError(f"Unknown task_id: {task_id}")
	return match