diff --git a/setools/dta.py b/setools/dta.py index ded88ff..1448820 100644 --- a/setools/dta.py +++ b/setools/dta.py @@ -257,7 +257,8 @@ class DomainTransitionAnalysis: if self.rebuildgraph: self._build_graph() - return nx.info(self.G) + return f"Graph nodes: {nx.number_of_nodes(self.G)}\n" \ + f"Graph edges: {nx.number_of_edges(self.G)}" # # Internal functions follow diff --git a/setools/infoflow.py b/setools/infoflow.py index 4b94a0c..2afb47e 100644 --- a/setools/infoflow.py +++ b/setools/infoflow.py @@ -260,7 +260,8 @@ class InfoFlowAnalysis: if self.rebuildgraph: self._build_graph() - return nx.info(self.G) + return f"Graph nodes: {nx.number_of_nodes(self.G)}\n" \ + f"Graph edges: {nx.number_of_edges(self.G)}" # # Internal functions follow