Skip to content

pixano_inference.tasks.utils

Task utilities.

get_tasks()

Get all tasks.

Source code in pixano_inference/tasks/utils.py 
22
23
24
def get_tasks() -> set[str]:
    """Get all tasks."""
    return {task for set in [STR_IMAGE_TASKS, STR_NLP_TASKS, STR_MULTIMODAL_TASKS, STR_VIDEO_TASKS] for task in set}

is_task(task)

Check if a task is valid.

Source code in pixano_inference/tasks/utils.py 
27
28
29
def is_task(task: str) -> bool:
    """Check if a task is valid."""
    return task in get_tasks()

str_to_task(task)

Convert a task string to its task.

Source code in pixano_inference/tasks/utils.py 
32
33
34
35
36
37
38
39
40
41
42
def str_to_task(task: str) -> Task:
    """Convert a task string to its task."""
    if task in STR_IMAGE_TASKS:
        return ImageTask(task)
    elif task in STR_NLP_TASKS:
        return NLPTask(task)
    elif task in STR_MULTIMODAL_TASKS:
        return MultimodalImageNLPTask(task)
    elif task in STR_VIDEO_TASKS:
        return VideoTask(task)
    raise ValueError(f"Invalid task '{task}'")