Skip to content

Commit

Permalink
Catch empty qubit declaration in writer
Browse files Browse the repository at this point in the history
  • Loading branch information
Guy Puts committed Oct 3, 2024
1 parent c883211 commit b1ac739
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions opensquirrel/writer/writer.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,10 +13,11 @@ def __init__(self, register_manager: RegisterManager) -> None:
self.register_manager = register_manager
qubit_register_size = self.register_manager.get_qubit_register_size()
qubit_register_name = self.register_manager.get_qubit_register_name()

bit_register_size = self.register_manager.get_bit_register_size()
bit_register_name = self.register_manager.get_bit_register_name()
self.output = "version 3.0\n\n{}\n{}\n".format(
f"qubit[{qubit_register_size}] {qubit_register_name}",
self.output = "version 3.0\n".format(
"\n{}\n{}\n"+f"qubit[{qubit_register_size}] {qubit_register_name}" if qubit_register_size > 0 else "",
f"bit[{bit_register_size}] {bit_register_name}\n" if bit_register_size > 0 else "",
)

Expand Down Expand Up @@ -71,7 +72,6 @@ def visit_comment(self, comment: Comment) -> None:

def circuit_to_string(circuit: Circuit) -> str:
writer_impl = _WriterImpl(circuit.register_manager)

circuit.ir.accept(writer_impl)

return writer_impl.output.rstrip() + "\n" # remove all trailing lines and leave only one

0 comments on commit b1ac739

Please sign in to comment.