Fix division by zero error caused in case of empty datasets.

ComNets-Bremen · Sep 19, 2023 · 35549b9 · 35549b9
1 parent 2f76914
commit 35549b9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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):