Merge pull request pegasystems#177 from pegasystems/pl_0_20_patch

improve polars patch PR
yusufuyanik1 · Dec 20, 2023 · b97a2c2 · b97a2c2
2 parents f69b76c + 8c3c8c2
commit b97a2c2
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/examples/articles/ADMExplained.ipynb b/examples/articles/ADMExplained.ipynb
@@ -641,12 +641,12 @@
                 "        pl.col(\"Contents\").cast(pl.Utf8)\n",
                 "    ).with_columns(\n",
                 "        pl.when(pl.col(\"Type\") == \"numeric\")\n",
-                "        .then(pl.col(\"Contents\").apply(lambda col: extract_numbers_in_contents(col, 0)))\n",
+                "        .then(pl.col(\"Contents\").map_elements(lambda col: extract_numbers_in_contents(col, 0)))\n",
                 "        .otherwise(pl.lit(-9999))\n",
                 "        .alias(\"BinLowerBound\")\n",
                 "        .cast(pl.Float32),\n",
                 "        pl.when(pl.col(\"Type\") == \"numeric\")\n",
-                "        .then(pl.col(\"Contents\").apply(lambda col: extract_numbers_in_contents(col, 1)))\n",
+                "        .then(pl.col(\"Contents\").map_elements(lambda col: extract_numbers_in_contents(col, 1)))\n",
                 "        .otherwise(pl.lit(-9999))\n",
                 "        .alias(\"BinUpperBound\")\n",
                 "        .cast(pl.Float32),\n",

diff --git a/python/pdstools/valuefinder/ValueFinder.py b/python/pdstools/valuefinder/ValueFinder.py
@@ -110,7 +110,7 @@ def __init__(
         self.StageOrder = (
             pl.DataFrame(
                 {"pyStage": self.NBADStages},
-                schema={"pyStage": pl.Categorical("physical")},
+                schema={"pyStage": pl.Enum(self.NBADStages)},
             )
             .select(pl.col("pyStage"))
             .lazy()

diff --git a/python/requirements.txt b/python/requirements.txt
@@ -3,7 +3,7 @@ plotly>=5.5.0
 requests
 pydot
 tqdm
-polars>=0.20.0, <=0.20.1
+polars>=0.20.0, <0.21.0
 pyarrow
 pyyaml
 aioboto3>=11.0