save

databricks · Jan 31, 2024 · 41fb93a · 41fb93a
1 parent adb494c
commit 41fb93a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/lilac/data/dataset_duckdb.py b/lilac/data/dataset_duckdb.py
@@ -1933,6 +1933,7 @@ def select_groups(
       ORDER BY {sort_by.value} {sort_order.value}, {value_column}
       {limit_query}
     """
+    print(query)
     df = self._query_df(query)
     counts = list(df.itertuples(index=False, name=None))
     if is_temporal(leaf.dtype):
@@ -3919,6 +3920,11 @@ def _auto_bins(stats: StatsResult, num_bins: int) -> list[Bin]:
   if stats.min_val is None or stats.max_val is None:
     return [('0', None, None)]
 
+  if stats.min_val == stats.max_val:
+    # Avoid division by zero when value_range = 0
+    const_val = cast(float, stats.min_val)
+    return [('0', const_val, None)]
+
   min_val = cast(float, stats.min_val)
   max_val = cast(float, stats.max_val)
   value_range = max_val - min_val