Handle tipper and impedance receivers

domfournier · domfournier · commit ab9adc453d7b · 2025-03-10T11:11:28.000-07:00
diff --git a/simpeg/dask/utils.py b/simpeg/dask/utils.py
@@ -57,13 +57,18 @@ def get_parallel_blocks(
                     block_count += 1
                     blocks.append([])
 
+                locs = rx.locations
+
+                if isinstance(locs, tuple):
+                    locs = locs[0]
+
                 blocks[block_count].append(
                     (
                         (s_id, r_id, ind),
                         (
                             chunk,
                             np.arange(row_index, row_index + chunk_size).astype(int),
-                            rx.locations.shape[0],
+                            locs.shape[0],
                         ),
                     )
                 )

Original file line number	Diff line number	Diff line change
`@@ -57,13 +57,18 @@ def get_parallel_blocks(`
`57`	`57`	`block_count += 1`
`58`	`58`	`blocks.append([])`
`59`	`59`
	`60`	`+ locs = rx.locations`
	`61`	`+`
	`62`	`+ if isinstance(locs, tuple):`
	`63`	`+ locs = locs[0]`
	`64`	`+`
`60`	`65`	`blocks[block_count].append(`
`61`	`66`	`(`
`62`	`67`	`(s_id, r_id, ind),`
`63`	`68`	`(`
`64`	`69`	`chunk,`
`65`	`70`	`np.arange(row_index, row_index + chunk_size).astype(int),`
`66`		`- rx.locations.shape[0],`
	`71`	`+ locs.shape[0],`
`67`	`72`	`),`
`68`	`73`	`)`
`69`	`74`	`)`