relax polars upper pin

Author: tswast

setup.py

@@ -77,9 +77,7 @@
         "google-cloud-pubsub >=2.21.4",
     ],
     # used for local engine
-    # TODO(tswast): relax upper pin when issue with test_engines_astype_int
-    # and test_divmods_series is resolved.
-    "polars": ["polars >= 1.21.0, <1.34.0"],
+    "polars": ["polars >= 1.21.0"],
     "scikit-learn": ["scikit-learn>=1.2.2"],
     # Packages required for basic development flow.
     "dev": [

tests/system/small/engines/test_generic_ops.py

@@ -22,7 +22,7 @@
 from bigframes.session import polars_executor
 from bigframes.testing.engine_utils import assert_equivalence_execution
 
-pytest.importorskip("polars")
+polars = pytest.importorskip("polars")
 
 # Polars used as reference as its fast and local. Generally though, prefer gbq engine where they disagree.
 REFERENCE_ENGINE = polars_executor.PolarsExecutor()
@@ -54,6 +54,12 @@ def apply_op(
 
 @pytest.mark.parametrize("engine", ["polars", "bq", "bq-sqlglot"], indirect=True)
 def test_engines_astype_int(scalars_array_value: array_value.ArrayValue, engine):
+    polars_version = tuple([int(part) for part in polars.__version__.split(".")])
+    if polars_version >= (1, 34, 0):
+        # TODO(https://github.com/pola-rs/polars/issues/24841): Remove this when
+        # polars fixes Decimal to Int cast.
+        scalars_array_value = scalars_array_value.drop_columns(["numeric_col"])
+
     arr = apply_op(
         scalars_array_value,
         ops.AsTypeOp(to_type=bigframes.dtypes.INT_DTYPE),