Expose method to get dataframe in a format that is JSON-serializable

model_server/base/roiset.py

@@ -538,9 +538,12 @@ class RoiSet(object):
 
         return InMemoryDataAccessor(om)
 
+    def get_serializable_dataframe(self) -> pd.DataFrame:
+        return self._df.drop(['expanded_slice', 'slice', 'relative_slice', 'binary_mask'], axis=1)
+
     def export_dataframe(self, csv_path: Path) -> str:
         csv_path.parent.mkdir(parents=True, exist_ok=True)
-        self._df.drop(['expanded_slice', 'slice', 'relative_slice', 'binary_mask'], axis=1).to_csv(csv_path, index=False)
+        self.get_serializable_dataframe().to_csv(csv_path, index=False)
         return csv_path.name
 
 

tests/base/test_roiset.py

@@ -5,7 +5,6 @@ import numpy as np
 from pathlib import Path
 
 import pandas as pd
-from skimage import draw
 
 from model_server.base.roiset import filter_df_overlap_bbox, filter_df_overlap_seg, RoiSetExportParams, RoiSetMetaParams
 from model_server.base.roiset import RoiSet