diff --git a/src/braket/circuits/text_diagram_builders/unicode_circuit_diagram.py b/src/braket/circuits/text_diagram_builders/unicode_circuit_diagram.py index 8136bb7cc..9e1779cb7 100644 --- a/src/braket/circuits/text_diagram_builders/unicode_circuit_diagram.py +++ b/src/braket/circuits/text_diagram_builders/unicode_circuit_diagram.py @@ -251,8 +251,9 @@ def _build_box( symbol = f"┤ {symbol} ├" return top, symbol, bottom - @staticmethod + @classmethod def _build_verbatim_box( + cls, symbol: Literal["StartVerbatim", "EndVerbatim"], connection: Literal["above", "below", "both", "none"], ) -> str: @@ -267,7 +268,7 @@ def _build_verbatim_box( top = "║" symbol = "╨" top = _fill_symbol(top, " ") - symbol = _fill_symbol(symbol, UnicodeCircuitDiagram._qubit_line_character()) + symbol = _fill_symbol(symbol, cls._qubit_line_character()) bottom = _fill_symbol(bottom, " ") return top, symbol, bottom