diff --git a/model_server/base/roiset.py b/model_server/base/roiset.py index dd3139a13ee5ffcce0e1df07da395d456b0dc2de..bbe3a5f8208db4b4874df54d88525ffc162664a6 100644 --- a/model_server/base/roiset.py +++ b/model_server/base/roiset.py @@ -428,7 +428,7 @@ class RoiSet(object): raw.data[:, :, ci, :] ) else: - if white_channel: # interpret as just a single channel + if white_channel is not None: # interpret as just a single channel assert white_channel < raw.chroma annotate_rgb = False for k in ['contour_channel', 'bounding_box_channel', 'mask_channel']: diff --git a/tests/test_roiset.py b/tests/test_roiset.py index e281b7a006b37f95a8ce52ea532e2bdddfe93ecd..9d32668ccf98a376bc421c4167a6a7fbc0d2b163 100644 --- a/tests/test_roiset.py +++ b/tests/test_roiset.py @@ -251,7 +251,7 @@ class TestRoiSetMultichannelProducts(BaseTestRoiSetMonoProducts, unittest.TestCa where = output_path / 'multichannel' / 'mono_2d_patches' df_res = self.roiset.export_patches( where, - white_channel=3, + white_channel=0, draw_bounding_box=True, expanded=True, pad_to=256,