From 9a23609b480896462125714c950425492d2bd243 Mon Sep 17 00:00:00 2001
From: Alex Stokes <r.alex.stokes@gmail.com>
Date: Wed, 6 Nov 2019 15:21:30 -0800
Subject: [PATCH] type protocol IDs to satisfy mypy

---
 libp2p/host/ping.py                  | 3 ++-
 libp2p/identity/identify/protocol.py | 4 ++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/libp2p/host/ping.py b/libp2p/host/ping.py
index 46a28e06..9075f0bb 100644
--- a/libp2p/host/ping.py
+++ b/libp2p/host/ping.py
@@ -4,8 +4,9 @@ import logging
 from libp2p.network.stream.exceptions import StreamEOF, StreamReset
 from libp2p.network.stream.net_stream_interface import INetStream
 from libp2p.peer.id import ID as PeerID
+from libp2p.typing import TProtocol
 
-ID = "/ipfs/ping/1.0.0"
+ID = TProtocol("/ipfs/ping/1.0.0")
 PING_LENGTH = 32
 RESP_TIMEOUT = 60
 
diff --git a/libp2p/identity/identify/protocol.py b/libp2p/identity/identify/protocol.py
index b42e51a9..390c0de2 100644
--- a/libp2p/identity/identify/protocol.py
+++ b/libp2p/identity/identify/protocol.py
@@ -4,11 +4,11 @@ from multiaddr import Multiaddr
 
 from libp2p.host.host_interface import IHost
 from libp2p.network.stream.net_stream_interface import INetStream
-from libp2p.typing import StreamHandlerFn
+from libp2p.typing import StreamHandlerFn, TProtocol
 
 from .pb.identify_pb2 import Identify
 
-ID = "/ipfs/id/1.0.0"
+ID = TProtocol("/ipfs/id/1.0.0")
 PROTOCOL_VERSION = "ipfs/0.1.0"
 # TODO dynamically generate the agent version
 AGENT_VERSION = "py-libp2p/alpha"