code: def __init__(self, socket, scheduler):
      self.socket = socket
      self.scheduler = scheduler
  
file_path: distributed/distributed/diagnostics/websocket.py
imports: from __future__ import annotations
  from dask.utils import key_split
  from distributed.diagnostics.plugin import SchedulerPlugin
  from distributed.diagnostics.task_stream import colors
label_desc: Use image node_id 1 to create a new WebsocketPlugin object from inherited base classes: SchedulerPlugin with example: obj = WebsocketPlugin(socket, scheduler)