Fix py311 StrEnum incompatibility.

eli64s · Aug 31, 2024 · d9ae89f · d9ae89f
1 parent 708e7ae
commit d9ae89f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "readmeai"
-version = "0.5.086"
+version = "0.5.087"
 description = "README file generator, powered by large language model APIs 👾"
 authors = ["Eli <egsalamie@gmail.com>"]
 license = "MIT"

diff --git a/readmeai/templates/header.py b/readmeai/templates/header.py
@@ -1,17 +1,17 @@
-import enum
+from enum import Enum
 from typing import Any, ClassVar
 
 from .base_template import BaseTemplate
 
 
-class HeaderStyle(enum.StrEnum):
+class HeaderStyle(Enum):
     """
     Enum containing header styles for the README file.
     """
 
-    CLASSIC = enum.auto()
-    COMPACT = enum.auto()
-    MODERN = enum.auto()
+    CLASSIC = "classic"
+    COMPACT = "compact"
+    MODERN = "modern"
 
 
 class HeaderTemplate(BaseTemplate):