Skip to content

Commit

Permalink
Update albumentations/core/serialization.py
Browse files Browse the repository at this point in the history
Co-authored-by: sourcery-ai[bot] <58596630+sourcery-ai[bot]@users.noreply.github.com>
  • Loading branch information
ternaus and sourcery-ai[bot] authored Mar 9, 2024
1 parent 3a7b9f6 commit 6f4f36a
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions albumentations/core/serialization.py
Original file line number Diff line number Diff line change
Expand Up @@ -173,9 +173,7 @@ def serialize_enum(obj: Any) -> Any:
return {k: serialize_enum(v) for k, v in obj.items()}
if isinstance(obj, list):
return [serialize_enum(v) for v in obj]
if isinstance(obj, Enum):
return obj.value # Convert Enum to its value
return obj
return obj.value if isinstance(obj, Enum) else obj


def save(
Expand Down

0 comments on commit 6f4f36a

Please sign in to comment.