fix mypy errors

fschlatt · Nov 16, 2021 · da91f5b · da91f5b
1 parent 6ddbcc6
commit da91f5b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/clubs/poker/evaluator.py b/clubs/poker/evaluator.py
@@ -3,7 +3,7 @@
 import itertools
 import operator
 from timeit import default_timer as timer
-from typing import Dict, Iterator, List, Optional, Tuple
+from typing import Dict, Iterable, Iterator, List, Optional, Tuple
 
 from clubs import error
 
@@ -576,9 +576,9 @@ def add_to_dict(
             # if different multiples (e.g. full house) order of
             # multiples is important
             if len(set(multiples)) > 1:
-                multiple_combinations = itertools.permutations(
-                    backwards_ranks, len(multiples)
-                )
+                multiple_combinations: Iterable[
+                    Tuple[int, ...]
+                ] = itertools.permutations(backwards_ranks, len(multiples))
             # if same multiples (e.g. two pair) order of multiples
             # is unimportant
             else:

diff --git a/clubs/render/graphic.py b/clubs/render/graphic.py
@@ -107,6 +107,7 @@ def _run_flask(self) -> None:
         monkey.patch_all()
         import flask
         import flask_socketio
+        import markupsafe
 
         config: Dict[str, Any] = {}
         dir_path = os.path.dirname(os.path.realpath(__file__))
@@ -124,7 +125,7 @@ def connect():  # type: ignore
         @app.route("/")
         def index():  # type: ignore
             svg = str(self.svg_poker.base_svg)
-            return flask.render_template("index.html", svg=flask.Markup(svg))
+            return flask.render_template("index.html", svg=markupsafe.Markup(svg))
 
         def listener() -> None:
             nonlocal config