Error with GATv2Conv

[Blaaake7]

Hi!
I have one question that truly need your help :
I am currently following this project.
When I run the classifier.py file and run the explainer.py file, I get the following error.

leesu0229@s2lab01:~/AdCPG$ python explainer.py
Traceback (most recent call last):
  File "/home/leesu0229/AdCPG/explainer.py", line 117, in <module>
    main()
  File "/home/leesu0229/AdCPG/explainer.py", line 114, in main
    generate_explanation(explanation_path, tokens, graph, node_data_path, explainer)
  File "/home/leesu0229/AdCPG/explainer.py", line 48, in generate_explanation
    explanation = explainer(data.x, data.edge_index, edge_attr=data.edge_attr, batch=data.batch, explain=True)
  File "/home/leesu0229/.local/lib/python3.10/site-packages/torch_geometric/explain/explainer.py", line 205, in call
    explanation = self.algorithm(
  File "/home/leesu0229/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1511, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
  File "/home/leesu0229/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1520, in _call_impl
    return forward_call(*args, **kwargs)
  File "/home/leesu0229/.local/lib/python3.10/site-packages/torch_geometric/explain/algorithm/gnn_explainer.py", line 87, in forward
    self._train(model, x, edge_index, target=target, index=index, **kwargs)
  File "/home/leesu0229/.local/lib/python3.10/site-packages/torch_geometric/explain/algorithm/gnn_explainer.py", line 123, in _train
    set_masks(model, self.edge_mask, edge_index, apply_sigmoid=True)
  File "/home/leesu0229/.local/lib/python3.10/site-packages/torch_geometric/explain/algorithm/utils.py", line 34, in set_masks
    module.explain = True
  File "/home/leesu0229/.local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1747, in setattr
    super().setattr(name, value)
  File "/home/leesu0229/.local/lib/python3.10/site-packages/torch_geometric/nn/conv/message_passing.py", line 747, in explain
    self.propagate = self.class._orig_propagate.get(
AttributeError: type object 'GATv2Conv' has no attribute '_orig_propagate'

I'm using

PyG : 2.5.2
Pytorch : 2.2.1

Can I ask how to solve this problem?
I really need your help..!

[rusty1s]

I pushed a fix in #9105. Do you mind trying out if it resolves your issues?

[muchen1453]

Hi!

I'm still running into such issues with 'TransformerConv': AttributeError: type object 'TransformerConv' has no attribute '_orig_propagate'. P.S. The issue for GATv2Conv has been solved!

Best