diff --git a/pyproject.toml b/pyproject.toml index e1fda5b..9dfe593 100644 --- 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 "] license = "MIT" diff --git a/readmeai/templates/header.py b/readmeai/templates/header.py index 70d8fa0..55c0ea8 100644 --- 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):