Skip to content

Commit 3a290fb

Browse files
committed
fix call to scikit-image rescale in Mask2D
1 parent fd98d50 commit 3a290fb

File tree

1 file changed

+16
-7
lines changed

1 file changed

+16
-7
lines changed

autoarray/mask/mask_2d_util.py

Lines changed: 16 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1115,13 +1115,22 @@ def rescaled_mask_2d_from(mask_2d: np.ndarray, rescale_factor: float) -> np.ndar
11151115

11161116
warnings.filterwarnings("ignore")
11171117

1118-
rescaled_mask_2d = rescale(
1119-
image=mask_2d,
1120-
scale=rescale_factor,
1121-
mode="edge",
1122-
anti_aliasing=False,
1123-
multichannel=False,
1124-
)
1118+
try:
1119+
rescaled_mask_2d = rescale(
1120+
image=mask_2d,
1121+
scale=rescale_factor,
1122+
mode="edge",
1123+
anti_aliasing=False,
1124+
multichannel=False,
1125+
)
1126+
except TypeError:
1127+
rescaled_mask_2d = rescale(
1128+
image=mask_2d,
1129+
scale=rescale_factor,
1130+
mode="edge",
1131+
anti_aliasing=False,
1132+
# multichannel=False,
1133+
)
11251134

11261135
rescaled_mask_2d[0, :] = 1
11271136
rescaled_mask_2d[rescaled_mask_2d.shape[0] - 1, :] = 1

0 commit comments

Comments
 (0)