TaskRegistry

GitHub source

TaskRegistry

evo.compute.tasks.common.runner.TaskRegistry

Registry mapping parameter types to their TaskRunner subclasses.

This is a singleton that stores the mapping from parameter class types to their corresponding TaskRunner subclass.

register

register(param_type: type[TParams], runner_cls: type[TaskRunner[TParams, TResultModel, TResult]]) -> None

Register a TaskRunner subclass for a parameter type.

Args: param_type: The parameter class (e.g., KrigingParameters). runner_cls: The TaskRunner subclass that handles this parameter type.

get_runner

get_runner(param_type: type[TParams]) -> type[TaskRunner[TParams, TResultModel, TResult]] | None

Get the TaskRunner subclass for a parameter type.

Args: param_type: The parameter class to look up.

Returns: The registered TaskRunner subclass, or None if not found.

get_runner_for_params

get_runner_for_params(params: TParams) -> type[TaskRunner[TParams, TResultModel, TResult]]

Get the TaskRunner subclass for a parameter instance.

Args: params: A parameter object instance.

Returns: The registered TaskRunner subclass.

Raises: TypeError: If no runner is registered for the parameter type.

clear

clear() -> None

Clear all registered runners (mainly for testing).