Ignore spurious mypy typing errors

dmentipl · Mar 14, 2021 · a1f5efe · a1f5efe
1 parent 794d8af
commit a1f5efe
Show file tree

Hide file tree

Showing 4 changed files with 18 additions and 17 deletions.
diff --git a/plonk/analysis/sinks.py b/plonk/analysis/sinks.py
@@ -254,7 +254,7 @@ def _norm(x):
     return np.sqrt(x[..., 0] ** 2 + x[..., 1] ** 2 + x[..., 2] ** 2)
 
 
-@numba.njit
+@numba.njit  # type: ignore
 def _potential(position, mass):
     """Get gravitational potential on particles.
 

diff --git a/plonk/analysis/sph.py b/plonk/analysis/sph.py
@@ -7,6 +7,7 @@
 import numba
 import numpy as np
 from numba.typed import List
+from numpy import ndarray
 
 from .._logging import logger
 from .._units import Quantity
@@ -216,7 +217,7 @@ def summation(
                     else:
                         logger.info('Finding neighbours...')
                 _neighbours = snap[indices].neighbours(np.arange(len(indices)))
-                neighbours = List()
+                neighbours: List[ndarray] = List()
                 for neigh in _neighbours:
                     neighbours.append(np.array(neigh))
 
@@ -242,7 +243,7 @@ def summation(
     return result
 
 
-@numba.njit
+@numba.njit  # type: ignore
 def _compute_derivative(
     indices,
     neighbours,
@@ -293,7 +294,7 @@ def _compute_derivative(
     return result
 
 
-@numba.njit
+@numba.njit  # type: ignore
 def _compute_grad_over_neighbours(
     posi,
     hi,
@@ -319,7 +320,7 @@ def _compute_grad_over_neighbours(
     return result
 
 
-@numba.njit
+@numba.njit  # type: ignore
 def _compute_div_over_neighbours(
     posi,
     hi,
@@ -345,7 +346,7 @@ def _compute_div_over_neighbours(
     return result
 
 
-@numba.njit
+@numba.njit  # type: ignore
 def _compute_curl_over_neighbours(
     posi,
     hi,

diff --git a/plonk/utils/kernels.py b/plonk/utils/kernels.py
@@ -4,7 +4,7 @@
 import numpy as np
 
 
-@numba.njit
+@numba.njit  # type: ignore
 def kernel_cubic(q):
     """Cubic kernel function.
 
@@ -29,7 +29,7 @@ def kernel_cubic(q):
         return 0.0
 
 
-@numba.njit
+@numba.njit  # type: ignore
 def kernel_gradient_cubic(q):
     """Cubic kernel gradient function.
 
@@ -54,7 +54,7 @@ def kernel_gradient_cubic(q):
         return 0.0
 
 
-@numba.njit
+@numba.njit  # type: ignore
 def kernel_quintic(q):
     """Quintic kernel function.
 
@@ -81,7 +81,7 @@ def kernel_quintic(q):
         return 0.0
 
 
-@numba.njit
+@numba.njit  # type: ignore
 def kernel_gradient_quintic(q):
     """Quintic kernel gradient function.
 
@@ -108,7 +108,7 @@ def kernel_gradient_quintic(q):
         return 0.0
 
 
-@numba.njit
+@numba.njit  # type: ignore
 def kernel_wendland_c4(q):
     """Wendland C4 kernel function.
 
@@ -133,7 +133,7 @@ def kernel_wendland_c4(q):
         return 0.0
 
 
-@numba.njit
+@numba.njit  # type: ignore
 def kernel_gradient_wendland_c4(q):
     """Wendland C4 kernel gradient function.
 

diff --git a/plonk/visualize/splash.py b/plonk/visualize/splash.py
@@ -19,7 +19,7 @@
 IVERBOSE = -1
 
 
-@numba.njit
+@numba.njit  # type: ignore
 def w_cubic(q2: float):
     """Cubic spline kernel.
 
@@ -41,7 +41,7 @@ def w_cubic(q2: float):
     return w
 
 
-@numba.njit
+@numba.njit  # type: ignore
 def setup_integratedkernel():
     """Set up integrated kernel.
 
@@ -76,7 +76,7 @@ def setup_integratedkernel():
     return coltable
 
 
-@numba.njit
+@numba.njit  # type: ignore
 def wfromtable(q2, coltable):
     """Interpolate from integrated kernel table values to give w(q).
 
@@ -103,7 +103,7 @@ def wfromtable(q2, coltable):
     return coltable[index] + dwdx * dxx
 
 
-@numba.njit
+@numba.njit  # type: ignore
 def interpolate_projection(
     x: ndarray,
     y: ndarray,
@@ -282,7 +282,7 @@ def interpolate_projection(
     return datsmooth.T
 
 
-@numba.njit
+@numba.njit  # type: ignore
 def interpolate_slice(
     x: ndarray,
     y: ndarray,