chore: Fix errors from new mypy version

open-contracting · Dec 6, 2023 · 4e5d355 · 4e5d355
1 parent f9bfcd3
commit 4e5d355
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/yapw/types.py b/yapw/types.py
@@ -69,7 +69,10 @@ class PublishKeywords(TypedDict, total=False):
     properties: pika.BasicProperties
 
 
-ConsumerCallback = Callable[[State, pika.channel.Channel, pika.spec.Basic.Deliver, pika.BasicProperties, Any], None]
+ConsumerCallback = Callable[
+    [State[Any], pika.channel.Channel, pika.spec.Basic.Deliver, pika.BasicProperties, Any], None
+]
 Decorator = Callable[
-    [Decode, ConsumerCallback, State, pika.channel.Channel, pika.spec.Basic.Deliver, pika.BasicProperties, bytes], None
+    [Decode, ConsumerCallback, State[Any], pika.channel.Channel, pika.spec.Basic.Deliver, pika.BasicProperties, bytes],
+    None,
 ]