diff --git a/framework/e2e/PaddleLT_new/PaddleLT.py b/framework/e2e/PaddleLT_new/PaddleLT.py index 398a73f923..d7f1e3ee0f 100644 --- 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 + ";"