From e70bab4383d436377842eb5f6668554e93ee5748 Mon Sep 17 00:00:00 2001
From: Christopher Rhodes <christopher.rhodes@embl.de>
Date: Tue, 2 Apr 2024 15:18:29 +0200
Subject: [PATCH] RoiSet init can now access both 2d and 3d segmentations

---
 tests/test_roiset.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_roiset.py b/tests/test_roiset.py
index c68b5409..c39c64f8 100644
--- a/tests/test_roiset.py
+++ b/tests/test_roiset.py
@@ -25,7 +25,7 @@ class TestRoiSetMonoProducts(BaseTestRoiSetMonoProducts, unittest.TestCase):
 
     def _make_roi_set(self, mask_type='boxes', **kwargs):
         id_map = _get_label_ids(self.seg_mask)
-        roiset = RoiSet.from_2d_obj_ids(
+        roiset = RoiSet(
             self.stack_ch_pa,
             id_map,
             params=RoiSetMetaParams(
@@ -62,7 +62,7 @@ class TestRoiSetMonoProducts(BaseTestRoiSetMonoProducts, unittest.TestCase):
         self.assertEqual(acc_zstack_slice.nz, 1)
         id_map = _get_label_ids(self.seg_mask)
 
-        roiset = RoiSet.from_2d_obj_ids(acc_zstack_slice, id_map, params=RoiSetMetaParams(mask_type='boxes'))
+        roiset = RoiSet(acc_zstack_slice, id_map, params=RoiSetMetaParams(mask_type='boxes'))
         zmask = roiset.get_zmask()
 
         zmask_acc = InMemoryDataAccessor(zmask)
-- 
GitLab