fix(lixuelin): to_ndarray fails to assign dtype for scalars (#708)

opendilab · Aug 18, 2023 · bc3ecd9 · bc3ecd9
1 parent aec6e9c
commit bc3ecd9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ding/torch_utils/data_helper.py b/ding/torch_utils/data_helper.py
@@ -218,7 +218,10 @@ def transform(d):
     elif isinstance(item, bool) or isinstance(item, str):
         return item
     elif np.isscalar(item):
-        return np.array(item)
+        if dtype is None:
+            return np.array(item)
+        else:
+            return np.array(item, dtype=dtype)
     elif item is None:
         return None
     else: