Skip to content

Commit 0d03647

Browse files
committed
Fix possible issue with empty block
1 parent 881756a commit 0d03647

1 file changed

Lines changed: 8 additions & 6 deletions

File tree

SimPEG/dask/electromagnetics/time_domain/simulation.py

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -207,13 +207,15 @@ def compute_J(self, f=None, Ainv=None):
207207
# )
208208
# else:
209209
# ATinv_df_duT_v = AdiagTinv * np.asarray(field_derivatives[isrc][:, block_ind])
210+
211+
if batch_block:
212+
f_blocks, j_blocks = process_blocks(
213+
self, AdiagTinv, d_count, batch_block, batch_indices, Asubdiag, f, tInd,
214+
solution_type, Jmatrix
215+
)
216+
field_deriv_blocks.append(dask.array.hstack(f_blocks))
217+
j_row_blocks += j_blocks
210218

211-
f_blocks, j_blocks = process_blocks(
212-
self, AdiagTinv, d_count, batch_block, batch_indices, Asubdiag, f, tInd,
213-
solution_type, Jmatrix
214-
)
215-
field_deriv_blocks.append(dask.array.hstack(f_blocks))
216-
j_row_blocks += j_blocks
217219
del field_derivatives
218220

219221
if self.store_sensitivities == "disk":

0 commit comments

Comments
 (0)