diff --git a/tests/onnx/auto_scan_test.py b/tests/onnx/auto_scan_test.py index e566611060cf65b8c9e07189907ee3313154e307..614b20e2e9642f7bc57808ca3203048ee659808f 100644 --- a/tests/onnx/auto_scan_test.py +++ b/tests/onnx/auto_scan_test.py @@ -48,7 +48,7 @@ settings.register_profile( derandomize=True, report_multiple_bugs=False) if float(os.getenv('TEST_NUM_PERCENT_CASES', default='1.0')) < 1 or \ - os.getenv('HYPOTHESIS_TEST_PROFILE', 'dev') == 'ci': + os.getenv('HYPOTHESIS_TEST_PROFILE', 'dev') == 'ci': settings.load_profile("ci") else: settings.load_profile("dev") diff --git a/tests/onnx/onnxbase.py b/tests/onnx/onnxbase.py index 7f2fcc75827c4b76da159a1f19c07f40cb4dba66..b5f9db718b8c8be7fe2d76c002cc04233029ee97 100644 --- a/tests/onnx/onnxbase.py +++ b/tests/onnx/onnxbase.py @@ -189,12 +189,12 @@ class ONNXConverter(object): paddle_tensor_feed = list() for i in range(len(self.input_feed)): paddle_numpy_feed.append(self.input_feed[self.inputs_name[i]]) - paddle_tensor_feed.append(paddle.to_tensor(self.input_feed[self.inputs_name[i]])) + paddle_tensor_feed.append( + paddle.to_tensor(self.input_feed[self.inputs_name[i]])) if self.run_dynamic: - paddle_path = os.path.join( - self.pwd, self.name, - self.name + '_' + str(ver) + '_paddle/') + paddle_path = os.path.join(self.pwd, self.name, + self.name + '_' + str(ver) + '_paddle/') import sys sys.path.append(paddle_path) from x2paddle_code import main