error at test : test_example_imputation

TorchSpatiotemporal / tsl

tsl: a PyTorch library for processing spatiotemporal data.

https://torch-spatiotemporal.readthedocs.io/

MIT License

267 stars 24 forks source link

error at test : test_example_imputation #16

Closed Tianxu-Jia closed 1 year ago

Tianxu-Jia commented 1 year ago

The error message: FAILED test_example_imputation.py::test_example_imputation - TypeError: on_train_batch_start() takes 3 positional arguments but 4 were given

Can you please fix the error.

Many thanks

marshka commented 1 year ago

Hi, please check issue #10, commit https://github.com/TorchSpatiotemporal/tsl/commit/1f55a6d95c7c361b2ce4b661ef4774d44be6fc5c should have fixed the problem.