6/5000 Error at run time

[SilenceRan]

I followed the guide step by step to configure the environment and run run_traffic_imputation.py. Here is my error:
Traceback (most recent call last):
File "./experiments/run_spatiotemporal_imputation.py", line 13, in
from tsl.data import SpatioTemporalDataModule, ImputationDataset
File "/home_nfs/miniconda3/envs/imputeformer/lib/python3.8/site-packages/tsl/data/init.py", line 1, in
from .batch import DisjointBatch, StaticBatch, static_graph_collate
File "/home_nfs/miniconda3/envs/imputeformer/lib/python3.8/site-packages/tsl/data/batch.py", line 14, in
from .data import Data
File "/home_nfs/miniconda3/envs/imputeformer/lib/python3.8/site-packages/tsl/data/data.py", line 14, in
from tsl.ops.connectivity import reduce_graph
ImportError: cannot import name 'reduce_graph' from 'tsl.ops.connectivity' (/home_nfs/miniconda3/envs/imputeformer/lib/python3.8/site-packages/tsl/ops/connectivity.py)

I think something is wrong with the tls package. How can I solve the problem? Look forward to your reply.

[tongnie]

Hi, what's the version of your tsl package. torch_spatiotemporal==0.1.1 is recommended and in this version we only need to import the following modules from tsl.data

[tongnie]

Hi @SilenceRan , thanks for your interests in our work! I have fixed these unsolved issues now.

In this latest version, we have now included the modified tsl (which is different from the pip-installed one) package in this repository so there is no need to install tsl.

In addition, I have provided the complete conda environments in conda_env.yml, and I have tested that this environment can successfully run the experiment script.

Now you can deal with all dependencies with:
conda env create -f conda_env.yml
conda activate imputeformer

Sorry for late reply and thanks again for your patience :)
Kind regards
Tong