to_yaml -> to_yamls

dask · Mar 17, 2022 · 4b65b81 · 4b65b81
1 parent 1ecb088
commit 4b65b81
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/distributed/cluster_dump.py b/distributed/cluster_dump.py
@@ -248,7 +248,7 @@ def _compact_state(self, state: dict, expand_keys: set[str]):
         result["general"] = general
         return result
 
-    def to_yaml(
+    def to_yamls(
         self,
         root_dir: str | Path | None = None,
         worker_expand_keys: Collection[str] = ("config", "log", "logs", "tasks"),

diff --git a/distributed/tests/test_cluster_dump.py b/distributed/tests/test_cluster_dump.py
@@ -158,7 +158,7 @@ def _expected_story(task_key):
 
 
 @gen_cluster(client=True)
-async def test_cluster_dump_to_yaml(c, s, a, b, tmp_path):
+async def test_cluster_dump_to_yamls(c, s, a, b, tmp_path):
     futs = c.map(inc, range(2))
     await c.gather(futs)
 
@@ -172,7 +172,7 @@ async def test_cluster_dump_to_yaml(c, s, a, b, tmp_path):
 
     dump = DumpArtefact.from_url(f"{filename}.msgpack.gz")
     yaml_path = Path(tmp_path / "dump")
-    dump.to_yaml(yaml_path)
+    dump.to_yamls(yaml_path)
 
     scheduler_files = {
         "events.yaml",