mypy

quantumlib · Jan 7, 2025 · 9a7209d · 9a7209d
1 parent be80886
commit 9a7209d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/qualtran/bloqs/gf_arithmetic/gf2_multiplication.py b/qualtran/bloqs/gf_arithmetic/gf2_multiplication.py
@@ -12,7 +12,7 @@
 #  See the License for the specific language governing permissions and
 #  limitations under the License.
 from functools import cached_property
-from typing import Dict, Sequence, Set, TYPE_CHECKING, Union
+from typing import Dict, Set, TYPE_CHECKING, Union
 
 import attrs
 import numpy as np
@@ -306,7 +306,9 @@ def on_classical_vals(self, g) -> Dict[str, 'ClassicalValT']:
         res = res[::-1]
         return {'g': res}
 
-    def build_composite_bloq(self, bb: 'BloqBuilder', g: Sequence['Soquet']) -> Dict[str, 'Soquet']:
+    def build_composite_bloq(
+        self, bb: 'BloqBuilder', g: np.ndarray['Soquet']
+    ) -> Dict[str, 'Soquet']:
         L, U, P = self.lup
         if is_symbolic(self.n):
             raise DecomposeTypeError(f"Symbolic decomposition isn't supported for {self}")