INNER CODE UNIT · Python

mlem_group_callback

iterative/mlem · mlem/cli/main.py:354

def mlem_group_callback(group: Typer, required: Optional[List[str]] = None):
    def decorator(f):
        @wraps(f)
        def inner(*args, **kwargs):
            ctx = click.get_current_context()
            if ctx.invoked_subcommand is not None:
                return None
            if required is not None:
                for req in required:
                    if req not in kwargs or kwargs[req] is None:
                        param = [
                            p
                            for p in ctx.command.get_params(ctx)
                            if p.name == req
                        ][0]
                        raise MissingParameter(ctx=ctx, param=param)
            return f(*args, **kwargs)

View source record →