update genn_astar

app.py

@@ -19,28 +19,27 @@ def _handle_ged_solve(
     gexf_1_path: str,
     gexf_2_path: str
 ):
+    # check the input files
     if gexf_1_path is None:
         raise gr.Error("Please upload file completely!")
     if gexf_2_path is None:
         raise gr.Error("Please upload file completely!")
     
-    start_time = time.time()
-    print("111")
+    # check the pretrained file
     dirs = pygm.utils.user_cache_dir("pygmtools")
-    print(dirs) 
-    print(os.path.exists(dirs))
     if not os.path.exists(dirs):
         os.makedirs(dirs)
-    print("222")
-    shutil.move(src=PRETRAINED_PATH, dst=os.path.join(dirs, PRETRAINED_PATH))
-    print("333")
+    if os.path.exists(PRETRAINED_PATH):
+        shutil.move(src=PRETRAINED_PATH, dst=os.path.join(dirs, PRETRAINED_PATH))
+    
+    # begin solve
+    start_time = time.time()
     astar(
         g1_path=gexf_1_path,
         g2_path=gexf_2_path,
         output_path="media",
         filename="ged_image"
     )
-    print("444")
     solved_time = time.time() - start_time
     
     message = "Successfully solve the GED problem, using time ({:.3f}s).".format(solved_time)

genn_astar.py

@@ -141,9 +141,7 @@ def astar(
     A2 = torch.tensor(pygm.utils.from_networkx(graph2)).float().to(device)
     
     # Caculate the ged
-    print("111111")
     x_pred = pygm.genn_astar(feat1, feat2, A1, A2, return_network=False)
-    print("222222")
      
     # Plot
     draw(graph1, colors1, labels1, output_filename.format(1), "Graph1")