"""
Task utility functions shared across modules.
This module contains task-related utilities to avoid circular imports.
"""

from typing import Tuple, List


def update_tasks_mnist(key: str) -> Tuple[List[int], List[int]]:
    """
    Get task label configurations for MNIST experiments.
    
    Args:
        key (str): Task configuration key
        
    Returns:
        Tuple[List[int], List[int]]: Task 1 labels, Task 2 labels
    """
    tasks_dict = {
        "half": ([0, 1, 2, 3, 4], [5, 6, 7, 8, 9]),
        "round": ([0, 3, 6, 8, 9], [1, 2, 4, 5, 7]),
        "top": ([0, 2, 3, 8, 9], [1, 4, 5, 6, 7]),
        "equal": ([0, 1, 2, 3, 4], [0, 1, 2, 3, 4]),
    }
    return tasks_dict[key]
