From b892210eb8814dbcb655a916eb209975515fd369 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Wed, 1 Jul 2020 13:19:43 -0500
Subject: [PATCH] Fix type annotation of dof_array.flatten
---
meshmode/dof_array.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/meshmode/dof_array.py b/meshmode/dof_array.py
index 12ffe33f..f4099b33 100644
--- a/meshmode/dof_array.py
+++ b/meshmode/dof_array.py
@@ -21,7 +21,7 @@ THE SOFTWARE.
"""
import numpy as np
-from typing import Optional, Iterable, TYPE_CHECKING
+from typing import Optional, Iterable, TYPE_CHECKING, Any
from functools import partial
from pytools import single_valued, memoize_in
@@ -170,7 +170,7 @@ def freeze(ary: np.ndarray) -> np.ndarray:
])
-def flatten(ary: np.ndarray) -> np.ndarray:
+def flatten(ary: np.ndarray) -> Any:
r"""Convert a :class:`DOFArray` into a "flat" array of degrees of freedom,
where the resulting type of the array is given by the
:attr:`DOFArray.array_context`.
--
GitLab