chore: Enable type checking in meltano.core.state_service (meltano#…

…7130)

pyproject.toml

@@ -540,7 +540,6 @@ module = [
   "meltano.core.settings_service",
   "meltano.core.settings_store",
   "meltano.core.sqlalchemy",
-  "meltano.core.state_service",
   "meltano.core.task_sets",
   "meltano.core.task_sets_service",
   "meltano.core.transform_add_service",

src/meltano/core/job/job.py

@@ -8,6 +8,7 @@
 from contextlib import asynccontextmanager, contextmanager, suppress
 from datetime import datetime, timedelta
 from enum import Enum
+from enum import IntFlag as EnumIntFlag
 
 from sqlalchemy import Column, literal, types
 from sqlalchemy.ext.hybrid import Comparator, hybrid_property
@@ -80,7 +81,7 @@ def current_trigger():
     return os.getenv("MELTANO_JOB_TRIGGER")
 
 
-class Payload(IntFlag):
+class Payload(EnumIntFlag):
     """Flag indicating whether a Job has state in its payload field."""
 
     STATE = 1

src/meltano/core/state_service.py

@@ -110,7 +110,7 @@ def validate_state(state: dict[str, Any]):
     def add_state(
         self,
         job: Job | str,
-        new_state: str | None,
+        new_state: str,
         payload_flags: Payload = Payload.STATE,
         validate=True,
     ):
@@ -157,7 +157,7 @@ def get_state(self, state_id: str):
             return json.loads(state.json_merged())
         return {}
 
-    def set_state(self, state_id: str, new_state: str | None, validate: bool = True):
+    def set_state(self, state_id: str, new_state: str, validate: bool = True):
         """Set the state for the state_id.
 
         Args: