evo-objects 0.4.2__tar.gz → 0.4.3__tar.gz

{evo_objects-0.4.2 → evo_objects-0.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: evo-objects
-Version: 0.4.2
+Version: 0.4.3
 Summary: Python SDK for using the Seequent Evo Geoscience Object API
 Project-URL: Source, https://github.com/SeequentEvo/evo-python-sdk
 Project-URL: Tracker, https://github.com/SeequentEvo/evo-python-sdk/issues

{evo_objects-0.4.2 → evo_objects-0.4.3}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "evo-objects"
 description = "Python SDK for using the Seequent Evo Geoscience Object API"
-version = "0.4.2"
+version = "0.4.3"
 requires-python = ">=3.10"
 license-files = ["LICENSE.md"]
 dynamic = ["readme"]

{evo_objects-0.4.2 → evo_objects-0.4.3}/src/evo/objects/utils/data.py

@@ -73,6 +73,7 @@ def _get_schema_from_dataframe(dataframe: "pd.DataFrame") -> pa.Schema:
     This hook centralizes dataframe-to-Arrow schema adjustments before the
     dataframe is converted to a table. It currently performs these conversions:
     - large_string -> string
+    - dictionary<*, large_string> -> dictionary<*, string>
 
     :param dataframe: The pandas dataframe to get the schema from.
     :return: A pyarrow schema with any configured type conversions applied.
@@ -82,6 +83,8 @@ def _get_schema_from_dataframe(dataframe: "pd.DataFrame") -> pa.Schema:
     for field in schema:
         if pa.types.is_large_string(field.type):
             fields.append(field.with_type(pa.string()))
+        elif pa.types.is_dictionary(field.type) and pa.types.is_large_string(field.type.value_type):
+            fields.append(field.with_type(pa.dictionary(field.type.index_type, pa.string())))
         else:
             fields.append(field)
     return pa.schema(fields)