search

SalesforceAIResearch / uni2ts

[ICML2024] Unified Training of Universal Time Series Forecasting Transformers
Apache License 2.0
611 stars 49 forks source link

plot_single() -> Exception ignored in: <generator object PyTorchPredictor.predict at 0x000002A2CAF23220> #16

Closed fritol closed 2 months ago

fritol commented 2 months ago

when running the Getting started script: at line plot_single(...) it throws:

Exception ignored in: <generator object PyTorchPredictor.predict at 0x0000020E3897EE30>
Traceback (most recent call last):
  File "H:\anaconda3\envs\310\lib\site-packages\gluonts\torch\model\predictor.py", line 89, in predict
  File "H:\anaconda3\envs\310\lib\site-packages\torch\autograd\grad_mode.py", line 84, in __exit__
TypeError: 'NoneType' object is not callable

python 3.10.13 gluonts 0.14.4 matplotlib 3.8.4 conda on Win 10

installed with pip install -e .

gorold commented 2 months ago

Hey, this is the same issue as #3, do not run this code in a script, try it in a notebook.