infoflow/dta: Correct typing for NetworkX.

Quote the annotations so NetworkX is optional again. Signed-off-by: Chris PeBenito <[email protected]>
SELinuxProject · Apr 22, 2024 · 0a1849b · 0a1849b
1 parent 0f7a123
commit 0a1849b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/setools/dta.py b/setools/dta.py
@@ -278,7 +278,7 @@ def results(self) -> Iterable[DTAPath] | Iterable[DomainTransition]:
                 case _:
                     raise ValueError(f"Unknown analysis mode: {self.mode}")
 
-    def graphical_results(self) -> nx.DiGraph:
+    def graphical_results(self) -> "nx.DiGraph":
 
         """
         Return the results of the analysis as a NetworkX directed graph.
@@ -697,7 +697,7 @@ class Edge(mixins.NetworkXGraphEdge):
                 The default is False.
     """
 
-    G: nx.DiGraph
+    G: "nx.DiGraph"
     source: policyrep.Type
     target: policyrep.Type
     create: InitVar[bool] = False

diff --git a/setools/infoflow.py b/setools/infoflow.py
@@ -202,7 +202,7 @@ def results(self) -> Iterable[InfoFlowPath] | Iterable["InfoFlowStep"]:
                 case _:
                     raise ValueError(f"Unknown analysis mode: {self.mode}")
 
-    def graphical_results(self) -> nx.DiGraph:
+    def graphical_results(self) -> "nx.DiGraph":
 
         """
         Return the results of the analysis as a NetworkX directed graph.
@@ -432,7 +432,7 @@ class InfoFlowStep(mixins.NetworkXGraphEdge):
                 The default is False.
     """
 
-    G: nx.DiGraph
+    G: "nx.DiGraph"
     source: policyrep.Type
     target: policyrep.Type
     create: InitVar[bool] = False