spell

Author: xadupre

.github/workflows/spell-check.yml

@@ -21,4 +21,4 @@ jobs:
 
       # Run codespell
       - name: Run codespell
-        run: codespell --skip="*.png,*.jpg,*.jpeg,*.gif,*.svg,*.ico,*.pdf,*.js,*.css,*.map" --ignore-words-list="nd,te,OT" --check-filenames
+        run: codespell --skip="*.png,*.jpg,*.jpeg,*.gif,*.svg,*.ico,*.pdf,*.js,*.css,*.map" --ignore-words-list="nd,te,OT,ags" --check-filenames

onnx_diagnostic/reference/torch_evaluator.py

@@ -219,7 +219,7 @@ def run(
                 # kernel execution
                 inputs = [(self.runtime_info[i].value if i else None) for i in kernel.input]
                 if kernel.has_subgraphs():
-                    res = kernel.run(*inputs, context=self.runtime_info)
+                    res = kernel.run(*inputs, context=self.runtime_info)  # type: ignore[call-arg]
                 else:
                     res = kernel.run(*inputs)
                 if isinstance(res, tuple):
@@ -271,6 +271,7 @@ def run_with_values(
             isinstance(a, torch_ops.OpRunValue) for a in args
         ), f"Unexpected type in args: {[type(a) for a in args]}"
         outputs = self.output_names
+        context = context or {}
 
         # sets constants
         for k, v in self.constants.items():

onnx_diagnostic/reference/torch_ops/control_flow.py

@@ -15,7 +15,7 @@ def __init__(
         self,
         node: onnx.NodeProto,
         version: Optional[int] = None,
-        parent: Optional["TorchOnnxEvaluator"] = None,  # noqa: F821
+        parent: Optional["onnx_diagnostic.reference.TorchOnnxEvaluator"] = None,  # noqa: F821
     ):
         super().__init__(node, version)
         assert (
@@ -36,5 +36,5 @@ class If_1(OpRunControlFlow):
     "If"
 
     def run(self, cond, context: Optional[Dict[str, Any]] = None):
-        rt = self.then_branch if cond.tensor.item() else self.else_branch
+        rt = self.then_branch if cond.tensor.item() else self.else_branch  # type: ignore[attr-defined]
         return rt.run_with_values(context=context)