pyzmq
используется в работе бэкэнда, поэтому возможно, что при обновлении возникает несовместимость с системой Colab.
В качестве обходного пути я бы рекомендовал установить pyzmq
илюбая библиотека требует обновления в virtualenv, чтобы изолировать ее от системного Python, используемого Colab.