fix: Default arrow conversion to string type (#122)

equinor · Dec 16, 2024 · 54004b1 · 54004b1
1 parent e7d6fab
commit 54004b1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/fmu/sumo/sim2sumo/_special_treatments.py b/src/fmu/sumo/sim2sumo/_special_treatments.py
@@ -26,11 +26,11 @@ def convert_to_arrow(frame):
         frame["DATE"] = pd.to_datetime(frame["DATE"])
     scheme = []
     for column_name in frame.columns:
-        if pd.api.types.is_string_dtype(frame[column_name]):
-            scheme.append((column_name, pa.string()))
+        if pd.api.types.is_numeric_dtype(frame[column_name]):
+            scheme.append((column_name, pa.float32()))
         else:
             scheme.append(
-                (column_name, standard.get(column_name, pa.float32()))
+                (column_name, standard.get(column_name, pa.string()))
             )
     table = pa.Table.from_pandas(frame, schema=pa.schema(scheme))
     return table