def register_trainable(name, trainable):
    """Register a trainable function or class.

    Args:
        name (str): Name to register.
        trainable (obj): Function or tune.Trainable class. Functions must
            take (config, status_reporter) as arguments and will be
            automatically converted into a class during registration.
    """

    from ray.tune.trainable import Trainable
    from ray.tune.function_runner import wrap_function

    if isinstance(trainable, type):
        logger.debug("Detected class for trainable.")
    elif isinstance(trainable, FunctionType):
        logger.debug("Detected function for trainable.")
        trainable = wrap_function(trainable)
    elif callable(trainable):
        logger.warning(
            "Detected unknown callable for trainable. Converting to class.")
        trainable = wrap_function(trainable)

    if not issubclass(trainable, Trainable):
        raise TypeError("Second argument must be convertable to Trainable",
                        trainable)
    _global_registry.register(TRAINABLE_CLASS, name, trainable)
def register_trainable ( name , trainable ) : from ray . tune . trainable import Trainable from ray . tune . function_runner import wrap_function if isinstance ( trainable , type ) : logger . debug ( "Detected class for trainable." ) elif isinstance ( trainable , FunctionType ) : logger . debug ( "Detected function for trainable." ) trainable = wrap_function ( trainable ) elif callable ( trainable ) : logger . warning ( "Detected unknown callable for trainable. Converting to class." ) trainable = wrap_function ( trainable ) if not issubclass ( trainable , Trainable ) : raise TypeError ( "Second argument must be convertable to Trainable" , trainable ) _global_registry . register ( TRAINABLE_CLASS , name , trainable )
Register a trainable function or class.

    Args:
        name (str): Name to register.
        trainable (obj): Function or tune.Trainable class. Functions must
            take (config, status_reporter) as arguments and will be
            automatically converted into a class during registration.