add domain assert

Quantco · Jul 23, 2024 · 5ea40d0 · 5ea40d0
1 parent ffc5e5d
commit 5ea40d0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_adapt.py b/tests/test_adapt.py
@@ -262,4 +262,5 @@ def test_if_adaptation_const():
     b = argument(Tensor(numpy.float32, ("N",)))
     cond = op18.equal(sq, b)
     out = op18.if_(cond, then_branch=lambda: [sq], else_branch=lambda: [sq])
-    build({"b": b}, {"out": out[0]})
+    model = build({"b": b}, {"out": out[0]})
+    assert model.domain == "" or model.domain == "ai.onnx"