Fix lint

Author: justinpolygon

polygon/websocket/__init__.py

@@ -58,7 +58,7 @@ def __init__(
             feed = feed.value
         if isinstance(market, Enum):
             market = market.value
-        self.url = f"ws{'s' if secure else ''}://{feed}/{market}"
+        self.url = f"ws{'s' if secure else ''}://{self.feed.value}/{self.market.value}"
         self.subscribed = False
         self.subs: Set[str] = set()
         self.max_reconnects = max_reconnects

polygon/websocket/models/__init__.py

@@ -50,10 +50,12 @@
 }
 
 
-def parse_single(data: Dict[str, Any], market: Market) -> Any:
+def parse_single(data: Dict[str, Any], market: Market, logger: logging.Logger) -> Any:
     event_type = data["ev"]
     # Look up the model class based on market and event type
-    model_class = MARKET_EVENT_MAP.get(market, {}).get(event_type)
+    model_class: Optional[Type[FromDictProtocol]] = MARKET_EVENT_MAP.get(
+        market, {}
+    ).get(event_type)
     if model_class:
         return model_class.from_dict(data)
     else: