Skip to content

Commit 3c9bcb0

Browse files
committed
Fix mypy errors
1 parent 0afa0c5 commit 3c9bcb0

1 file changed

Lines changed: 3 additions & 4 deletions

File tree

src/parcels/_reprs.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -220,8 +220,7 @@ def fieldset_describe(fieldset: FieldSet) -> str:
220220
models[parent_id] = models.get(parent_id, len(models))
221221
model_id = models[parent_id]
222222

223-
type_ = field.__class__.__name__
224-
assert type_ in ("Field", "VectorField")
223+
type_ = cast(Literal["Field", "VectorField", "Context value"], field.__class__.__name__)
225224

226225
rows.append(
227226
_FieldSetDescriptionRow(
@@ -245,5 +244,5 @@ def fieldset_describe(fieldset: FieldSet) -> str:
245244

246245
def _get_parent_model(field: Field | VectorField) -> ModelData:
247246
if isinstance_noimport(field, "Field"):
248-
return field.model
249-
return field.U.model
247+
return field.model # type:ignore[union-attr]
248+
return field.U.model # type:ignore[union-attr]

0 commit comments

Comments
 (0)