From ce50d7de71e8be1644cda36ed35a9130d8f2ce91 Mon Sep 17 00:00:00 2001
From: Matthias Diener <mdiener@illinois.edu>
Date: Mon, 7 Jun 2021 23:28:31 -0500
Subject: [PATCH] mypy+flake8
---
arraycontext/impl/pytato.py | 7 ++++---
1 file changed, 4 insertions(+), 3 deletions(-)
diff --git a/arraycontext/impl/pytato.py b/arraycontext/impl/pytato.py
index 8c09226..d6d7bac 100644
--- a/arraycontext/impl/pytato.py
+++ b/arraycontext/impl/pytato.py
@@ -59,7 +59,7 @@ class _PytatoFakeNumpyNamespace(BaseFakeNumpyNamespace):
"sinh", "cosh", "tanh", "exp", "log", "log10", "isnan",
"sqrt", "exp"]
if name in pt_funcs:
- import pytato as pt
+ import pytato as pt # type: ignore
from functools import partial
return partial(rec_map_array_container, getattr(pt, name))
@@ -264,7 +264,7 @@ class PytatoArrayContext(ArrayContext):
return cl_array.get(queue=self.queue)
def call_loopy(self, program, **kwargs):
- from pytato.loopy import call_loopy
+ from pytato.loopy import call_loopy # type: ignore
import pyopencl.array as cla
entrypoint, = set(program.callables_table)
@@ -305,7 +305,8 @@ class PytatoArrayContext(ArrayContext):
# }}}
def compile(self, f: Callable[[Any], Any],
- inputs_like: Tuple[Union[Number, np.ndarray], ...]) -> Callable[..., Any]:
+ inputs_like: Tuple[Union[Number, np.ndarray], ...]) -> Callable[
+ ..., Any]:
from pytools.obj_array import flat_obj_array
from arraycontext.impl import _is_meshmode_dofarray
from meshmode.dof_array import DOFArray
--
GitLab