fix: add back return statements for closed streams (#356)

momentohq · Jul 17, 2023 · f2c9376 · f2c9376
1 parent df6c9d0
commit f2c9376
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/prepy310/topic_subscribe_async.py b/examples/prepy310/topic_subscribe_async.py
@@ -62,7 +62,7 @@ async def poll_subscription(subscription: TopicSubscribe.SubscriptionAsync):
         elif isinstance(item, TopicSubscriptionItem.Error):
             print("stream closed")
             print(item.inner_exception.message)
-
+            return
 
 if __name__ == "__main__":
     asyncio.run(main())
diff --git a/examples/py310/topic_subscribe_async.py b/examples/py310/topic_subscribe_async.py
@@ -64,7 +64,7 @@ async def poll_subscription(subscription: TopicSubscribe.SubscriptionAsync):
             case TopicSubscriptionItem.Error():
                 print("stream closed")
                 print(item.inner_exception.message)
-
+                return
 
 if __name__ == "__main__":
     asyncio.run(main())