diff --git a/src/polymarket_insider_tracker/ingestor/metadata_sync.py b/src/polymarket_insider_tracker/ingestor/metadata_sync.py index 52e87cd..b85fae2 100644 --- a/src/polymarket_insider_tracker/ingestor/metadata_sync.py +++ b/src/polymarket_insider_tracker/ingestor/metadata_sync.py @@ -11,7 +11,7 @@ import logging from collections.abc import Callable from dataclasses import dataclass from datetime import UTC, datetime -from enum import Enum +from enum import StrEnum from redis.asyncio import Redis @@ -27,7 +27,7 @@ DEFAULT_CACHE_TTL_SECONDS = 600 # 10 minutes DEFAULT_REDIS_KEY_PREFIX = "polymarket:market:" -class SyncState(str, Enum): +class SyncState(StrEnum): """State of the metadata synchronizer.""" STOPPED = "stopped" diff --git a/src/polymarket_insider_tracker/pipeline.py b/src/polymarket_insider_tracker/pipeline.py index 0d4478e..9d95ca7 100644 --- a/src/polymarket_insider_tracker/pipeline.py +++ b/src/polymarket_insider_tracker/pipeline.py @@ -11,7 +11,7 @@ import contextlib import logging from dataclasses import dataclass from datetime import UTC, datetime -from enum import Enum +from enum import StrEnum from typing import TYPE_CHECKING from redis.asyncio import Redis @@ -43,7 +43,7 @@ if TYPE_CHECKING: logger = logging.getLogger(__name__) -class PipelineState(str, Enum): +class PipelineState(StrEnum): """Pipeline lifecycle states.""" STOPPED = "stopped"