Add support for polars 0.19 (#22)

* Add support for polars 0.19

* bump version

* fix tests for 0.19

* update readme

Author: pavelzw

.github/workflows/ci.yml

@@ -26,6 +26,7 @@ jobs:
           - { PYTHON_VERSION: 'python=3.9', POLARS_VERSION: 'polars=0.16' }
           - { PYTHON_VERSION: 'python=3.9', POLARS_VERSION: 'polars=0.17' }
           - { PYTHON_VERSION: 'python=3.9', POLARS_VERSION: 'polars=0.18' }
+          - { PYTHON_VERSION: 'python=3.9', POLARS_VERSION: 'polars=0.19' }
           - { PYTHON_VERSION: 'python=3.10', POLARS_VERSION: '' }
           - { PYTHON_VERSION: 'python=3.11', POLARS_VERSION: '' }
     steps:

README.md

@@ -170,17 +170,17 @@ To get a more detailed understanding of what's happening under the hood, check o
 ### conda
 
 ```bash
-$ conda install -c conda-forge polarify
+conda install -c conda-forge polarify
 # or micromamba
-$ micromamba install -c conda-forge polarify
+micromamba install -c conda-forge polarify
 # or pixi
-$ pixi add polarify
+pixi add polarify
 ```
 
 ### pip
 
 ```bash
-$ pip install polarify
+pip install polarify
 ```
 
 ## ⚠️ Limitations
@@ -211,8 +211,8 @@ TODO: Add some benchmarks
 
 ## 📥 Development installation
 
-```
-$ pixi install
-$ pixi run postinstall
-$ pixi run test
+```bash
+pixi install
+pixi run postinstall
+pixi run test
 ```

pixi.toml

@@ -2,7 +2,7 @@
 # https://github.com/prefix-dev/pixi/issues/79
 [project]
 name = "polarify"
-version = "0.1.2"
+version = "0.1.3"
 description = "Simplifying conditional Polars Expressions with Python 🐍 🐻‍❄️"
 authors = ["Bela Stoyan <[email protected]>", "Pavel Zwerschke <[email protected]>"]
 channels = ["conda-forge"]
@@ -16,7 +16,7 @@ lint = "pre-commit run --all"
 [dependencies]
 python = ">=3.9"
 pip = "*"
-polars = ">=0.14.24,<0.19"
+polars = ">=0.14.24,<0.20"
 # build
 hatchling = "*"
 # test

pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "hatchling.build"
 [project]
 name = "polarify"
 description = "Simplifying conditional Polars Expressions with Python 🐍 🐻‍❄️"
-version = "0.1.2"
+version = "0.1.3"
 readme = "README.md"
 license = "MIT"
 requires-python = ">=3.9"
@@ -20,7 +20,7 @@ classifiers = [
     "Programming Language :: Python :: 3.11",
 ]
 dependencies = [
-    "polars >=0.14.24,<0.19",
+    "polars >=0.14.24,<0.20",
 ]
 
 [project.urls]

tests/test_parse_body.py

@@ -50,8 +50,16 @@ def test_funcs(request):
 def test_transform_function(df: polars.DataFrame, test_funcs):
     x = polars.col("x")
     transformed_func, original_func = test_funcs
+
+    if pl_version < Version("0.19.0"):
+        df_with_transformed_func = df.select(transformed_func(x).alias("apply"))
+        df_with_applied_func = df.apply(lambda r: original_func(r[0]))
+    else:
+        df_with_transformed_func = df.select(transformed_func(x).alias("map"))
+        df_with_applied_func = df.map_rows(lambda r: original_func(r[0]))
+
     assert_frame_equal(
-        df.select(transformed_func(x).alias("apply")),
-        df.apply(lambda r: original_func(r[0])),
+        df_with_transformed_func,
+        df_with_applied_func,
         check_dtype=False,
     )