Internal change

PiperOrigin-RevId: 429084023

pybind11_abseil/register_status_bindings.cc

@@ -146,8 +146,12 @@ void RegisterStatusBindings(module m) {
       .def("to_string", [](const absl::Status& s) { return s.ToString(); })
       .def("to_string", [](const absl::Status& s) { return s.ToString(); })
       .def("__repr__", [](const absl::Status& s) { return s.ToString(); })
       .def("__repr__", [](const absl::Status& s) { return s.ToString(); })
       .def_static("OkStatus", DoNotThrowStatus(&absl::OkStatus))
       .def_static("OkStatus", DoNotThrowStatus(&absl::OkStatus))
-      .def("raw_code", &absl::Status::code)
-      .def("CanonicalCode", &absl::Status::code)
+      .def("raw_code", [](const absl::Status& self) {
+          return static_cast<int>(self.code());
+      })
+      .def("CanonicalCode", [](const absl::Status& self) {
+          return static_cast<int>(self.code());
+      })
       .def("error_message", &absl::Status::message)
       .def("error_message", &absl::Status::message)
       .def("IgnoreError", &absl::Status::IgnoreError);
       .def("IgnoreError", &absl::Status::IgnoreError);
 
 

pybind11_abseil/tests/status_test.py

@@ -129,8 +129,8 @@ class StatusTest(absltest.TestCase):
   def test_create_ok_status(self):
   def test_create_ok_status(self):
     ok_status = status.Status.OkStatus()
     ok_status = status.Status.OkStatus()
     self.assertEqual(ok_status.to_string(), 'OK')
     self.assertEqual(ok_status.to_string(), 'OK')
-    self.assertEqual(ok_status.raw_code(), status.StatusCode.OK)
-    self.assertEqual(ok_status.CanonicalCode(), status.StatusCode.OK)
+    self.assertEqual(ok_status.raw_code(), 0)
+    self.assertEqual(ok_status.CanonicalCode(), 0)
     self.assertEqual(ok_status.error_message(), '')
     self.assertEqual(ok_status.error_message(), '')
     self.assertIsNone(ok_status.IgnoreError())
     self.assertIsNone(ok_status.IgnoreError())