Merge 2cd2e24f57 into 11330e464b

src/compute/models/gaussian_nb.rs

@@ -59,7 +59,9 @@ impl GaussianNB {
             let bits = label.to_bits();
             groups.entry(bits).or_default().push(i);
         }
-        assert!(!groups.is_empty(), "No class labels found in y"); //-- panicked earlier
+        if groups.is_empty() {
+            panic!("No class labels found in y");
+        }
 
         // Extract and sort class labels
         self.classes = groups.keys().cloned().map(f64::from_bits).collect();
@@ -207,5 +209,6 @@ mod tests {
         let y = Matrix::from_vec(vec![0.0], 1, 1);
         let mut clf = GaussianNB::new(1e-9, false);
         clf.fit(&x, &y);
+        clf.predict(&x);
     }
 }