mwtkrayer
/
suspendtools
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

numpy asks for tuples

This commit is contained in:
Michael Krayer 2021-06-03 16:41:20 +02:00
parent d21bf975fd
commit c73368781b
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
parallel.py
View File

@ -430,6 +430,7 @@ class PPP:
sl[axis] = slice( sl[axis] = slice(
self.proc_grid[key][2*axis][pos[axis]]-1, self.proc_grid[key][2*axis][pos[axis]]-1,
self.proc_grid[key][2*axis+1][pos[axis]]) self.proc_grid[key][2*axis+1][pos[axis]])
sl = tuple(sl)
# Receive data and put it in a good spot # Receive data and put it in a good spot
if rank_src==0: if rank_src==0:
integ[sl] += integ_local integ[sl] += integ_local
@ -466,6 +467,7 @@ class PPP:
for axis in range(3): for axis in range(3):
if not self.field[key_out].has_same_dim(field_ref,axis=axis): if not self.field[key_out].has_same_dim(field_ref,axis=axis):
sl[axis] = slice(0,field_ref.dim(axis=axis)) sl[axis] = slice(0,field_ref.dim(axis=axis))
sl = tuple(sl)
self.field[key_out].data = self.field[key_out].data[sl] self.field[key_out].data = self.field[key_out].data[sl]
# Copy metadata of pressure grid # Copy metadata of pressure grid
self.copy('p',key_out,skip_data=True,skip_symmetries=True) self.copy('p',key_out,skip_data=True,skip_symmetries=True)