Format

Author: fcourtial

distributed/http/scheduler/api.py

@@ -30,9 +30,7 @@ def clean_dict(d):
                 if not isinstance(d, dict):
                     return d
                 return {
-                    k: clean_dict(v)
-                    for k, v in d.items()
-                    if not isinstance(k, tuple)
+                    k: clean_dict(v) for k, v in d.items() if not isinstance(k, tuple)
                 }
 
             workers_info = clean_dict(workers_info)

distributed/http/scheduler/tests/test_scheduler_http.py

@@ -732,6 +732,7 @@ async def test_retire_workers_with_tuple_keys(c, s, a, b):
     aiohttp = pytest.importorskip("aiohttp")
 
     worker_address = "tcp://172.17.0.3:39571"
+
     async def mock_retire_workers(*args, **kwargs):
         # Return problematic data structure
         # tuple are not json serializable
@@ -740,10 +741,16 @@ async def mock_retire_workers(*args, **kwargs):
                 "type": "Worker",
                 "metrics": {
                     "digests_total_since_heartbeat": {
-                        ("execute", "slowadd", "thread-cpu", "seconds"): 0.0003396660000000093,
+                        (
+                            "execute",
+                            "slowadd",
+                            "thread-cpu",
+                            "seconds",
+                        ): 0.0003396660000000093,
                     },
-                }
-            }}
+                },
+            }
+        }
 
     # Replace the method with our mock
     s.retire_workers = mock_retire_workers