RuntimeError: Undirected subgraphs not yet supported

[NBStarry]

I upgraded my PyG to 2.3.0 and got an error when using NeighborLoader to undirected data, the code did work with PyG=2.0.3 but failed with 2.3.0. How to solve the problem?

File "main.py", line 186, in train                                                                                                                
    for data_tuple in zip(*train_loader_list):                                                                                                      
  File "/etc/anaconda3/envs/mpit/lib/python3.8/site-packages/torch_geometric/loader/base.py", line 36, in __next__                                  
    return self.transform_fn(next(self.iterator))                                                                                                   
  File "/etc/anaconda3/envs/mpit/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 681, in __next__                                 
    data = self._next_data()                                                                                                                        
  File "/etc/anaconda3/envs/mpit/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 721, in _next_data
    data = self._dataset_fetcher.fetch(index)  # may raise StopIteration
  File "/etc/anaconda3/envs/mpit/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 52, in fetch
    return self.collate_fn(data)
  File "/etc/anaconda3/envs/mpit/lib/python3.8/site-packages/torch_geometric/loader/node_loader.py", line 117, in collate_fn
    out = self.node_sampler.sample_from_nodes(input_data)
File "/etc/anaconda3/envs/mpit/lib/python3.8/site-packages/torch_geometric/sampler/neighbor_sampler.py", line 174, in sample_from_nodes
    return node_sample(inputs, self._sample)
  File "/etc/anaconda3/envs/mpit/lib/python3.8/site-packages/torch_geometric/sampler/neighbor_sampler.py", line 358, in node_sample
    out = sample_fn(seed, seed_time)
  File "/etc/anaconda3/envs/mpit/lib/python3.8/site-packages/torch_geometric/sampler/neighbor_sampler.py", line 282, in _sample
    out = torch.ops.pyg.neighbor_sample(
  File "/etc/anaconda3/envs/mpit/lib/python3.8/site-packages/torch/_ops.py", line 143, in __call__
    return self._op(*args, **kwargs or {})
RuntimeError: Undirected subgraphs not yet supported

[rusty1s]

We are in the process of adding undirected support to pyg-lib. In the meantime, please call torch_geometric.typing.WITH_PYG_LIB = False at the start of your script for undirected sampling.