updated requirements

run_prothgt_app.py

@@ -18,8 +18,7 @@ class ProtHGT(torch.nn.Module):
 
         self.convs = torch.nn.ModuleList()
         for _ in range(num_layers):
-            # conv = HGTConv(hidden_channels, hidden_channels, data.metadata(), num_heads, group='sum')
-            conv = HGTConv(hidden_channels, hidden_channels, data.metadata(), num_heads)
+            conv = HGTConv(hidden_channels, hidden_channels, data.metadata(), num_heads, group='sum')
             self.convs.append(conv)
         
         self.mlp = MLP(mlp_hidden_layers , dropout=mlp_dropout, norm=None)

setup.sh

@@ -3,4 +3,4 @@ pip uninstall torch-scatter torch_sparse torch_geometric torch_cluster torch_geo
 # pip install torch-scatter -f https://data.pyg.org/whl/torch-1.12.0+cpu.html
 pip install torch_sparse -f https://data.pyg.org/whl/torch-1.12.0+cpu.html
 # pip install torch-cluster -f https://data.pyg.org/whl/torch-1.12.0+cpu.html
-pip install torch_geometric
+pip install torch_geometric==2.2.0