fix: Fix Int128 dtype serialization (#20629)

pola-rs · Jan 9, 2025 · 323b88c · 323b88c
1 parent f3da50f
commit 323b88c
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/crates/polars-core/src/datatypes/_serde.rs b/crates/polars-core/src/datatypes/_serde.rs
@@ -85,6 +85,7 @@ enum SerializableDataType {
     Int16,
     Int32,
     Int64,
+    Int128,
     Float32,
     Float64,
     String,
@@ -130,6 +131,7 @@ impl From<&DataType> for SerializableDataType {
             Int16 => Self::Int16,
             Int32 => Self::Int32,
             Int64 => Self::Int64,
+            Int128 => Self::Int128,
             Float32 => Self::Float32,
             Float64 => Self::Float64,
             String => Self::String,
@@ -186,6 +188,7 @@ impl From<SerializableDataType> for DataType {
             Int16 => Self::Int16,
             Int32 => Self::Int32,
             Int64 => Self::Int64,
+            Int128 => Self::Int128,
             Float32 => Self::Float32,
             Float64 => Self::Float64,
             String => Self::String,

diff --git a/py-polars/tests/unit/io/test_pickle.py b/py-polars/tests/unit/io/test_pickle.py
@@ -19,7 +19,12 @@ def test_pickle() -> None:
 
 
 def test_pickle_expr() -> None:
-    for e in [pl.all(), pl.len(), pl.duration(weeks=10, days=20, hours=3)]:
+    for e in [
+        pl.all(),
+        pl.len(),
+        pl.duration(weeks=10, days=20, hours=3),
+        pl.col("a").cast(pl.Int128),
+    ]:
         f = io.BytesIO()
         pickle.dump(e, f)