Skip to content

Commit 34a680e

Browse files
committed
MAINT: Fixing mypy errors
1 parent c544363 commit 34a680e

3 files changed

Lines changed: 4 additions & 12 deletions

File tree

pyproject.toml

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -175,27 +175,19 @@ exclude = [
175175
"src/parcels/_core/utils/interpolation.py",
176176
"src/parcels/_core/utils/unstructured.py",
177177
"src/parcels/_core/particle.py",
178-
"src/parcels/_core/field.py",
179178
"src/parcels/_core/kernel.py",
180179
"src/parcels/_core/particleset.py",
181180
"src/parcels/_core/xgrid.py",
182-
"src/parcels/_core/warnings.py",
183-
"src/parcels/_core/constants.py",
184-
"src/parcels/_core/utils/sgrid.py",
185-
"src/parcels/_core/utils/string.py",
186181
"src/parcels/_core/particlefile.py",
187182
"src/parcels/_core/uxgrid.py",
188183
"src/parcels/_core/spatialhash.py",
189-
"src/parcels/_core/statuscodes.py",
190184
"src/parcels/_core/particlesetview.py",
191185
"src/parcels/_core/fieldset.py",
192186
"src/parcels/_core/index_search.py",
193187
"src/parcels/interpolators/_xinterpolators.py",
194-
"src/parcels/interpolators/__init__.py",
195188
"src/parcels/interpolators/_uxinterpolators.py",
196189
"src/parcels/kernels/_sigmagrids.py",
197190
"src/parcels/kernels/_advectiondiffusion.py",
198-
"src/parcels/kernels/__init__.py",
199191
"src/parcels/kernels/_advection.py",
200192
]
201193

src/parcels/_core/field.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ def _deal_with_errors(error, key, vector_type: VectorType):
3535
else:
3636
raise RuntimeError(f"{error}. Error could not be handled because particles was not part of the Field Sampling.")
3737

38-
if vector_type and "3D" in vector_type:
38+
if vector_type == "3D":
3939
return (0, 0, 0)
4040
elif vector_type == "2D":
4141
return (0, 0)
@@ -219,7 +219,7 @@ def __init__(
219219
_assert_same_time_interval((U, V, W))
220220

221221
self.time_interval = U.time_interval
222-
222+
self.vector_type: VectorType
223223
if self.W:
224224
self.vector_type = "3D"
225225
else:
@@ -285,7 +285,7 @@ def eval(self, time: datetime, z, y, x, particles=None):
285285
_update_particle_states_interp_value(particles, vel)
286286
_mask_outofbounds_values(grid_positions, vel)
287287

288-
if "3D" in self.vector_type:
288+
if "3D" == self.vector_type:
289289
return (u, v, w)
290290
else:
291291
return (u, v)

src/parcels/_core/statuscodes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ def __init__(self, particles, fieldset=None, msg=None):
117117
super().__init__(message)
118118

119119

120-
AllParcelsErrorCodes = {
120+
AllParcelsErrorCodes: dict[type[Exception], int] = {
121121
FieldInterpolationError: StatusCode.ErrorInterpolation,
122122
FieldOutOfBoundError: StatusCode.ErrorOutOfBounds,
123123
FieldOutOfBoundSurfaceError: StatusCode.ErrorThroughSurface,

0 commit comments

Comments
 (0)