plt fix PaddleLT, test=model

PaddlePaddle · Oct 10, 2024 · 18be77d · 18be77d
1 parent 0c25d57
commit 18be77d
Showing 1 changed file with 12 additions and 23 deletions.
diff --git a/framework/e2e/PaddleLT_new/PaddleLT.py b/framework/e2e/PaddleLT_new/PaddleLT.py
@@ -24,34 +24,23 @@ def test_module_layer(title, layerfile, testing, device_place_id):
     allure.dynamic.feature("case")
 
     flags_str = ""
-    if os.environ.get("FRAMEWORK") == "paddle":
-        import paddle
+    # if os.environ.get("FRAMEWORK") == "paddle":
+    #     import paddle
 
-        flags_str += "paddle_commit=" + paddle.__git_commit__ + ";"
+    #     flags_str += "paddle_commit=" + paddle.__git_commit__ + ";"
 
     for key, value in os.environ.items():
         if key.startswith("FLAGS_"):
             flags_str = flags_str + key + "=" + value + ";"
-    flags_str += (
-        "TESTING="
-        + os.environ.get("TESTING")
-        + ";"
-        + "CUDA_VISIBLE_DEVICES="
-        + os.environ.get("CUDA_VISIBLE_DEVICES")
-        + ";"
-        + "FRAMEWORK="
-        + os.environ.get("FRAMEWORK")
-        + ";"
-        + "USE_PADDLE_MODEL="
-        + os.environ.get("USE_PADDLE_MODEL")
-        + ";"
-        + "wheel_url="
-        + os.environ.get("wheel_url")
-        + ";"
-        + "docker_image="
-        + os.environ.get("docker_image")
-        + ";"
-    )
+
+    flags_str = f"paddle_commit={os.environ.get('paddle_commit')};"
+    flags_str += f"TESTING={os.environ.get('TESTING')};"
+    flags_str += f"CUDA_VISIBLE_DEVICES={os.environ.get('CUDA_VISIBLE_DEVICES')};"
+    flags_str += f"FRAMEWORK={os.environ.get('FRAMEWORK')};"
+    flags_str += f"USE_PADDLE_MODEL={os.environ.get('USE_PADDLE_MODEL')};"
+    flags_str += f"wheel_url={os.environ.get('wheel_url')};"
+    flags_str += f"docker_image={os.environ.get('docker_image')};"
+
     for key, value in os.environ.items():
         if key.startswith("PLT_"):
             flags_str = flags_str + key + "=" + value + ";"