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'