Fix inconsistent value types in dataframe_resolver. (#1931)

Fixes #1930 Signed-off-by: Ye Cao <caoye.cao@alibaba-inc.com>
v6d-io · Jul 1, 2024 · 5aaac91 · 5aaac91
1 parent e936885
commit 5aaac91
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/vineyard/data/dataframe.py b/python/vineyard/data/dataframe.py
@@ -509,6 +509,8 @@ def pandas_dataframe_resolver(obj, resolver):
         elif len(np_value.shape) == 1:
             values = np.expand_dims(np_value, 0).view(ndarray)
             setattr(values, '__vineyard_ref', getattr(np_value, '__vineyard_ref', None))
+            if isinstance(np_value, ndarray):
+                values = np.asarray(values)
             block = Block(values, placement, ndim=2)
         else:
             values = NDArrayArray(np.expand_dims(np_value, 0).view(ndarray), dim=2)