def get_subgraph_with_ancestors(self, nodes): subgraph_nodes = set(nodes) for node in nodes: subgraph_nodes |= nx.ancestors(self._nx_dag, node) return self._nx_dag.subgraph(subgraph_nodes)