Use astor instead of ast to unparse source code of paddle model, to avoid unstable result between different python versions.

Author: Xreki

graph_net/paddle/validate.py

@@ -12,6 +12,7 @@
 import graph_net
 import os
 import ast
+import astor
 import paddle
 
 
@@ -29,7 +30,7 @@ def _get_sha_hash(content):
     return m.hexdigest()
 
 
-def _extract_forward_source(model_path):
+def _extract_forward_source(model_path, class_name):
     source = None
     with open(f"{model_path}/model.py", "r") as f:
         source = f.read()
@@ -38,18 +39,18 @@ def _extract_forward_source(model_path):
     forward_code = None
 
     for node in tree.body:
-        if isinstance(node, ast.ClassDef) and node.name == "GraphModule":
+        if isinstance(node, ast.ClassDef) and node.name == class_name:
             for fn in node.body:
                 if isinstance(fn, ast.FunctionDef) and fn.name == "forward":
-                    return ast.unparse(fn)
+                    return astor.to_source(fn)
     return None
 
 
 def check_graph_hash(args):
     model_path = args.model_path
     file_path = f"{model_path}/graph_hash.txt"
     if args.dump_graph_hash_key:
-        model_str = _extract_forward_source(model_path)
+        model_str = _extract_forward_source(model_path, class_name="GraphModule")
         assert model_str is not None, f"model_str of {args.model_path} is None."
         new_hash_text = _get_sha_hash(model_str)
 

requirements.txt

@@ -1 +1,2 @@
 pre-commit
+astor

tools/ci/check_validate.sh

@@ -37,6 +37,7 @@ function prepare_torch_env() {
     LOG "[INFO] Update pip ..."
     env http_proxy="" https_proxy="" pip install -U pip > /dev/null
     [ $? -ne 0 ] && LOG "[FATAL] Update pip failed!" && exit -1
+    pip install astor
     # install torch
     pip install torch==2.6.0 --index-url https://download.pytorch.org/whl/cu118 > /dev/null
     [ $? -ne 0 ] && LOG "[FATAL] Install torch2.6.0 failed!" && exit -1
@@ -56,7 +57,7 @@ function prepare_paddle_env() {
     env http_proxy="" https_proxy="" pip install -U pip > /dev/null
     [ $? -ne 0 ] && LOG "[FATAL] Update pip failed!" && exit -1
     # install paddle
-    pip uninstall torch==2.7.0 --yes
+    pip install astor
     LOG "[INFO] Install paddlepaddle-develop ..."
     python -m pip install --pre paddlepaddle-gpu -i https://www.paddlepaddle.org.cn/packages/nightly/cu118/ > /dev/null
     [ $? -ne 0 ] && LOG "[FATAL] Install paddlepaddle-develop failed!" && exit -1