diff --git a/pyproject.toml b/pyproject.toml
index 856b2b2..74ff432 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -26,6 +26,7 @@ dependencies = [
   "xarray",
   "numpy",
   "dask",
+  "dask[diagnostics]",
   "spectral",
   "pandas",
   "matplotlib"