223031012
/
DAMASK_EICMD
mirror of https://github.com/achalhp/DAMASK_EICMD.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
python/damask/_grid.py
View File

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