Ignore mypy warnings from pre-existing code

microsoft · Jul 25, 2024 · 9d9db37 · 9d9db37
1 parent a0cc3db
commit 9d9db37
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/onnxscript/tools/benchmark/benchmark_helpers.py b/onnxscript/tools/benchmark/benchmark_helpers.py
@@ -287,7 +287,7 @@ def common_export(
     if exporter == "script":
         torch.onnx.export(
             model,
-            inputs,
+            inputs,  # type: ignore[arg-type]
             filename,
             do_constant_folding=False,
             input_names=[f"input{i}" for i in range(len(inputs))],

diff --git a/onnxscript/tools/transformers_models/__init__.py b/onnxscript/tools/transformers_models/__init__.py
@@ -41,7 +41,7 @@ def export_to_onnx(
             prog = torch.onnx.export(model, args, dynamo=True)  # pylint: disable=no-value-for-parameter
         else:
             prog = torch.onnx.dynamo_export(model, *args)
-    model_proto = prog.model_proto
+    model_proto = prog.model_proto  # type: ignore[union-attr]
     if optimize:
         model_proto = onnxscript.optimizer.optimize(
             model_proto,