From 3a343d3167270cfcd3956b116efee674732acb1e Mon Sep 17 00:00:00 2001
From: Matthew McDermott <mattmcdermott8@gmail.com>
Date: Thu, 22 Aug 2024 13:34:11 -0400
Subject: [PATCH] Upgraded polars and pyarrow.

---
 pyproject.toml                       | 4 ++--
 tests/test_check_static_variables.py | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 8ee261a..9d6f4a5 100644
--- 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
index fbedc19..870271f 100644
--- 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."):