consistent and correct type hinting/documentation for comments

This commit is contained in:
Philip Eisenlohr 2022-05-19 12:19:55 -04:00
parent 164c7b4bc3
commit 92a4883344
3 changed files with 7 additions and 6 deletions

View File

@ -50,15 +50,16 @@ class Grid:
Coordinates of grid origin in meter. Defaults to [0.0,0.0,0.0].
initial_conditions : dictionary, optional
Labels and values of the inital conditions at each material point.
comments : (list of) str, optional
Comments, e.g. history of operations.
comments : str or iterable of str, optional
Additional, human-readable information, e.g. history of operations.
"""
self.material = material
self.size = size # type: ignore
self.origin = origin # type: ignore
self.initial_conditions = {} if initial_conditions is None else initial_conditions
self.comments = [] if comments is None else comments # type: ignore
comments_ = [comments] if isinstance(comments,str) else comments
self.comments = [] if comments_ is None else [str(c) for c in comments_]
def __repr__(self) -> str:
"""Give short human-readable summary."""

View File

@ -14,7 +14,7 @@ class Table:
def __init__(self,
shapes: dict,
data: np.ndarray,
comments: Union[str, Iterable] = None):
comments: Union[str, Iterable[str]] = None):
"""
New spreadsheet.
@ -30,7 +30,7 @@ class Table:
"""
comments_ = [comments] if isinstance(comments,str) else comments
self.comments = [] if comments_ is None else [c for c in comments_]
self.comments = [] if comments_ is None else [str(c) for c in comments_]
self.shapes = { k:(v,) if isinstance(v,(np.int64,np.int32,int)) else v for k,v in shapes.items() }
self.data = pd.DataFrame(data=data)
self._relabel('uniform')

View File

@ -103,7 +103,7 @@ class VTK:
Parameters
----------
comments : str or list of str
comments : str or sequence of str
Comments.
"""