From dfbe171905b349fdd933424ae7af2b6217a28133 Mon Sep 17 00:00:00 2001
From: Matthew Rocklin <mrocklin@gmail.com>
Date: Tue, 17 Nov 2020 15:15:05 -0800
Subject: [PATCH] Add dask/distributed versions to performance_report (#4249)

Co-authored-by: James Bourbeau <jrbourbeau@gmail.com>
---
 distributed/scheduler.py         | 5 +++++
 distributed/tests/test_client.py | 1 +
 2 files changed, 6 insertions(+)

diff --git a/distributed/scheduler.py b/distributed/scheduler.py
index 1890f7c1c1..451d9fca64 100644
--- a/distributed/scheduler.py
+++ b/distributed/scheduler.py
@@ -5231,6 +5231,7 @@ def profile_to_figure(state):
         bandwidth_types.update()
 
         from bokeh.models import Panel, Tabs, Div
+        import distributed
 
         # HTML
         html = """
@@ -5251,6 +5252,8 @@ def profile_to_figure(state):
           <li> Workers: {nworkers} </li>
           <li> Threads: {threads} </li>
           <li> Memory: {memory} </li>
+          <li> Dask Version: {dask_version} </li>
+          <li> Dask.Distributed Version: {distributed_version} </li>
         </ul>
 
         <h2> Calling Code </h2>
@@ -5266,6 +5269,8 @@ def profile_to_figure(state):
             threads=sum(w.nthreads for w in self.workers.values()),
             memory=format_bytes(sum(w.memory_limit for w in self.workers.values())),
             code=code,
+            dask_version=dask.__version__,
+            distributed_version=distributed.__version__,
         )
         html = Div(text=html)
 
diff --git a/distributed/tests/test_client.py b/distributed/tests/test_client.py
index 100b5b1335..f34b8b4c74 100644
--- a/distributed/tests/test_client.py
+++ b/distributed/tests/test_client.py
@@ -6154,6 +6154,7 @@ async def f():
     assert "Dask Performance Report" in data
     assert "x = da.random" in data
     assert "Threads: 4" in data
+    assert dask.__version__ in data
 
 
 @pytest.mark.asyncio