hotfix/catch-zerodivision-error

ersilia-os · Nov 28, 2024 · 8ee13d3 · 8ee13d3
1 parent 2a3f196
commit 8ee13d3
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/ersilia/core/tracking.py b/ersilia/core/tracking.py
@@ -329,8 +329,13 @@ def get_file_sizes(self, input_file, output_file):
         input_size = sys.getsizeof(input_file) / 1024
         output_size = sys.getsizeof(output_file) / 1024
 
-        input_avg_row_size = input_size / len(input_file)
-        output_avg_row_size = output_size / len(output_file)
+        try:
+            input_avg_row_size = input_size / len(input_file)
+            output_avg_row_size = output_size / len(output_file)
+        except ZeroDivisionError:
+            self.logger.warning("Encountered a ZeroDivisionError. No data in input or output file")
+            input_avg_row_size = -1
+            output_avg_row_size = -1
 
         return {
             "input_size": input_size,