Fix mypy issues.

CQCL · Aug 23, 2023 · 313f91d · 313f91d
1 parent 9f4d0a0
commit 313f91d
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/pytket/tests/simulator/tket_sim_backend.py b/pytket/tests/simulator/tket_sim_backend.py
@@ -101,10 +101,12 @@ def required_predicates(self) -> List[Predicate]:
             NoFastFeedforwardPredicate(),
         ]
 
-    def rebase_pass(self) -> BasePass:
+    def rebase_pass(self, **kwargs: KwargTypes) -> BasePass:
         return auto_rebase_pass(_GATE_SET)
 
-    def default_compilation_pass(self, optimisation_level: int = 1) -> BasePass:
+    def default_compilation_pass(
+        self, optimisation_level: int = 1, **kwargs: KwargTypes
+    ) -> BasePass:
         assert optimisation_level in range(3)
         if optimisation_level == 0:
             return SequencePass([DecomposeBoxes(), self.rebase_pass()])
@@ -155,7 +157,9 @@ class TketSimShotBackend(TketSimBackend):
     _supports_state = False
     _supports_contextual_optimisation = True
 
-    def default_compilation_pass(self, optimisation_level: int = 1) -> BasePass:
+    def default_compilation_pass(
+        self, optimisation_level: int = 1, **kwargs: KwargTypes
+    ) -> BasePass:
         assert optimisation_level in range(3)
         if optimisation_level == 0:
             return SequencePass([DecomposeBoxes(), self.rebase_pass()])