change the debug info of fx.GraphModule replay error

Author: JewelRoam

graph_net/torch/fx_graph_module_util.py

@@ -15,6 +15,7 @@ def get_torch_module_and_inputs(model_path):
 def _get_torch_module(model_path):
     py_module = load_module(f"{model_path}/model.py")
     torch_module_cls = py_module.GraphModule
+    torch_module_cls.__graph_net_file_path__ = model_path
     return torch_module_cls()
 
 

graph_net/torch/fx_graph_parse_util.py

@@ -221,15 +221,16 @@ def handle_underscore_suffix_difference():
 
     zip_filter_names = get_zip_filter_names()
 
-    def zip_filter_names_str():
+    def zip_filter_names_error_str():
         for triple in zip_filter_names:
             print(triple)
-        return "<printed before>"
+        error_model_path = module.__graph_net_file_path__
+        return f"{error_model_path=}"
 
     from pathlib import Path
 
     Path("/tmp/a.py").write_text(traced_module.code)
-    assert len(zip_filter_names) == 0, f"{zip_filter_names_str()=}"
+    assert len(zip_filter_names) == 0, zip_filter_names_error_str()
     return traced_module