diff --git a/mlair/helpers/helpers.py b/mlair/helpers/helpers.py index 4cc7310db32c2ef3bbdb9f70896a2f8455a974fc..981999fbc111f133cdce1b182e8147f0d9bf6807 100644 --- a/mlair/helpers/helpers.py +++ b/mlair/helpers/helpers.py @@ -96,6 +96,8 @@ def remove_items(obj: Union[List, Dict, Tuple], items: Any): return {k: v for k, v in dict_obj.items() if k not in key_list} items = to_list(items) + if len(items) == 0: + return obj if isinstance(obj, list): return remove_from_list(obj, items) elif isinstance(obj, dict):