typing

src/sentry/eventstream/kafka/backend.py

@@ -12,8 +12,9 @@
 
 from sentry import options
 from sentry.conf.types.kafka_definition import Topic
-from sentry.eventstream.base import EventStreamEventType, GroupStates
+from sentry.eventstream.base import GroupStates
 from sentry.eventstream.snuba import KW_SKIP_SEMANTIC_PARTITIONING, SnubaProtocolEventStream
+from sentry.eventstream.types import EventStreamEventType
 from sentry.killswitches import killswitch_matches_context
 from sentry.utils import json
 from sentry.utils.kafka_config import get_kafka_producer_cluster_options, get_topic_definition

src/sentry/ingest/types.py

@@ -1,4 +1,7 @@
-class ConsumerType:
+from enum import Enum
+
+
+class ConsumerType(Enum):
     """
     Defines the types of ingestion consumers
     """

tests/sentry/eventstream/test_eventstream.py

@@ -12,9 +12,9 @@
 from sentry import nodestore
 from sentry.event_manager import EventManager
 from sentry.eventstore.models import Event
-from sentry.eventstream.base import EventStreamEventType
 from sentry.eventstream.kafka.backend import KafkaEventStream
 from sentry.eventstream.snuba import SnubaEventStream, SnubaProtocolEventStream
+from sentry.eventstream.types import EventStreamEventType
 from sentry.receivers import create_default_projects
 from sentry.snuba.dataset import Dataset, EntityKey
 from sentry.testutils.cases import SnubaTestCase, TestCase