b7222818e9dd4d33ee933388e73a11ad8d0bf89b,mlflow/keras.py,,load_pyfunc,#Any#,58

Before Change

    :param model_file: Path to Keras model file.
    :return: PyFunc model.
    
    return _KerasModelWrapper(_load_model(model_file))


def load_model(path, run_id=None):
    

After Change

        // We create an use new Graph and Session and store them with the model
        // This way the model is independent on the global state.
        with graph.as_default():
            with sess.as_default():  // pylint:disable=not-context-manager
                K.set_learning_phase(0)
                m = _load_model(model_file)
        return _KerasModelWrapper(m, graph, sess)
    else:
        raise Exception("Unsupported backend "%s"" % K._BACKEND)

Instances