37d2a3fee7048675cb8f34df68c7b616553918ac,tutorials/tensorflow/mlflow_gcp/trainer/task.py,,train_and_evaluate,#Any#,144

Before Change

                                    tf_signature_def_key="serving_default",
                                    artifact_path="model")
        // Reloading the model
        pyfunc_model = mlflow.pyfunc.load_model(
            mlflow.get_artifact_uri("model"))
        logging.info("Uploading TensorFlow events as a run artifact.")
        mlflow.log_artifacts(tensorboard_path)
        logging.info(
            "Launch TensorBoard with:\n\ntensorboard --logdir=%s" %

After Change

                                    artifact_path="model")
        // Reloading the model
        if args.model_reload:
            mlflow.pyfunc.load_model(mlflow.get_artifact_uri("model"))

        logging.info("Uploading TensorFlow events as a run artifact.")
        mlflow.log_artifacts(tensorboard_path)
        logging.info(

Instances