code: def __init__(self, data, expected_notifications=None):
      self.data = data
      self.expected_notifications = expected_notifications
  
file_path: distributed/distributed/diagnostics/tests/test_worker_plugin.py
imports: from __future__ import annotations
  import asyncio
  import warnings
  import pytest
  from distributed import Worker, WorkerPlugin
  from distributed.protocol.pickle import dumps
  from distributed.utils_test import async_poll_for, gen_cluster, inc
label_desc: Use image node_id 1 to create a new MyPlugin object from inherited base classes: WorkerPlugin with example: obj = MyPlugin(data, expected_notifications)