fix small bug

Jabbawukis · Jan 2, 2025 · 8724694 · 8724694
1 parent ef31f00
commit 8724694
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/src/utility/utility.py b/src/utility/utility.py
@@ -153,12 +153,15 @@ def count_increasing_occurrences_in_slices(path_to_files: str) -> dict:
                             "occurrences_increase": [],
                             "obj_id": fact["obj_id"],
                         }
-                    increase = sum(
-                        count["occurrences"]
-                        for count in increasing_occurrences_in_slices[relation_id][entity_id]["occurrences_increase"]
-                    )
+                    try:
+                        total = increasing_occurrences_in_slices[relation_id][entity_id]["occurrences_increase"][-1][
+                            "total"
+                        ]
+                        total = total + fact["occurrences"]
+                    except IndexError:
+                        total = fact["occurrences"]
                     increasing_occurrences_in_slices[relation_id][entity_id]["occurrences_increase"].append(
-                        {"Slice": files.index(file), "occurrences": fact["occurrences"], "total": increase}
+                        {"Slice": files.index(file), "occurrences": fact["occurrences"], "total": total}
                     )
     return increasing_occurrences_in_slices