fix: improve types according to new typecheck

This commit is contained in:
Akash Mondal
2025-06-09 19:10:15 +00:00
parent 47ae20d29c
commit 13d730ae5c
2 changed files with 7 additions and 9 deletions

View File

@ -193,7 +193,7 @@ async def run_enhanced_demo(
print("🖥️ ENHANCED ECHO SERVER MODE")
print("=" * 60)
# Set the enhanced stream handler
# type: ignore: Stream is type of NetStream
host.set_stream_handler(PROTOCOL_ID, enhanced_echo_handler)
print(
@ -217,7 +217,8 @@ async def run_enhanced_demo(
# Create stream and run enhanced demo
stream = await host.new_stream(info.peer_id, [PROTOCOL_ID])
await enhanced_client_demo(stream)
if isinstance(stream, NetStream):
await enhanced_client_demo(stream)
print("\n" + "=" * 60)
print("CLIENT DEMO COMPLETE")