INNER CODE UNIT · Python

assert_upload_runs

operatorai/modelstore · workflows/actions/cli.py:34

def assert_upload_runs(domain: str, model_path: str) -> str:
    """Runs the 'python -m modelstore upload' command"""
    assert os.path.exists(model_path)
    model_id = _run_cli_command(
        [
            "upload",
            domain,
            model_path,
        ]
    )
    assert model_id is not None
    assert model_id != ""
    print(f"✅  CLI command uploaded model={model_id}")
    return model_id


def assert_download_runs(domain: str, model_id: str):
    """Runs the 'python -m modelstore download' command

View source record →