fix mypy

getsentry · Sep 6, 2024 · f0bf12d · f0bf12d
1 parent 9db1c55
commit f0bf12d
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/seer/automation/autofix/state.py b/src/seer/automation/autofix/state.py
@@ -11,8 +11,10 @@
 @dataclasses.dataclass
 class ContinuationState(DbState[AutofixContinuation]):
     @classmethod
-    def from_id(cls, id: int, model: type[BaseModel]) -> "ContinuationState":
-        return cast(ContinuationState, super().from_id(id, model, type=DbStateRunTypes.AUTOFIX))
+    def from_id(
+        cls, id: int, model: type[BaseModel], type: DbStateRunTypes = DbStateRunTypes.AUTOFIX
+    ) -> "ContinuationState":
+        return cast(ContinuationState, super().from_id(id, model, type))
 
     def set(self, state: AutofixContinuation):
         state.mark_updated()