[Core] return dict from ModelWrapper.get_opset_imports

Author: maltanar

src/qonnx/core/modelwrapper.py

@@ -740,5 +740,5 @@ def set_tensor_sparsity(self, tensor_name, sparsity_dict):
             qnt_annotations.append(qa)
 
     def get_opset_imports(self):
-        """Returns a list of imported opsets as (domain, version) tuples."""
-        return [(opset.domain, opset.version) for opset in self._model_proto.opset_import]
+        """Returns a list of imported opsets as a {domain, version} dictionary."""
+        return {opset.domain: opset.version for opset in self._model_proto.opset_import}

tests/core/test_modelwrapper.py

@@ -68,7 +68,7 @@ def test_modelwrapper():
     inp_sparsity = {"dw": {"kernel_shape": [3, 3]}}
     model.set_tensor_sparsity(first_conv_iname, inp_sparsity)
     assert model.get_tensor_sparsity(first_conv_iname) == inp_sparsity
-    assert model.get_opset_imports() == [("", 8)]
+    assert model.get_opset_imports() == {"": 8}
 
 
 def test_modelwrapper_set_get_rm_initializer():