[#3988] Fix test deprecation warnings (#4556)

automatic commit by git-black, original commits: 5aeb088
dbt-labs · Feb 8, 2022 · 64fce5f · 64fce5f
1 parent a09aaf4
commit 64fce5f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 5 deletions.
diff --git a/core/dbt/compilation.py b/core/dbt/compilation.py
@@ -160,7 +160,7 @@ def write_graph(self, outfile: str, manifest: Manifest):
         for node_id in self.graph:
             data = manifest.expect(node_id).to_dict(omit_none=True)
             out_graph.add_node(node_id, **data)
-        with open(outfile, 'wb') as outfh:
+        with open(outfile, "wb") as outfh:
             pickle.dump(out_graph, outfh, protocol=pickle.HIGHEST_PROTOCOL)
 
 

diff --git a/core/dbt/graph/graph.py b/core/dbt/graph/graph.py
@@ -35,10 +35,7 @@ def ancestors(
         # This used to use nx.utils.reversed(self.graph), but that is deprecated,
         # so changing to use self.graph.reverse(copy=False) as recommeneded
         G = self.graph.reverse(copy=False) if self.graph.is_directed() else self.graph
-        anc = nx.single_source_shortest_path_length(G=G,
-                                                    source=node,
-                                                    cutoff=max_depth)\
-            .keys()
+        anc = nx.single_source_shortest_path_length(G=G, source=node, cutoff=max_depth).keys()
         return anc - {node}
 
     def descendants(