Skip to content

Commit 3e26bb3

Browse files
committed
Fix array splitting
1 parent aa7d2c2 commit 3e26bb3

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

SimPEG/dask/electromagnetics/time_domain/simulation.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,9 @@ def get_parallel_blocks(source_list: list, m_size: int, max_chunk_size: int):
223223
for s_id, src in enumerate(source_list):
224224
for r_id, rx in enumerate(src.receiver_list):
225225
indices = np.arange(rx.nD).astype(int)
226-
chunks = np.split(indices, int(np.ceil(len(indices) / data_block_size)))
226+
chunks = np.array_split(
227+
indices, int(np.ceil(len(indices) / data_block_size))
228+
)
227229

228230
for ind, chunk in enumerate(chunks):
229231
chunk_size = len(chunk)

0 commit comments

Comments
 (0)