feat: expose C++ properties to Pybind and update Sinter build dependencies

TAG=agy

Author: aria-googler

BUILD

@@ -19,9 +19,9 @@ py_wheel(
     name="tesseract_decoder_wheel",
     distribution = "tesseract_decoder",
     deps=[
-        "//src:tesseract_decoder",
+        "//src:_core",
         "//src/py:generated_stubs",
-        "//src/py/_tesseract_py_util:_tesseract_py_util",
+        "//src/py:tesseract_decoder",
         ":package_data",
     ],
     version = "$(VERSION)",

src/error_correlations.test.cc

@@ -1,58 +1,61 @@
-#include "gtest/gtest.h"
 #include "error_correlations.h"
+
 #include <vector>
 
+#include "gtest/gtest.h"
+
 using namespace tesseract;
 
 TEST(TwoPassCorrelationsTest, JointProbabilities) {
-    stim::DetectorErrorModel dem(R"DEM(
+  stim::DetectorErrorModel dem(R"DEM(
         error(0.1) D0 ^ D1
         error(0.2) D0
     )DEM");
 
-    auto joint = get_hyperedge_joint_probabilities(dem);
-    
-    Hyperedge h0 = {0};
-    Hyperedge h1 = {1};
-    
-    // P(D0) = 0.1 XOR 0.2 = 0.1*(1-0.2) + 0.2*(1-0.1) = 0.08 + 0.18 = 0.26
-    EXPECT_NEAR(joint[h0][h0], 0.26, 1e-6);
-    // P(D1) = 0.1
-    EXPECT_NEAR(joint[h1][h1], 0.1, 1e-6);
-    // P(D0 and D1) = 0.1
-    EXPECT_NEAR(joint[h0][h1], 0.1, 1e-6);
-    EXPECT_NEAR(joint[h1][h0], 0.1, 1e-6);
+  std::vector<int> global_det_to_comp_id = {0, 1};
+  auto joint = get_hyperedge_joint_probabilities(dem, global_det_to_comp_id);
+
+  Hyperedge h0 = {0};
+  Hyperedge h1 = {1};
+
+  // P(D0) = 0.1 XOR 0.2 = 0.1*(1-0.2) + 0.2*(1-0.1) = 0.08 + 0.18 = 0.26
+  EXPECT_NEAR(joint[h0][h0], 0.26, 1e-6);
+  // P(D1) = 0.1
+  EXPECT_NEAR(joint[h1][h1], 0.1, 1e-6);
+  // P(D0 and D1) = 0.1
+  EXPECT_NEAR(joint[h0][h1], 0.1, 1e-6);
+  EXPECT_NEAR(joint[h1][h0], 0.1, 1e-6);
 }
 
 TEST(TwoPassCorrelationsTest, ImpliedProbabilities) {
-    JointProbsMap joint;
-    Hyperedge h0 = {0};
-    Hyperedge h1 = {1};
-    
-    joint[h0][h0] = 0.2;
-    joint[h1][h1] = 0.1;
-    joint[h0][h1] = 0.05;
-    joint[h1][h0] = 0.05;
-
-    auto implied = get_implied_hyperedge_probabilities(joint);
-    
-    // P(D1 | D0) = 0.05 / 0.2 = 0.25
-    bool found = false;
-    for (const auto& imp : implied[h0]) {
-        if (imp.affected_hyperedge == h1) {
-            EXPECT_NEAR(imp.probability, 0.25, 1e-6);
-            found = true;
-        }
+  JointProbsMap joint;
+  Hyperedge h0 = {0};
+  Hyperedge h1 = {1};
+
+  joint[h0][h0] = 0.2;
+  joint[h1][h1] = 0.1;
+  joint[h0][h1] = 0.05;
+  joint[h1][h0] = 0.05;
+
+  auto implied = get_implied_hyperedge_probabilities(joint);
+
+  // P(D1 | D0) = 0.05 / 0.2 = 0.25
+  bool found = false;
+  for (const auto& imp : implied[h0]) {
+    if (imp.affected_hyperedge == h1) {
+      EXPECT_NEAR(imp.probability, 0.25, 1e-6);
+      found = true;
     }
-    EXPECT_TRUE(found);
-
-    // P(D0 | D1) = 0.05 / 0.1 = 0.5
-    found = false;
-    for (const auto& imp : implied[h1]) {
-        if (imp.affected_hyperedge == h0) {
-            EXPECT_NEAR(imp.probability, 0.5, 1e-6);
-            found = true;
-        }
+  }
+  EXPECT_TRUE(found);
+
+  // P(D0 | D1) = 0.05 / 0.1 = 0.5
+  found = false;
+  for (const auto& imp : implied[h1]) {
+    if (imp.affected_hyperedge == h0) {
+      EXPECT_NEAR(imp.probability, 0.5, 1e-6);
+      found = true;
     }
-    EXPECT_TRUE(found);
+  }
+  EXPECT_TRUE(found);
 }

src/multi_pass_sinter_compat.pybind.h

@@ -126,7 +126,8 @@ void pybind_multi_pass_sinter_compat(py::module& m) {
     py::class_<MultiPassSinterCompiledDecoder>(m, "MultiPassSinterCompiledDecoder")
         .def_property_readonly("num_components", &MultiPassSinterCompiledDecoder::num_components)
         .def("decode_shots_bit_packed", &MultiPassSinterCompiledDecoder::decode_shots_bit_packed,
-             py::kw_only(), py::arg("bit_packed_detection_event_data"));
+             py::kw_only(), py::arg("bit_packed_detection_event_data"),
+             py::call_guard<py::scoped_ostream_redirect, py::scoped_estream_redirect>());
 
     py::class_<MultiPassSinterDecoder>(m, "MultiPassSinterDecoder")
         .def(py::init<size_t>(), py::arg("num_passes") = 2)

src/py/BUILD

@@ -31,6 +31,11 @@ py_library(
     data = [":copy_core_so"],
     imports = ["."],
     visibility = ["//visibility:public"],
+    deps = [
+        "@pypi//stim",
+        "@pypi//numpy",
+        "@pypi//sinter",
+    ],
 )
 
 py_library(