Error when running timesfm[pax] on Google Colab

[woo2071]

I encountered an error while installing and running the timesfm[pax] package on Google Colab. After running pip install timesfm[pax], the following warning and compatibility issues appeared:

WARNING: The following packages were previously imported in this runtime:
[IPython, google]
You must restart the runtime in order to use newly installed versions.

WARNING: Upgrading ipython, ipykernel, tornado, prompt-toolkit, pyzmq can
cause your runtime to repeatedly crash or behave in unexpected ways and is not
recommended. If your runtime won't connect or execute code, you can reset it
with "Disconnect and delete runtime" from the "Runtime" menu.

This is followed by various dependency conflicts related to core packages like protobuf, ipython, ipykernel, and others, causing version incompatibilities:
google-ai-generativelanguage 0.6.10 requires protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.2, but you have protobuf 3.19.6 which is incompatible.
google-cloud-aiplatform 1.70.0 requires protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.2, but you have protobuf 3.19.6 which is incompatible.
google-cloud-functions 1.16.5 requires protobuf!=4.21.0,!=4.21.1,!=4.21.2,!=4.21.3,!=4.21.4,!=4.21.5,<6.0.0dev,>=3.20.2, but you have protobuf 3.19.6 which is incompatible.
...

After restarting the runtime, I'm no longer able to reconnect to the Colab environment. Even after resetting the runtime, the connection fails repeatedly. The logs show the following errors:

Oct 21, 2024, 5:33:55 PM WARNING AttributeError: module 'IPython.utils.traitlets' has no attribute 'Unicode'
Oct 21, 2024, 5:33:55 PM WARNING format_type = _traitlets.Unicode(_JAVASCRIPT_MODULE_MIME_TYPE)
Oct 21, 2024, 5:33:55 PM WARNING File "/usr/local/lib/python3.10/dist-packages/google/colab/data_table.py", line 320, in _JavascriptModuleFormatter
Oct 21, 2024, 5:33:55 PM WARNING class _JavascriptModuleFormatter(_IPython.core.formatters.BaseFormatter):
Oct 21, 2024, 5:33:55 PM WARNING File "/usr/local/lib/python3.10/dist-packages/google/colab/data_table.py", line 319, in
Oct 21, 2024, 5:33:55 PM WARNING from google.colab import data_table as _data_table
Oct 21, 2024, 5:33:55 PM WARNING File "/usr/local/lib/python3.10/dist-packages/google/colab/_interactive_table_hint_button.py", line 18, in
Oct 21, 2024, 5:33:55 PM WARNING from google.colab import _interactive_table_hint_button
Oct 21, 2024, 5:33:55 PM WARNING File "/usr/local/lib/python3.10/dist-packages/google/colab/init.py", line 19, in
Oct 21, 2024, 5:33:55 PM WARNING module = import(package, fromlist=[obj])

At this point, I'm unable to use the runtime at all and the connection won't re-establish. Could you please provide guidance on how to resolve this issue?

[siriuz42]

Ooops, such a not-officially-supported-Google-product.

Joking aside, I'm looking into this issue. Meanwhile please try an earlier version without the extra dependencies for now pip install timesfm==1.2.0 or earlier.