Skip to content

Datasets

wrap_map_func(func, result_key)

Wrapper to adapt functions to datasets map interface.

Source code in src/kibad_llm/utils/datasets.py 
 7
 8
 9
10
11
12
13
def wrap_map_func(func, result_key: str) -> Callable:
    """Wrapper to adapt functions to datasets map interface."""

    def wrapper(*args, **kwargs) -> dict[str, Any]:
        return {result_key: func(*args, **kwargs)}

    return wrapper