Cleanup new exception traceback (#4125)

dask · Sep 28, 2020 · a968158 · a968158
1 parent 4dd0e1b
commit a968158
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/distributed/comm/core.py b/distributed/comm/core.py
@@ -140,12 +140,12 @@ def handshake_configuration(local, remote):
                     local["pickle-protocol"], remote["pickle-protocol"]
                 )
             }
-        except KeyError:
+        except KeyError as e:
             raise ValueError(
                 "Your Dask versions may not be in sync. "
                 "Please ensure that you have the same version of dask "
                 "and distributed on your client, scheduler, and worker machines"
-            )
+            ) from e
 
         if local["compression"] == remote["compression"]:
             out["compression"] = local["compression"]