From 63aaef9be87be3924e7340be9268a4f6f2b40342 Mon Sep 17 00:00:00 2001 From: mamingjie-China Date: Tue, 24 Dec 2019 10:30:28 +0800 Subject: [PATCH] run pre-commit --- x2paddle/convert.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/x2paddle/convert.py b/x2paddle/convert.py index 2c3e157..a7e4b60 100644 --- a/x2paddle/convert.py +++ b/x2paddle/convert.py @@ -193,7 +193,7 @@ def main(): assert args.framework is not None, "--framework is not defined(support tensorflow/caffe/onnx)" assert args.save_dir is not None, "--save_dir is not defined" - + if args.framework == "onnx": try: import onnxruntime as rt @@ -202,8 +202,10 @@ def main(): print("onnxruntime==1.0.0 is required") return except: - print("onnxruntime is not installed, use \"pip install onnxruntime==1.0.0\".") - + print( + "onnxruntime is not installed, use \"pip install onnxruntime==1.0.0\"." + ) + try: import paddle v0, v1, v2 = paddle.__version__.split('.') -- GitLab