Error:General Union types are not currently supported. Only Union[T, NoneType] (i.e. Optional[T]) is supported.: File "/home/lvqy/anaconda3/envs/ZernikeMetric/lib/python3.8/site-packages/torch_cluster/rw.py", line 18

[dabianzhixing]

Hi, when I execute the command:

python script/pretrain.py -c config/pretrain/mc_gearnet_edge.yaml --gpus [0]

There is an error occurring. How to solve This problem?

RuntimeError: General Union types are not currently supported. Only Union[T, NoneType] (i.e. Optional[T]) is supported.: File "/home/lvqy/anaconda3/envs/ZernikeMetric/lib/python3.8/site-packages/torch_cluster/rw.py", line 18 num_nodes: Optional[int] = None, return_edge_indices: bool = False, ) -> Union[Tensor, Tuple[Tensor, Tensor]]:

    """Samples random walks of length :obj:`walk_length` from all node indices
    in :obj:`start` in the graph given by :obj:`(row, col)` as described in the

[dabianzhixing]

solved, pip install torch_cluster==1.5.9