869e56edc7e3153b534f065e6a73d924ec28bec3,vision/automl/edge_container_predict/automl_vision_edge_container_predict_test.py,,edge_container_predict_server_port,#,60

Before Change

        env={"DOCKER_API_VERSION": "1.38"})

    // Get the sample saved model.
    if not os.path.exists(MODEL_PATH):
        os.mkdir(MODEL_PATH)
    subprocess.check_output(
        ["gsutil", "-m", "cp", SAMPLE_SAVED_MODEL, MODEL_PATH])

    // Start the CPU docker.

After Change

        ["docker", "pull", CPU_DOCKER_GCS_PATH],
        env={"DOCKER_API_VERSION": "1.38"})

    if os.environ.get("TRAMPOLINE_V2") == "true":
        // Use /tmp
        model_path = tempfile.TemporaryDirectory()
    else:
        // Use project directory with Trampoline V1.
        model_path = tempfile.TemporaryDirectory(dir=os.path.dirname(__file__))
    print("Using model_path: {}".format(model_path))
    // Get the sample saved model.
    subprocess.check_output(
        ["gsutil", "-m", "cp", SAMPLE_SAVED_MODEL, model_path.name])

Instances