complex

xadupre · xadupre · commit 6527538c0236 · 2024-11-12T16:30:23.000+01:00
diff --git a/onnx_array_api/npx/npx_jit_eager.py b/onnx_array_api/npx/npx_jit_eager.py
@@ -167,7 +167,7 @@ def make_key(self, *values: List[Any], **kwargs: Dict[str, Any]) -> Tuple[Any, .
                         f"to the attribute list, v={v}."
                     )
                 res.append(v.key)
-            elif isinstance(v, (int, float, bool, DType)):
+            elif isinstance(v, (int, float, bool, complex, DType)):
                 if iv in self.kwargs_to_input_:
                     res.append(self.kwargs_to_input_[iv])
                 res.append(type(v))

Original file line number	Diff line number	Diff line change
`@@ -167,7 +167,7 @@ def make_key(self, values: List[Any], *kwargs: Dict[str, Any]) -> Tuple[Any, .`
`167`	`167`	`f"to the attribute list, v={v}."`
`168`	`168`	`)`
`169`	`169`	`res.append(v.key)`
`170`		`- elif isinstance(v, (int, float, bool, DType)):`
	`170`	`+ elif isinstance(v, (int, float, bool, complex, DType)):`
`171`	`171`	`if iv in self.kwargs_to_input_:`
`172`	`172`	`res.append(self.kwargs_to_input_[iv])`
`173`	`173`	`res.append(type(v))`