try with latest pyg version

requirements.txt

@@ -1,5 +1,5 @@
 pandas
-torch==1.12.1
-# torch-sparse -f https://data.pyg.org/whl/torch-1.13.0+cpu.html
-torch-geometric==2.2.0
+torch
+torch-sparse
+torch-geometric
 gdown

run_prothgt_app.py

@@ -18,7 +18,8 @@ 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, group='sum')
+            conv = HGTConv(hidden_channels, hidden_channels, data.metadata(), num_heads)
             self.convs.append(conv)
         
         self.mlp = MLP(mlp_hidden_layers , dropout=mlp_dropout, norm=None)

setup.sh

@@ -1,6 +0,0 @@
-#!/bin/bash
-pip install torch==1.12.1
-pip uninstall torch-scatter torch-sparse torch-geometric torch-cluster torch-geometric --y
-# 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