benedekrozemberczki / pytorch_geometric_temporal

PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)
MIT License
2.58k stars 367 forks source link

Fix `test_tsagcn` #253

Open cshjin opened 9 months ago

cshjin commented 9 months ago

Fix the issue in #186 Changes

test/attention_test.py ........ [ 8%] test/batch_test.py ............................. [ 39%] test/dataset_test.py .......................................... [ 84%] test/heterogeneous_test.py . [ 86%] test/recurrent_test.py ............. [100%]

======================================= 93 passed in 119.87s (0:01:59) =======================================