Typing issues

Author: mpharrigan

qualtran/drawing/bloq_counts_graph.py

@@ -99,7 +99,7 @@ def get_graph(self) -> pydot.Dot:
 
     def get_svg_bytes(self) -> bytes:
         """Get the SVG code (as bytes) for drawing the graph."""
-        return self.get_graph().create(prog='dot', format='svg', encoding='utf-8')
+        return self.get_graph().create(prog='dot', format='svg', encoding='utf-8')  # type: ignore[return-value]
 
     def get_svg(self) -> IPython.display.SVG:
         """Get an IPython SVG object displaying the graph."""

qualtran/drawing/bloq_counts_graph_test.py

@@ -48,7 +48,7 @@ def test_format_counts_graph_markdown():
 
 def _get_node_labels_from_pydot_graph(drawer: _CallGraphDrawerBase) -> List[str]:
     graph = drawer.get_graph()
-    node_labels = [node.get_label() for node in graph.get_node_list()]
+    node_labels = [node.get_label() for node in graph.get_node_list()]  # type: ignore[attr-defined]
     random.shuffle(node_labels)  # don't rely on order of graphviz nodes
     return node_labels
 

qualtran/drawing/graphviz.py

@@ -340,7 +340,7 @@ def add_cxn(self, graph: pydot.Graph, cxn: Connection) -> pydot.Graph:
         graph.add_edge(self.cxn_edge(left, right, cxn))
         return graph
 
-    def get_graph(self) -> pydot.Dot:
+    def get_graph(self) -> pydot.Graph:
         """Get the graphviz graph representing the Bloq.
 
         This is the main entry-point to this class.
@@ -360,7 +360,7 @@ def get_graph(self) -> pydot.Dot:
 
     def get_svg_bytes(self) -> bytes:
         """Get the SVG code (as bytes) for drawing the graph."""
-        return self.get_graph().create(prog='dot', format='svg', encoding='utf-8')
+        return self.get_graph().create(prog='dot', format='svg', encoding='utf-8')  # type: ignore[attr-defined,return-value]
 
     def get_svg(self) -> IPython.display.SVG:
         """Get an IPython SVG object displaying the graph."""