fix extraction of users

rwth-acis · Feb 27, 2024 · 066478a · 066478a
1 parent 1141782
commit 066478a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/discovery/main.py b/discovery/main.py
@@ -1,4 +1,5 @@
 import pm4py
+import json
 
 def discover_petri_net(event_log,algorithm="inductive"):
     """
@@ -84,6 +85,9 @@ def bot_statistics(event_log):
     stats['numberOfStates'] = event_log['concept:name'].nunique()
     if 'user' in event_log.columns:
         stats['numberOfUsers'] = event_log['user'].nunique()
+    else:
+        users = event_log['REMAKRS'].apply(lambda x: json.loads(x)['user'])
+        stats['numberOfUsers'] = users.nunique()
     stats['averageConversationLength'] = event_log.groupby(
         'case:concept:name').size().mean()
     stats['averageConversationDuration'] = event_log.groupby(