diff --git a/giskard_vision/object_detection/dataloaders/loaders.py b/giskard_vision/object_detection/dataloaders/loaders.py index d34bbdb4..95fc343a 100644 --- a/giskard_vision/object_detection/dataloaders/loaders.py +++ b/giskard_vision/object_detection/dataloaders/loaders.py @@ -324,9 +324,15 @@ def get_meta(self, idx: int) -> Optional[Dict[str, Any]]: "faceRectangle_height", ] return MetaData( - data=meta.data + {k: v for k, v in meta.data.items() if k not in excludes}, + data={ + **meta.data, + **{k: v for k, v in meta.data.items() if k not in excludes}, + }, categories=meta.categories + [c for c in meta.categories if c not in excludes], - issue_groups=meta.issue_groups + {k: v for k, v in meta.issue_groups.items() if k not in excludes}, + issue_groups={ + **meta.issue_groups, + **{k: v for k, v in meta.issue_groups.items() if k not in excludes}, + }, )