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]]:
"""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'
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]]: