Issue with LogReader Class During XES File Processing

[Leon-AW]

Dear authors,

We at Humboldt University are trying to replicate your project but are encountering an issue during execution. When running the following command:

Command: "python dg_training.py -f Helpdesk.xes -m lstm -e 1 -o bayesian"

Error:
(deep_generator) (base) leon@DESKTOP-6US0IG2:~/uni/process_prediction/GenerativeLSTM$ python dg_training.py -f Helpdesk.xes -m lstm -e 1 -o bayesian
WARNING: All log messages before absl::InitializeLog() is called are written to STDERR
E0000 00:00:1733769813.573576 13020 cuda_dnn.cc:8310] Unable to register cuDNN factory: Attempting to register factory for plugin cuDNN when one has already been registered
E0000 00:00:1733769813.590714 13020 cuda_blas.cc:1418] Unable to register cuBLAS factory: Attempting to register factory for plugin cuBLAS when one has already been registered
/home/leon/miniconda3/envs/deep_generator/lib/python3.9/site-packages/pm4py/util/dt_parsing/parser.py:82: UserWarning: ISO8601 strings are not fully supported with strpfromiso for Python versions below 3.11
warnings.warn(
parsing log, completed traces :: 100%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 4580/4580 [00:00<00:00, 4691.03it/s]
Rearranging log traces ... Traceback (most recent call last):
File "/home/leon/uni/process_prediction/GenerativeLSTM/dg_training.py", line 88, in
main(sys.argv[1:])
File "/home/leon/uni/process_prediction/GenerativeLSTM/dg_training.py", line 84, in main
tr.ModelTrainer(parameters)
File "/home/leon/uni/process_prediction/GenerativeLSTM/model_training/model_trainer.py", line 31, in init
self.log = self.load_log(params)
File "/home/leon/uni/process_prediction/GenerativeLSTM/model_training/model_trainer.py", line 97, in load_log
log = lr.LogReader(os.path.join('input_files', params['file_name']),
File "/home/leon/miniconda3/envs/deep_generator/lib/python3.9/site-packages/readers/log_reader.py", line 35, in init
self.load_data_from_file()
File "/home/leon/miniconda3/envs/deep_generator/lib/python3.9/site-packages/readers/log_reader.py", line 44, in load_data_from_file
self.get_xes_events_data()
File "/home/leon/miniconda3/envs/deep_generator/lib/python3.9/site-packages/readers/log_reader.py", line 81, in get_xes_events_data
self.append_csv_start_end()
File "/home/leon/miniconda3/envs/deep_generator/lib/python3.9/site-packages/readers/log_reader.py", line 198, in append_csv_start_end
for case, group in pd.DataFrame(self.data).groupby('caseid'):
File "/home/leon/miniconda3/envs/deep_generator/lib/python3.9/site-packages/pandas/core/frame.py", line 8402, in groupby
return DataFrameGroupBy(
File "/home/leon/miniconda3/envs/deep_generator/lib/python3.9/site-packages/pandas/core/groupby/groupby.py", line 965, in init
grouper, exclusions, obj = get_grouper(
File "/home/leon/miniconda3/envs/deep_generator/lib/python3.9/site-packages/pandas/core/groupby/grouper.py", line 888, in get_grouper
raise KeyError(gpr)
KeyError: 'caseid'

Tried to fix it with Ai tools but they also don't know how to fix the problem. Could you please help us? My setup is Linux so on debian I ran it with WSL2 and Python 3.9.
Thank you very much in advance.

SIncerly Leon