def METHOD_NAME(cls: T, migration_step: T) -> Callable:
    """Register the migration step.
    :param migration_step: The migration step to register
    :return: The registered migration step
    """
    def wrapper(func):
        """Add the dunder collector to the func.
        :param func: The function to decorate
        :returns: The decorated function
        """
        migration_step.function_name = func.__name__
        func.__migration_step__ = migration_step
        return func
    return wrapper