Fix mypy errors

qir-alliance · Jun 4, 2024 · 4810634 · 4810634
1 parent 462b7e1
commit 4810634
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pyqir/pyqir/_native.pyi b/pyqir/pyqir/_native.pyi
@@ -298,7 +298,7 @@ class Builder:
         """
         ...
 
-    def phi(self, value: Type) -> Instruction:
+    def phi(self, value: Type) -> Phi:
         """
         Inserts a phi node.
 
@@ -768,7 +768,7 @@ class Opcode(Enum):
 class Phi(Instruction):
     """A phi node instruction."""
 
-    def add_incoming(self, value: Value, block: BasicBlock):
+    def add_incoming(self, value: Value, block: BasicBlock) -> None:
         """Adds an incoming value to the end of the phi list."""
         ...
 

diff --git a/pyqir/tests/test_phi.py b/pyqir/tests/test_phi.py
@@ -53,7 +53,7 @@ def test_ghz_loop() -> None:
     for i in range(num_qubits):
         pyqir.rt.result_record_output(builder, pyqir.result(context, i), nullptr)
 
-    builder.ret()
+    builder.ret(None)
     mod.verify()
 
     ir = str(mod)