fix misplaced type annotations

onnx · Sep 6, 2020 · e527581 · e527581
1 parent 755c49d
commit e527581
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/onnxoptimizer/__init__.py b/onnxoptimizer/__init__.py
@@ -39,8 +39,7 @@
 get_available_passes = C.get_available_passes
 
 
-# type: (ModelProto, Optional[Sequence[Text]], bool) -> ModelProto
-def optimize(model, passes=None, fixed_point=False):
+def optimize(model, passes=None, fixed_point=False):  # type: (ModelProto, Optional[Sequence[Text]], bool) -> ModelProto
     if passes is None:
         passes = ['eliminate_nop_transpose',
                   'eliminate_nop_pad',