RuntimeError: General Union types are not currently supported.

[chiku-parida]

Need Solution!

fn = torch._C._jit_script_compile( qualified_name, ast, _rcb, get_default_args(obj) ) RuntimeError: General Union types are not currently supported. Only Union[T, NoneType] (i.e. Optional[T]) is supported.: /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]]: ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ <--- HERE """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 ` when loading module 'cdvae.pl_modules.gnn.DimeNetPlusPlusWrap'