Raise helpful error when using the wrong plugin base classes

dask · Oct 15, 2024 · 1d2044d · 1d2044d
1 parent 9bec841
commit 1d2044d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/distributed/client.py b/distributed/client.py
@@ -5101,6 +5101,14 @@ def _register_plugin(
     ):
         if isinstance(plugin, type):
             raise TypeError("Please provide an instance of a plugin, not a type.")
+        if any(
+            "dask.distributed.diagnostics.plugin" in str(c)
+            for c in plugin.__class__.__bases__
+        ):
+            raise TypeError(
+                "Importing plugin base classes from `from dask.distributed.diagnostics.plugin` "
+                "is not supported. Please import directly from `distributed.diagnostics.plugin` instead."
+            )
         raise TypeError(
             "Registering duck-typed plugins is not allowed. Please inherit from "
             "NannyPlugin, WorkerPlugin, or SchedulerPlugin to create a plugin."