From b03fd811b4b65b01650b15cbfff2a3aa79a2cdbb Mon Sep 17 00:00:00 2001 From: Peter Andreas Entschev Date: Wed, 26 May 2021 13:59:07 +0200 Subject: [PATCH] Fix UCX scrub config logging (#4850) --- distributed/comm/ucx.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/distributed/comm/ucx.py b/distributed/comm/ucx.py index 4d10a898004..de349d1f7e4 100644 --- a/distributed/comm/ucx.py +++ b/distributed/comm/ucx.py @@ -520,11 +520,9 @@ def _scrub_ucx_config(): options["NET_DEVICES"] = net_devices # ANY UCX options defined in config will overwrite high level dask.ucx flags - valid_ucx_keys = list(get_config().keys()) - for k, v in dask.config.get("ucx").items(): - if k in valid_ucx_keys: - options[k] = v - else: + valid_ucx_vars = list(get_config().keys()) + for k, v in options.items(): + if k not in valid_ucx_vars: logger.debug( "Key: %s with value: %s not a valid UCX configuration option" % (k, v) )