AI Platform restore checkpoint not working in estimator train_and_evaluate

I am using AI Platform to train a Tensorflow model using the Estimator API. However, when the model saves a checkpoint and attempts to restore the checkpoint, it throws the error tensorflow.python.framework.errors_impl.NotFoundError: Unsuccessful TensorSliceReader constructor: Failed to find any matching files for gs://path/keras/keras_model.ckpt

It appears that this is an issue with restoring the metadata graph in Tensorflow, which is code that goes in the session setup (TensorFlow, why there are 3 files after saving the model?). However, since AI Platform abstracts this away from my configuration, how can I fix my issue?