Почему я получаю эту ошибку, когда запускаю пример документации PyTorch L Lighting на TPU?

Question

Я пробую учебник PyTorch Lightning по MNIST на colab TPU. Но, запустив его, я получаю

Исключение в устройстве =TPU:0: модуль 'torch_xla.core.xla_model' не имеет атрибута 'rendezvous'.

Заранее благодарны за Вашу помощь.