fix(pm4py): significant performance improvement filter_ocel_objects

process-intelligence-solutions · Jul 15, 2022 · ef575a8 · ef575a8
1 parent f14ab86
commit ef575a8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pm4py/filtering.py b/pm4py/filtering.py
@@ -1017,7 +1017,8 @@ def filter_ocel_objects(ocel: OCEL, object_identifiers: Collection[str], positiv
         while level > 1:
             curr = list(object_identifiers)
             for el in curr:
-                object_identifiers = object_identifiers.union(graph[el])
+                for el2 in graph[el]:
+                    object_identifiers.add(el2)
             level = level - 1
     from copy import copy
     from pm4py.objects.ocel.util import filtering_utils