Handle string unpacking failure

rapidsai · Nov 25, 2024 · 9b54437 · 9b54437
1 parent b7d2bf1
commit 9b54437
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 3 deletions.
diff --git a/python/cudf_polars/cudf_polars/dsl/expressions/boolean.py b/python/cudf_polars/cudf_polars/dsl/expressions/boolean.py
@@ -58,7 +58,11 @@ class Name(IntEnum):
         @classmethod
         def from_polars(cls, obj: pl_expr.BooleanFunction) -> Self:
             """Convert from polars' `BooleanFunction`."""
-            function, name = str(obj).split(".", maxsplit=1)
+            try:
+                function, name = str(obj).split(".", maxsplit=1)
+            except ValueError:
+                # Failed to unpack string
+                function = None
             if function != "BooleanFunction":
                 raise ValueError("BooleanFunction required")
             return getattr(cls, name)

diff --git a/python/cudf_polars/cudf_polars/dsl/expressions/datetime.py b/python/cudf_polars/cudf_polars/dsl/expressions/datetime.py
@@ -79,7 +79,11 @@ class Name(IntEnum):
         @classmethod
         def from_polars(cls, obj: pl_expr.TemporalFunction) -> Self:
             """Convert from polars' `TemporalFunction`."""
-            function, name = str(obj).split(".", maxsplit=1)
+            try:
+                function, name = str(obj).split(".", maxsplit=1)
+            except ValueError:
+                # Failed to unpack string
+                function = None
             if function != "TemporalFunction":
                 raise ValueError("TemporalFunction required")
             return getattr(cls, name)

diff --git a/python/cudf_polars/cudf_polars/dsl/expressions/string.py b/python/cudf_polars/cudf_polars/dsl/expressions/string.py
@@ -83,7 +83,11 @@ class Name(IntEnum):
         @classmethod
         def from_polars(cls, obj: pl_expr.StringFunction) -> Self:
             """Convert from polars' `StringFunction`."""
-            function, name = str(obj).split(".", maxsplit=1)
+            try:
+                function, name = str(obj).split(".", maxsplit=1)
+            except ValueError:
+                # Failed to unpack string
+                function = None
             if function != "StringFunction":
                 raise ValueError("StringFunction required")
             return getattr(cls, name)