req

Author: xadupre

onnx_diagnostic/export/dynamic_shapes.py

@@ -196,14 +196,14 @@ def _generic_walker(self, processor: Callable, args_kwargs: bool = False):
             kwargs.update(self.kwargs)
             res = self._generic_walker_step(processor, kwargs, self.dynamic_shapes)
             if args_kwargs:
-                ags = [None for _ in range(len(self.args))]
+                pgs = [None for _ in range(len(self.args))]
                 kws = {}
                 for k, v in res.items():
                     if k not in self.kwargs:
-                        ags[self.args_names.index(k)] = v
+                        pgs[self.args_names.index(k)] = v
                     else:
                         kws[k] = v
-                return ags, kws
+                return pgs, kws
             return res
 
         raise NotImplementedError(

requirements-dev.txt

@@ -4,7 +4,7 @@ diffusers>=0.30.0
 furo
 huggingface_hub
 matplotlib
-onnx-array-api
+onnx-array-api>=0.3.1
 git+https://github.com/microsoft/onnxscript.git
 openpyxl
 packaging