make mypy happy...

This commit is contained in:
Philip Eisenlohr 2022-04-22 13:39:23 -04:00
parent 06fccc6125
commit 61265ff994
1 changed files with 3 additions and 5 deletions

View File

@ -3,7 +3,6 @@ import copy
import warnings
import multiprocessing as mp
from functools import partial
import collections
import typing
from typing import Union, Optional, TextIO, List, Sequence, Dict
from pathlib import Path
@ -150,15 +149,14 @@ class Grid:
@property
def initial_conditions(self) -> Dict[str,np.ndarray]:
"""Fields of initial conditions."""
self._ic = dict(zip(self._ic.keys(),
[v if isinstance(v, collections.Sequence) else
np.broadcast_to(v,self.cells) for v in self._ic.values()]))
self._ic = dict(zip(self._ic.keys(), # type: ignore
[v if isinstance(v,np.ndarray) else
np.broadcast_to(v,self.cells) for v in self._ic.values()])) # type: ignore
return self._ic
@initial_conditions.setter
def initial_conditions(self,
ic: Dict[str,np.ndarray]):
if not isinstance(ic,dict):
raise TypeError('initial conditions is not a dictionary')