Пример #1
0
def test_run_git():
    with TempDir() as tmp:
        with update_temp_env(
            {mlflow.tracking._TRACKING_URI_ENV_VAR: tmp.path()}):
            res = invoke_cli_runner(
                cli.run, [GIT_PROJECT_URI, "--no-conda", "-P", "alpha=0.5"])
            assert "python train.py 0.5 0.1" in res.output
Пример #2
0
def test_run_local():
    with TempDir() as tmp:
        with update_temp_env({mlflow.tracking._TRACKING_URI_ENV_VAR: tmp.path()}):
            excitement_arg = 2
            name = "friend"
            invoke_cli_runner(cli.run, [TEST_PROJECT_DIR, "-e", "greeter", "-P",
                                        "greeting=hi", "-P", "name=%s" % name,
                                        "-P", "excitement=%s" % excitement_arg])
Пример #3
0
def test_run_local():
    with TempDir() as tmp:
        with update_temp_env(
            {mlflow.tracking._TRACKING_URI_ENV_VAR: tmp.path()}):
            excitement_arg = 2
            res = invoke_cli_runner(cli.run, [
                TEST_PROJECT_DIR, "-e", "greeter", "-P", "greeting=hi",
                "--no-conda", "-P", "name=friend", "-P",
                "excitement=%s" % excitement_arg
            ])
            _assert_succeeded(res.output)
Пример #4
0
def test_run_git():
    with TempDir() as tmp:
        with update_temp_env({mlflow.tracking._TRACKING_URI_ENV_VAR: tmp.path()}):
            invoke_cli_runner(cli.run, [GIT_PROJECT_URI, "--no-conda", "-P", "alpha=0.5"])