Add trivial conversion of np.int32 and np.float32 in service message …

…conversion
facebookresearch · Mar 31, 2022 · 9fe45c1 · 9fe45c1
1 parent c3cae40
commit 9fe45c1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/compiler_gym/service/proto/py_converters.py b/compiler_gym/service/proto/py_converters.py
@@ -371,7 +371,9 @@ def make_message_default_converter() -> TypeBasedConverter:
     conversion_map = {
         bool: convert_trivial,
         int: convert_trivial,
+        np.int32: convert_trivial,
         float: convert_trivial,
+        np.float32: convert_trivial,
         str: convert_trivial,
         bytes: convert_bytes_to_numpy,
         BooleanTensor: convert_tensor_message_to_numpy,