Fix num_neighbours type hint

oscarhiggott · Jun 2, 2022 · 9ff3966 · 9ff3966
1 parent 55722d6
commit 9ff3966
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pymatching/matching.py b/src/pymatching/matching.py
@@ -620,7 +620,7 @@ def num_detectors(self) -> int:
 
     def decode(self,
                z: Union[np.ndarray, List[int]],
-               num_neighbours: int = 30,
+               num_neighbours: Union[int, None] = 30,
                return_weight: bool = False
                ) -> Union[np.ndarray, Tuple[np.ndarray, int]]:
         """Decode the syndrome `z` using minimum-weight perfect matching