From 35549b90eb4b306576305990d12d7202b497c393 Mon Sep 17 00:00:00 2001 From: Jens Dede Date: Tue, 19 Sep 2023 22:00:37 +0200 Subject: [PATCH] Fix division by zero error caused in case of empty datasets. --- simplelabel/models.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/simplelabel/models.py b/simplelabel/models.py index ff9be9e..365d2ac 100644 --- a/simplelabel/models.py +++ b/simplelabel/models.py @@ -54,7 +54,9 @@ def get_number_polls(self): get_number_polls.short_description = "Number of polls" def get_percentage_done(self): - return round((self.get_number_polls() / (self.dataset_max_polls * self.get_number_images()))*100.0) + if self.dataset_max_polls > 0 and self.get_number_images() > 0: + return round((self.get_number_polls() / (self.dataset_max_polls * self.get_number_images()))*100.0) + return 0.0 get_percentage_done.short_description = "Percentage done" def __str__(self):