complex

xadupre · xadupre · commit 685d073704c1 · 2024-11-12T17:38:54.000+01:00
diff --git a/onnx_array_api/graph_api/graph_builder.py b/onnx_array_api/graph_api/graph_builder.py
@@ -536,6 +536,8 @@ def make_nodes(
             if isinstance(value, TensorProto):
                 value.name = name
             self.initializers_dict[name] = value
+            if name in builder.initializers_dict_sources:
+                self.initializers_dict_sources[name] = builder.initializers_dict_sources[name]
 
             self.constants_[name] = None
             self.set_shape(name, builder._known_shapes[init])
diff --git a/onnx_array_api/npx/npx_jit_eager.py b/onnx_array_api/npx/npx_jit_eager.py
@@ -204,7 +204,7 @@ def make_key(self, *values: List[Any], **kwargs: Dict[str, Any]) -> Tuple[Any, .
                 if k in self.kwargs_to_input_:
                     res.append(type(v))
                     res.append(v)
-                elif isinstance(v, (int, float, str, type, bool, DType)):
+                elif isinstance(v, (int, float, str, type, bool, complex, DType)):
                     res.append(k)
                     res.append(type(v))
                     res.append(v)
