def METHOD_NAME(source_type: Union[TYPE_PYTHON, TYPE_SQL]) -> Callable:
    if source_type == TYPE_PYTHON:
        return NotebookStepFuncFactory.run_python_step
    elif source_type == TYPE_SQL:
        return NotebookStepFuncFactory.run_sql_step
    else:
        raise NotImplementedError(
            f"Run function for source type {source_type} is not implemented."
        )