5d3c9c8861ee658b7931bb374603f86987c17433,python/ray/tune/examples/mlflow_example.py,,,#,33

Before Change

    client = MlflowClient()
    experiment_id = client.create_experiment("test")

    trials = tune.run(
        easy_objective,
        name="mlflow",
        num_samples=5,
        loggers=DEFAULT_LOGGERS + (MLFLowLogger, ),
        config={
            "logger_config": {
                "mlflow_experiment_id": experiment_id,
            },
            "width": tune.randint(10, 100),
            "height": tune.randint(0, 100),
        })

    df = mlflow.search_runs([experiment_id])
    print(df)

After Change

if __name__ == "__main__":
    import argparse

    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--smoke-test", action="store_true", help="Finish quickly for testing")
    args, _ = parser.parse_known_args()

    if args.smoke_test:
        mlflow_tracking_uri = os.path.join(tempfile.gettempdir(), "mlruns")
    else:

Instances