Skip to content

Commit

Permalink
Add type hints
Browse files Browse the repository at this point in the history
Co-authored-by: Sudarsan Surendralal <[email protected]>
  • Loading branch information
pmrv and sudarsan-surendralal authored Oct 22, 2021
1 parent 4c09aec commit c0d37d9
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions pyiron_base/generic/flattenedstorage.py
Original file line number Diff line number Diff line change
Expand Up @@ -339,7 +339,7 @@ def get_array(self, name, frame=None):
else:
raise KeyError(f"no array named {name}")

def get_array_ragged(self, name):
def get_array_ragged(self, name: str) -> np.ndarray:
"""
Return elements of array `name` in all chunks. Values are returned in a ragged array of dtype=object.
Expand All @@ -357,7 +357,7 @@ def get_array_ragged(self, name):
return np.array([self.get_array(name, i) for i in range(len(self))],
dtype=object)

def get_array_filled(self, name):
def get_array_filled(self, name: str) -> np.ndarray:
"""
Return elements of array `name` in all chunks. Arrays are padded to be all of the same length.
Expand Down

0 comments on commit c0d37d9

Please sign in to comment.