From 8fbc1f8bf346f9db143b5bc84dbf3bed4a897771 Mon Sep 17 00:00:00 2001
From: Stefan Krawczyk <stefan@dagworks.io>
Date: Wed, 28 Aug 2024 15:21:38 -0700
Subject: [PATCH] Fixes polars hist test

---
 ui/sdk/tests/tracking/test_polars_col_stats.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/ui/sdk/tests/tracking/test_polars_col_stats.py b/ui/sdk/tests/tracking/test_polars_col_stats.py
index 226669d3d..05a54d85a 100644
--- a/ui/sdk/tests/tracking/test_polars_col_stats.py
+++ b/ui/sdk/tests/tracking/test_polars_col_stats.py
@@ -66,10 +66,9 @@ def test_quantiles(example_df):
 
 def test_histogram(example_df):
     assert pcs.histogram(example_df["a"], num_hist_bins=3) == {
-        "(-inf, 0.0]": 0,
-        "(0.0, 2.0]": 2,
-        "(2.0, 4.0]": 2,
-        "(4.0, inf]": 1,
+        "(0.996, 2.333333]": 2,
+        "(2.333333, 3.666667]": 1,
+        "(3.666667, 5.0]": 2,
     }