autointent.schemas.node_validation.get_optuna_class

autointent.schemas.node_validation.get_optuna_class(param_type)

Get the Optuna class for the given parameter type.

If the (possibly annotated or union) type includes int or float, this function returns the corresponding search space class.

Parameters:

param_type (type) – Parameter type (could be a union, annotated type, or container)

Returns:

ParamSpaceInt if the type matches int, ParamSpaceFloat if it matches float, else None.

Return type:

type[ParamSpaceInt | ParamSpaceFloat] | None