Cleanup new exception traceback
@@ -140,12 +140,12 @@ class Comm(ABC):
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"]
