fix: allow upload string list (#47)

zeno-ml · Dec 8, 2023 · 9eae224 · 9eae224
1 parent 36c69fc
commit 9eae224
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zeno-client"
-version = "0.1.13"
+version = "0.1.14"
 description = "Python client for creating new Zeno projects and uploading data."
 authors = ["Zeno Team <[email protected]>"]
 license = "MIT"

diff --git a/zeno_client/util.py b/zeno_client/util.py
@@ -33,17 +33,17 @@ def df_to_pa(df: pd.DataFrame, id_column: str) -> pa.Table:
 
         if pa.types.is_struct(col_type):
             new_column = array_to_utf8_json_array(data)
+        elif pa.types.is_list(col_type):
+            if pa.types.is_integer(col_type.value_type) or pa.types.is_floating(
+                col_type.value_type
+            ):
+                new_column = data
+            else:
+                new_column = array_to_utf8_json_array(data)
         elif (
             pa.types.is_integer(col_type)
             or pa.types.is_floating(col_type)
             or pa.types.is_boolean(col_type)
-            or (
-                pa.types.is_list(col_type)
-                and (
-                    pa.types.is_integer(col_type.value_type)
-                    or pa.types.is_floating(col_type.value_type)
-                )
-            )
         ):
             new_column = data
         else: