google-research / timesfm

TimesFM (Time Series Foundation Model) is a pretrained time-series foundation model developed by Google Research for time-series forecasting.
https://research.google/blog/a-decoder-only-foundation-model-for-time-series-forecasting/
Apache License 2.0
3.78k stars 322 forks source link

Error when running timesfm[pax] on Google Colab #163

Open woo2071 opened 3 weeks ago

woo2071 commented 3 weeks ago

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 commented 3 weeks ago

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.