Support float32 precision for mjtNum in Python bindings.

PiperOrigin-RevId: 922959317 Change-Id: I384b3bdeabdc4d3f66c71cae41dabfaf51387be7
2026-05-28 13:56:09 -07:00
parent b612d352d4
commit df6942eb39
6 changed files with 119 additions and 104 deletions
@@ -17,6 +17,7 @@

 #include <cstddef>
 #include <iostream>
+#include <type_traits>

 #include <mujoco/mjtype.h>

@@ -30,7 +31,8 @@ namespace mujoco::python::_impl {

 static_assert(sizeof(char) == 1);
 static_assert(sizeof(int) == 4);
-static_assert(sizeof(mjtNum) == 8);
+static_assert((std::is_same_v<mjtNum, float> && sizeof(mjtNum) == 4) ||
+              (std::is_same_v<mjtNum, double> && sizeof(mjtNum) == 8));

 inline void WriteChar(std::ostream& output, char c) {
  output.write(&c, 1);