jan/gridlock
2
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

use dtype when allocating

Browse source
This commit is contained in:
Jan Petykiewicz 2021-10-31 19:02:49 -07:00
commit 85d9ff7ace
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
gridlock/grid.py
View file

@ -250,9 +250,9 @@ class Grid:
The allocated array
"""
if fill_value is None:
return numpy.empty(self.cell_data_shape)
return numpy.empty(self.cell_data_shape, dtype=dtype)
else:
return numpy.full(self.cell_data_shape, fill_value)
return numpy.full(self.cell_data_shape, fill_value, dtype=dtype)
def __init__(self,
pixel_edge_coordinates: Sequence[numpy.ndarray],