Fix return type of get_encoding_tableau (#209)

* Update circuits.py * Update circuits.py
Infleqtion · Jan 29, 2025 · 1918b0c · 1918b0c
1 parent 294c86e
commit 1918b0c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/qldpc/circuits.py b/qldpc/circuits.py
@@ -41,7 +41,7 @@ def qubit_func(*args: object, **kwargs: object) -> stim.Circuit:
 
 
 @restrict_to_qubits
-def get_encoding_tableau(code: codes.QuditCode) -> stim.Circuit:
+def get_encoding_tableau(code: codes.QuditCode) -> stim.Tableau:
     """Tableau to encode physical states at its input into logical states of the given code.
 
     For all j in {0, 1, ..., code.dimension - 1}, this tableau maps weight-one X_j and Z_j operators
@@ -85,7 +85,7 @@ def get_encoding_tableau(code: codes.QuditCode) -> stim.Circuit:
 
 
 @restrict_to_qubits
-def get_encoding_circuit(code: codes.QuditCode) -> stim.Tableau:
+def get_encoding_circuit(code: codes.QuditCode) -> stim.Circuit:
     """Circuit to encode physical states at its input into logical states of the given code.
 
     For all j in {0, 1, ..., code.dimension - 1}, this circuit maps weight-one X_j and Z_j operators