No public description

PiperOrigin-RevId: 655904046
google-research · Jul 25, 2024 · c4bcd66 · c4bcd66
1 parent 0cb1825
commit c4bcd66
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vmoe/checkpoints/serialization.py b/vmoe/checkpoints/serialization.py
@@ -287,7 +287,7 @@ def _msgpack_ext_pack(x):
     return msgpack.ExtType(_MsgpackExtType.index_info, _index_info_to_bytes(x))
   if isinstance(x, (np.ndarray, jax.Array)):
     return msgpack.ExtType(_MsgpackExtType.ndarray, _ndarray_to_bytes(x))
-  if np.issctype(type(x)):
+  if isinstance(x, np.generic):
     return msgpack.ExtType(_MsgpackExtType.npscalar,
                            _ndarray_to_bytes(np.asarray(x)))
   return x