test: Validate the query logic to validate metrics calculation

jyejare · jyejare · commit 3da4ddeffcb8 · 2026-05-04T21:10:05.000+05:30
Signed-off-by: Jitendra Yejare &lt;11752425+jyejare@users.noreply.github.com&gt;
diff --git a/sdk/python/feast/infra/offline_stores/dask.py b/sdk/python/feast/infra/offline_stores/dask.py
@@ -803,7 +803,7 @@ def _dask_filter_arrow_by_timestamp(
 def _dask_compute_numeric_metrics(
     column: pyarrow.ChunkedArray, histogram_bins: int
 ) -> Dict[str, Any]:
-    total = column.length
+    total = len(column)
     null_count = column.null_count
     result: Dict[str, Any] = {
         "feature_type": "numeric",
@@ -856,7 +856,7 @@ def _dask_compute_numeric_metrics(
 def _dask_compute_categorical_metrics(
     column: pyarrow.ChunkedArray, top_n: int
 ) -> Dict[str, Any]:
-    total = column.length
+    total = len(column)
     null_count = column.null_count
     result: Dict[str, Any] = {
         "feature_type": "categorical",
diff --git a/sdk/python/feast/infra/offline_stores/duckdb.py b/sdk/python/feast/infra/offline_stores/duckdb.py
@@ -262,11 +262,10 @@ def _duckdb_numeric_histogram(
         ).fetchone()[0]
         return {"bins": [min_val, max_val], "counts": [cnt], "bin_width": 0.0}
 
-    upper = max_val + (max_val - min_val) * 1e-10
     bin_width = (max_val - min_val) / bins
 
     query = (
-        f"SELECT width_bucket(CAST({q_col} AS DOUBLE), {min_val}, {upper}, {bins}) AS bucket, "
+        f"SELECT LEAST(FLOOR((CAST({q_col} AS DOUBLE) - {min_val}) / {bin_width}) + 1, {bins}) AS bucket, "
         f"COUNT(*) AS cnt "
         f"FROM {from_expr} AS _src "
         f"WHERE {q_col} IS NOT NULL AND {tw} "
diff --git a/sdk/python/tests/unit/monitoring/test_compute_correctness.py b/sdk/python/tests/unit/monitoring/test_compute_correctness.py
