doc fix

Author: Strilanc

.github/workflows/ci.yml

@@ -53,7 +53,7 @@ jobs:
       MACOSX_DEPLOYMENT_TARGET: "10.15"
       CIBW_BUILD: "${{ matrix.os_dist.dist }}"
       CIBW_ARCHS_MACOS: "${{ matrix.os_dist.macosarch }}"
-      CIBW_TEST_REQUIRES: pytest stim sinter pygltflib
+      CIBW_TEST_REQUIRES: pytest stim~=1.14.dev sinter pygltflib
       CIBW_TEST_COMMAND: pytest {project}/src
     steps:
       - uses: actions/checkout@v3

doc/chromobius.pyi

@@ -240,7 +240,8 @@ class CompiledDecoder:
             >>> # Decode with Chromobius.
             >>> dem = repetition_color_code.detector_error_model()
             >>> decoder = chromobius.compile_decoder_for_dem(dem)
-            >>> pred, weights = decoder.predict_obs_flips_from_dets_bit_packed(dets)
+            >>> result = decoder.predict_weighted_obs_flips_from_dets_bit_packed(dets)
+            >>> pred, weights = result
         """
 def compile_decoder_for_dem(
     dem: stim.DetectorErrorModel,

doc/chromobius_api_reference.md

@@ -351,6 +351,7 @@ def predict_weighted_obs_flips_from_dets_bit_packed(
         >>> # Decode with Chromobius.
         >>> dem = repetition_color_code.detector_error_model()
         >>> decoder = chromobius.compile_decoder_for_dem(dem)
-        >>> pred, weights = decoder.predict_obs_flips_from_dets_bit_packed(dets)
+        >>> result = decoder.predict_weighted_obs_flips_from_dets_bit_packed(dets)
+        >>> pred, weights = result
     """
 ```

src/chromobius/pybind/chromobius.pybind.cc

@@ -315,7 +315,8 @@ PYBIND11_MODULE(chromobius, m) {
                 >>> # Decode with Chromobius.
                 >>> dem = repetition_color_code.detector_error_model()
                 >>> decoder = chromobius.compile_decoder_for_dem(dem)
-                >>> pred, weights = decoder.predict_obs_flips_from_dets_bit_packed(dets)
+                >>> result = decoder.predict_weighted_obs_flips_from_dets_bit_packed(dets)
+                >>> pred, weights = result
         )DOC")
             .data());