Update inverse_confusion_matrix.py

unitaryfund · Feb 9, 2024 · 502ebfd · 502ebfd
1 parent f686a78
commit 502ebfd
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/mitiq/rem/inverse_confusion_matrix.py b/mitiq/rem/inverse_confusion_matrix.py
@@ -36,8 +36,11 @@ def sample_probability_vector(
 
     # split the binary strings into an array of ints
     bitstrings = (
-        np.apply_along_axis(
-            np.fromstring, 1, binary_strings[:, None], dtype="U1"
+        np.apply_along_axis(  # type: ignore
+            func1d=np.fromstring,
+            axis=1,
+            arr=binary_strings[:, None],
+            dtype="U1",  # type: ignore
         )
         .astype(np.uint8)
         .tolist()