diff --git a/opensquirrel/writer/writer.py b/opensquirrel/writer/writer.py index b5686a9..fe9fce5 100644 --- a/opensquirrel/writer/writer.py +++ b/opensquirrel/writer/writer.py @@ -16,9 +16,11 @@ def __init__(self, register_manager: RegisterManager) -> None: 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".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 "", + self.output = "version 3.0{}{}{}{}\n".format( + "\n\n" if qubit_register_size > 0 or bit_register_size > 0, + f"qubit[{qubit_register_size}] {qubit_register_name}" if qubit_register_size > 0 else "", + "\n" if qubit_register_size > 0 and bit_register_size > 0 else "", + f"bit[{bit_register_size}] {bit_register_name}" if bit_register_size > 0 else "", ) def visit_bit(self, bit: Bit) -> str: