again

huggingface · Oct 25, 2024 · cfef85e · cfef85e
1 parent 12d5197
commit cfef85e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/datasets/arrow_dataset.py b/src/datasets/arrow_dataset.py
@@ -303,7 +303,7 @@ def _get_output_signature(
                 tf_dtype = tf.float32
                 np_dtype = np.float32
             elif np_arrays[0].dtype.kind == "U":  # Unicode strings
-                np_dtype = np.unicode_
+                np_dtype = np.str_
                 tf_dtype = tf.string
             else:
                 raise RuntimeError(