From eb65fa209450f10eb97baeccb3bf6cca8ec8c29f Mon Sep 17 00:00:00 2001 From: Matthew Smith <mjsmith6@illinois.edu> Date: Wed, 13 Oct 2021 13:01:31 -0500 Subject: [PATCH] fix type annotations --- grudge/reductions.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/grudge/reductions.py b/grudge/reductions.py index 2d23f706..6ec71c1d 100644 --- a/grudge/reductions.py +++ b/grudge/reductions.py @@ -92,7 +92,7 @@ def _norm(dcoll: DiscretizationCollection, vec, p, dd): raise NotImplementedError("Unsupported value of p") -def norm(dcoll: DiscretizationCollection, vec, p, dd=None) -> float: +def norm(dcoll: DiscretizationCollection, vec, p, dd=None): r"""Return the vector p-norm of a function represented by its vector of degrees of freedom *vec*. @@ -128,7 +128,7 @@ def norm(dcoll: DiscretizationCollection, vec, p, dd=None) -> float: return _norm(dcoll, vec, p, dd) -def nodal_sum(dcoll: DiscretizationCollection, dd, vec) -> float: +def nodal_sum(dcoll: DiscretizationCollection, dd, vec): r"""Return the nodal sum of a vector of degrees of freedom *vec*. :arg dd: a :class:`~grudge.dof_desc.DOFDesc`, or a value @@ -148,7 +148,7 @@ def nodal_sum(dcoll: DiscretizationCollection, dd, vec) -> float: comm.allreduce(actx.to_numpy(nodal_sum_loc(dcoll, dd, vec)), op=MPI.SUM)) -def nodal_sum_loc(dcoll: DiscretizationCollection, dd, vec) -> float: +def nodal_sum_loc(dcoll: DiscretizationCollection, dd, vec): r"""Return the rank-local nodal sum of a vector of degrees of freedom *vec*. :arg dd: a :class:`~grudge.dof_desc.DOFDesc`, or a value @@ -176,7 +176,7 @@ def nodal_sum_loc(dcoll: DiscretizationCollection, dd, vec) -> float: return sum([device_sum(grp_ary) for grp_ary in vec]) -def nodal_min(dcoll: DiscretizationCollection, dd, vec) -> float: +def nodal_min(dcoll: DiscretizationCollection, dd, vec): r"""Return the nodal minimum of a vector of degrees of freedom *vec*. :arg dd: a :class:`~grudge.dof_desc.DOFDesc`, or a value @@ -196,7 +196,7 @@ def nodal_min(dcoll: DiscretizationCollection, dd, vec) -> float: comm.allreduce(actx.to_numpy(nodal_min_loc(dcoll, dd, vec)), op=MPI.MIN)) -def nodal_min_loc(dcoll: DiscretizationCollection, dd, vec) -> float: +def nodal_min_loc(dcoll: DiscretizationCollection, dd, vec): r"""Return the rank-local nodal minimum of a vector of degrees of freedom *vec*. @@ -230,7 +230,7 @@ def nodal_min_loc(dcoll: DiscretizationCollection, dd, vec) -> float: vec, as_device_scalar(np.inf)) -def nodal_max(dcoll: DiscretizationCollection, dd, vec) -> float: +def nodal_max(dcoll: DiscretizationCollection, dd, vec): r"""Return the nodal maximum of a vector of degrees of freedom *vec*. :arg dd: a :class:`~grudge.dof_desc.DOFDesc`, or a value @@ -250,7 +250,7 @@ def nodal_max(dcoll: DiscretizationCollection, dd, vec) -> float: comm.allreduce(actx.to_numpy(nodal_max_loc(dcoll, dd, vec)), op=MPI.MAX)) -def nodal_max_loc(dcoll: DiscretizationCollection, dd, vec) -> float: +def nodal_max_loc(dcoll: DiscretizationCollection, dd, vec): r"""Return the rank-local nodal maximum of a vector of degrees of freedom *vec*. @@ -284,7 +284,7 @@ def nodal_max_loc(dcoll: DiscretizationCollection, dd, vec) -> float: vec, as_device_scalar(-np.inf)) -def integral(dcoll: DiscretizationCollection, dd, vec) -> float: +def integral(dcoll: DiscretizationCollection, dd, vec): """Numerically integrates a function represented by a :class:`~meshmode.dof_array.DOFArray` of degrees of freedom. -- GitLab