diff --git a/evaluation/evalutils.py b/evaluation/evalutils.py index fdb8fc87ee0c8c10e809ecc47fda37df3497bda6..299469ee5981e44704433df0d25da65e81728a2b 100644 --- a/evaluation/evalutils.py +++ b/evaluation/evalutils.py @@ -110,7 +110,9 @@ def savefig(name: str) -> None: def parse_directory_name(name: str) -> dict: """Extract information from the directory name structure.""" - _, game, _, model, experiment, episode, _ = str(name).split('/') + + splits = str(name).split('/') + game, _, model, experiment, episode, _ = splits[-6], splits[-5], splits[-4], splits[-3], splits[-2], splits[-1] return {'game': game, 'model': model, 'experiment': experiment,