mirror of
https://github.com/OpenHands/OpenHands.git
synced 2025-12-26 05:48:36 +08:00
8b473397d1
Co-authored-by: openhands <openhands@all-hands.dev> Co-authored-by: Xingyao Wang <xingyao@all-hands.dev> Co-authored-by: Engel Nyst <enyst@users.noreply.github.com>
21 lines
745 B
Python
21 lines
745 B
Python
def remove_fields(obj: dict | list | tuple, fields: set[str]) -> None:
|
|
"""Remove fields from an object.
|
|
|
|
Parameters:
|
|
- obj: The dictionary, or list of dictionaries to remove fields from
|
|
- fields (set[str]): A set of field names to remove from the object
|
|
"""
|
|
if isinstance(obj, dict):
|
|
for field in fields:
|
|
if field in obj:
|
|
del obj[field]
|
|
for _, value in obj.items():
|
|
remove_fields(value, fields)
|
|
elif isinstance(obj, (list, tuple)):
|
|
for item in obj:
|
|
remove_fields(item, fields)
|
|
if hasattr(obj, '__dataclass_fields__'):
|
|
raise ValueError(
|
|
'Object must not contain dataclass, consider converting to dict first'
|
|
)
|