_pickle.PicklingError: Can't pickle typing.Union[torch.Tensor, NoneType]

[prey176]

We are working on Google Colab and were facing this issue, we were trying to train the model as stated in the READ ME with the following command ("python3 -m subgraph_matching.train --node_anchored"). The full error is stated as "_pickle.PicklingError: Can't pickle typing.Union[torch.Tensor, NoneType]: it's not the same object as typing.Union". We have also tried on the Colab GPU, but was getting the same error.

[qema]

Thanks for bringing up the issue. This seems to be a versioning issue with python/pytorch/pytorch geometric -- pyg-team/pytorch_geometric#1541 may be related?

The code has been tested with Python 3.7.4, PyTorch 1.4.0 and Pytorch Geometric 1.4.3. If you are using the latest version of the libraries, upgrading Python may help?

[prey176]

Thanks, I will look into it.