search

Graph-Machine-Learning-Group / spin

Official repository for the paper "Learning to Reconstruct Missing Data from Spatiotemporal Graphs with Sparse Observations" (NeurIPS 2022)
https://arxiv.org/abs/2205.13479
MIT License
41 stars 8 forks source link

BUG: TypeError: on_train_batch_start() takes 3 positional arguments but 4 were given #10

Closed guo-huojian closed 11 months ago

guo-huojian commented 11 months ago 
File "/home/yu/anaconda3/envs/spin/lib/python3.8/site-packages/tsl/imputers/imputer.py", line 134, in on_train_batch_start
    super(Imputer, self).on_train_batch_start(batch, batch_idx, unused)

TypeError: on_train_batch_start() takes 3 positional arguments but 4 were given

marshka commented 11 months ago

Hi, this issue is caused by a change in pytorch lightning's API. Please see #4.