Use StrEnum for python 3.11 compatibility (Fixes #16)

.github/workflows/test.yml

@@ -13,7 +13,7 @@ jobs:
 
     strategy:
       matrix:
-        python-version: [ "3.8", "3.9", "3.10" ]
+        python-version: [ "3.8", "3.9", "3.10", "3.11" ]
       fail-fast: false
 
     steps:

arcee/dalm.py

@@ -1,7 +1,7 @@
-from enum import Enum
 from typing import Any, Dict, List, Literal, Optional
 
 from pydantic import BaseModel, model_validator
+from strenum import StrEnum
 
 from arcee.api_handler import make_request
 from arcee.schemas.routes import Route
@@ -12,7 +12,7 @@ def check_model_status(name: str) -> Dict[str, str]:
     return make_request("get", route)
 
 
-class FilterType(str, Enum):
+class FilterType(StrEnum):
     fuzzy_search = "fuzzy_search"
     strict_search = "strict_search"
 

arcee/schemas/routes.py

@@ -1,7 +1,7 @@
-from enum import Enum
+from strenum import StrEnum
 
 
-class Route(str, Enum):
+class Route(StrEnum):
     contexts = "contexts"
     train_model = "models/train"
     train_model_status = "models/status/{id_or_name}"

pyproject.toml

@@ -17,7 +17,8 @@ dependencies = [
     "requests",
     "typer",
     "rich",
-    "pydantic"
+    "pydantic",
+    "StrEnum"
 ]
 
 [project.scripts]

tasks.py

@@ -189,9 +189,9 @@ def _bump_version(version: str, bump: Optional[BumpType] = None) -> str:
     from packaging.version import Version
 
     v = Version(version)
-    if bump == BumpType.MAJOR:
+    if bump == BumpType.MAJOR.value:
         v = Version(f"{v.major + 1}.0.0")
-    elif bump == BumpType.MINOR:
+    elif bump == BumpType.MINOR.value:
         v = Version(f"{v.major}.{v.minor + 1}.0")
     else:
         v = Version(f"{v.major}.{v.minor}.{v.micro + 1}")