Return None/Null instead of 0 for empty input

appian · Nov 18, 2024 · 2045914 · 2045914
1 parent 97526fb
commit 2045914
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 4 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "polars-tdigest"
-version = "0.1.4"
+version = "0.1.5"
 edition = "2021"
 
 [lib]

diff --git a/src/expressions.rs b/src/expressions.rs
@@ -264,6 +264,11 @@ fn estimate_quantile(inputs: &[Series], kwargs: MergeTDKwargs) -> PolarsResult<S
 
     let tdigests: Vec<TDigest> = tdigest_json.into_iter().map(|td| td.tdigest).collect();
     let tdigest = TDigest::merge_digests(tdigests);
-    let ans = tdigest.estimate_quantile(kwargs.quantile);
-    Ok(Series::new("", vec![ans]))
+    if tdigest.is_empty() {
+        let v: &[Option<f64>] = &[None];
+        Ok(Series::new("", v))
+    } else {
+        let ans = tdigest.estimate_quantile(kwargs.quantile);
+        Ok(Series::new("", vec![ans]))
+    }
 }