Authentication error when running the run command on Colab

When running the Colab example I am able tu run the code and successfully save the model inside the created bucket, but the run command keeps trying multiple GETs returning:

"Request is missing required authentication credential. Expected OAuth 2 access token, login cookie or other valid authentication credential. See https://developers.google.com/identity/sign-in/web/devconsole-project."

The run command I am using is

f = open('requirements.txt', 'w')
f.write('tensorflow-datasets\n')
f.write('pandas')
f.close()

tfc.run(
    entry_point=None,
    distribution_strategy="auto",
    requirements_txt="requirements.txt",
    chief_config=tfc.MachineConfig(
        cpu_cores=8,
        memory=30,
        accelerator_type=tfc.AcceleratorType.NVIDIA_TESLA_P100,
        accelerator_count=2,
    ),
    docker_image_bucket_name=BUCKET_NAME,
    docker_base_image='tensorflow/tensorflow:2.2.0-gpu',
    worker_count=0
)

I had to add docker_base_image='tensorflow/tensorflow:2.2.0-gpu' since I was getting a message saying that there is no docker for the TF version 2.2.0.