update acurding to seongjunyun#41

OmerBenHayun · Jul 1, 2023 · 0391639 · 0391639
1 parent 0df0c69
commit 0391639
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -5,9 +5,9 @@
 from model_fastgtn import FastGTNs
 import pickle
 import argparse
-from torch_geometric.utils import f1_score, add_self_loops
+from torch_geometric.utils import  add_self_loops
 from sklearn.metrics import f1_score as sk_f1_score
-from utils import init_seed, _norm
+from utils import init_seed, _norm,f1_score
 import copy
 
 if __name__ == '__main__':