Better error for non-numeric types

danijar · Jul 2, 2024 · 0d0a1d7 · 0d0a1d7
1 parent d886c9a
commit 0d0a1d7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/zerofun/__init__.py b/zerofun/__init__.py
@@ -1,4 +1,4 @@
-__version__ = '2.1.1'
+__version__ = '2.1.2'
 
 import multiprocessing as mp
 try:

diff --git a/zerofun/sockets.py b/zerofun/sockets.py
@@ -223,6 +223,8 @@ def pack(data):
   dtypes, shapes, buffers = [], [], []
   for value in leaves:
     value = np.asarray(value)
+    if value.dtype == object:
+      raise TypeError(data)
     assert value.data.c_contiguous, (
         "Array is not contiguous in memory. Use np.asarray(arr, order='C') " +
         "before passing the data into pack().")