Define custom StrEnum

This only exists in 3.11
torchbox · Aug 17, 2023 · 166ce2f · 166ce2f
1 parent c7489f1
commit 166ce2f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/heroku_audit/format.py b/heroku_audit/format.py
@@ -1,6 +1,6 @@
 import csv
 import json
-from enum import StrEnum
+from enum import Enum
 from io import StringIO
 from typing import Annotated, Any
 
@@ -22,7 +22,7 @@ def default(self, o: Any) -> Any:
         return super().default(o)
 
 
-class Format(StrEnum):
+class Format(str, Enum):
     TABLE = "table"
     CSV = "csv"
     JSON = "json"