Merge pull request #111 from justin13601/upgrade_polars

Upgraded polars and pyarrow.
justin13601 · Aug 22, 2024 · 94b121f · 94b121f
2 parents 7c2c5cf + 3a343d3
commit 94b121f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,14 +18,14 @@ classifiers = [
     "Operating System :: OS Independent",
 ]
 dependencies = [
-    "polars == 0.20.*",
+    "polars == 1.5.*",
     "bigtree == 0.18.*",
     "ruamel.yaml == 0.18.*",
     "loguru == 0.7.*",
     "hydra-core == 1.3.*",
     "pytimeparse == 1.1.*",
     "networkx == 3.3.*",
-    "pyarrow == 16.1.*",
+    "pyarrow == 17.*",
     "meds == 0.3",
 ]
 

diff --git a/tests/test_check_static_variables.py b/tests/test_check_static_variables.py
@@ -44,7 +44,7 @@ def test_check_static_variables():
             "is_C": [1, 1, 1, 0],
         }
     )
-    assert filtered_df.frame_equal(expected_df)
+    assert filtered_df.equals(expected_df)
 
     # Test ValueError when demographic column is missing
     with pytest.raises(ValueError, match="Static predicate 'female' not found in the predicates dataframe."):