record entry returned by object class map export is no longer a dictionary of...

record entry returned by object class map export is no longer a dictionary of paths, but rather one path for object map

model_server/base/roiset.py

@@ -506,11 +506,11 @@ class RoiSet(object):
             if k == 'annotated_zstacks':
                 self.export_annotated_zstack(subdir, prefix=pr, **kp)
             if k == 'object_classes':
-                record[k] = {}
+                # record[k] = {}
                 for kc, acc in self.object_class_maps.items():
                     fp = subdir / kc / (pr + '.tif')
                     write_accessor_data_to_file(fp, acc)
-                    record[k][kc] = fp.__str__()
+                    record[f'{k}_{kc}'] = fp.__str__()
             if k == 'dataframe':
                 dfpa = subdir / (pr + '.csv')
                 dfpa.parent.mkdir(parents=True, exist_ok=True)

tests/test_roiset.py

@@ -148,9 +148,9 @@ class TestRoiSetMonoProducts(BaseTestRoiSetMonoProducts, unittest.TestCase):
             'obmap',
             RoiSetExportParams(object_classes=True)
         )
-        opa = record['object_classes']['dummy_class']
-        acc = generate_file_accessor(opa)
+        opa = record['object_classes_dummy_class']
         self.assertTrue(Path(opa).exists())
+        acc = generate_file_accessor(opa)
         self.assertTrue(all(np.unique(acc.data) == [0, 1]))