ImportError: cannot import name 'cat_all_gather' from 'pytorchvideo.layers.distributed'

Hi Im facing this issue when running the last code block in the install.md file. Im trying to run the implementation on the kinetics400 dataset but still keep facing this error.

Please reply back ASAP. Thanks