From 81000023d3f80ac2c997489120d1830b29bd19a9 Mon Sep 17 00:00:00 2001
From: Andrey Aristov <aaristov@pasteur.fr>
Date: Thu, 9 Jun 2022 17:34:47 +0200
Subject: [PATCH] prevent ValueError
---
src/droplet_growth/poisson.py | 7 ++++++-
1 file changed, 6 insertions(+), 1 deletion(-)
diff --git a/src/droplet_growth/poisson.py b/src/droplet_growth/poisson.py
index 17aba4f..14a53ab 100644
--- a/src/droplet_growth/poisson.py
+++ b/src/droplet_growth/poisson.py
@@ -8,7 +8,12 @@ def fit(numbers:np.ndarray, max_value=None, xlabel='Initial number of cells',
title='', plot=True, save_fig_path=None):
if max_value is None:
max_value = numbers.max()
- bins = np.arange(max_value + 1) - .5
+ try:
+ bins = np.arange(max_value + 1) - .5
+ except ValueError as e:
+ print('poisson error in bins:', *e.args)
+ return None
+
vector = bins[:-1] +.5
hist, bins= np.histogram(numbers, bins=bins, density=True)
popt, pcov = curve_fit(poisson.pmf, vector, hist, p0=(1.,))
--
GitLab