From c0dc34cc3c26a035043d6ed39ea4f497d72487ca Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Fri, 20 Aug 2021 11:45:54 -0500
Subject: [PATCH] flake8
---
pytato/transform.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/pytato/transform.py b/pytato/transform.py
index ce75526..79ba5c9 100644
--- a/pytato/transform.py
+++ b/pytato/transform.py
@@ -24,7 +24,8 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
THE SOFTWARE.
"""
-from typing import Any, Callable, Dict, FrozenSet, Union, TypeVar, Set, Generic
+from typing import (Any, Callable, Dict, FrozenSet, Union, TypeVar, Set, Generic,
+ List)
from pytato.array import (
Array, IndexLambda, Placeholder, MatrixProduct, Stack, Roll,
@@ -566,13 +567,14 @@ class TopoSortMapper(CachedWalkMapper):
def __init__(self) -> None:
super().__init__()
- self.topological_order = []
+ self.topological_order: List[Array] = []
- def post_visit(self, expr: Any) -> bool:
+ def post_visit(self, expr: Any) -> None:
self.topological_order.append(expr)
# }}}
+
# {{{ mapper frontends
def copy_dict_of_named_arrays(source_dict: DictOfNamedArrays,
--
GitLab