MNT update to polars 0.17.2 (#73)

lorentzenchr · Apr 14, 2023 · e6a33b4 · e6a33b4
1 parent 2dbace0
commit e6a33b4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -21,7 +21,7 @@ classifiers = [
 ]
 keywords = ["machine learning", "model diagnostics", "calibration"]
 dependencies = [
-  "polars >= 0.16.8",
+  "polars >= 0.17.2",
   "matplotlib >= 3.6.1",
   "scikit-learn >=1.2.0",  # We rely on the set_output API.
 ]

diff --git a/src/model_diagnostics/calibration/tests/test_identification.py b/src/model_diagnostics/calibration/tests/test_identification.py
@@ -407,7 +407,9 @@ def test_compute_bias_keeps_null_values():
         feature=feature,
         n_bins=1,
     )
-    assert_series_equal(df_bias["feature"], pl.Series("feature", [None]))
+    assert_series_equal(
+        df_bias["feature"], pl.Series("feature", [None], dtype=pl.Float64)
+    )
     assert_series_equal(
         df_bias["bias_count"], pl.Series("bias_count", [2], dtype=pl.UInt32)
     )