Switch to StrEnum

anna-money · Nov 10, 2024 · e732d5a · e732d5a
1 parent 5ea77df
commit e732d5a
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/marshmallow_recipe/options.py b/marshmallow_recipe/options.py
@@ -8,12 +8,9 @@
 _OPTIONS_KEY = "__marshmallow_recipe_options__"
 
 
-class NoneValueHandling(str, enum.Enum):
-    IGNORE = "IGNORE"
-    INCLUDE = "INCLUDE"
-
-    def __str__(self) -> str:
-        return self.value
+class NoneValueHandling(enum.StrEnum):
+    IGNORE = enum.auto()
+    INCLUDE = enum.auto()
 
 
 @dataclasses.dataclass(kw_only=True, slots=True, frozen=True)