소스 검색

Lint fixes

Lutz Roeder 6 년 전
부모
커밋
e3129a78a7
2개의 변경된 파일8개의 추가작업 그리고 8개의 파일을 삭제
  1. 7 7
      tools/onnx-script.py
  2. 1 1
      tools/tf-script.py

+ 7 - 7
tools/onnx-script.py

@@ -209,7 +209,7 @@ def metadata():
     generate_json(schemas, '../src/onnx-metadata.json')
 
 def convert():
-    file = sys.argv[2];
+    file = sys.argv[2]
     base, extension = os.path.splitext(file)
     if extension == '.mlmodel':
         pip_import('coremltools')
@@ -242,8 +242,8 @@ def convert():
 def optimize():
     import onnx
     from onnx import optimizer
-    file = sys.argv[2];
-    base, extension = os.path.splitext(file)
+    file = sys.argv[2]
+    base = os.path.splitext(file)
     onnx_model = onnx.load(file)
     passes = optimizer.get_available_passes()
     optimized_model = optimizer.optimize(onnx_model, passes)
@@ -253,13 +253,13 @@ def infer():
     import onnx
     import onnx.shape_inference
     from onnx import shape_inference
-    file = sys.argv[2];
-    base, extension = os.path.splitext(file)
+    file = sys.argv[2]
+    base = os.path.splitext(file)
     onnx_model = onnx.load(base + '.onnx')
-    onnx_model = onnx.shape_inference.infer_shapes(onnx_model);
+    onnx_model = onnx.shape_inference.infer_shapes(onnx_model)
     onnx.save(onnx_model, base + '.shape.onnx')
 
 if __name__ == '__main__':
     command_table = { 'metadata': metadata, 'convert': convert, 'optimize': optimize, 'infer': infer }
-    command = sys.argv[1];
+    command = sys.argv[1]
     command_table[command]()

+ 1 - 1
tools/tf-script.py

@@ -173,7 +173,7 @@ def convert_attr_value(attr_value):
 
 api_def_map = read_api_def_map('../third_party/tensorflow/tensorflow/core/api_def/base_api')
 
-input_file = '../third_party/tensorflow/tensorflow/core/ops/ops.pbtxt';
+input_file = '../third_party/tensorflow/tensorflow/core/ops/ops.pbtxt'
 
 ops_list = op_def_pb2.OpList()
 with open(input_file) as input_handle: