Fix mypy

zxcalc · Nov 13, 2023 · cfe78b4 · cfe78b4
1 parent a0729d3
commit cfe78b4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/zxlive/mainwindow.py b/zxlive/mainwindow.py
@@ -392,6 +392,7 @@ def handle_export_tikz_proof_action(self) -> bool:
             return False
         with open(path, "w") as f:
             f.write(proof_to_tikz(self.active_panel.proof_model))
+        return True
 
     def cut_graph(self) -> None:
         assert self.active_panel is not None

diff --git a/zxlive/tikz.py b/zxlive/tikz.py
@@ -11,6 +11,8 @@ def proof_to_tikz(proof: ProofModel) -> str:
     max_width = settings.value("tikz/layout/max-width")
     draw_scalar = False
 
+    assert isinstance(vspace, float) and isinstance(hspace, float) and isinstance(max_width, float)
+
     xoffset = -max_width
     yoffset = -10
     idoffset = 0