fix: slide_etl pandera schema dropping columns

msk-mind · Nov 15, 2023 · afc13cc · afc13cc
1 parent 7232cac
commit afc13cc
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/luna/pathology/cli/slide_etl.py b/src/luna/pathology/cli/slide_etl.py
@@ -13,7 +13,7 @@
 from tiffslide import TiffSlide
 
 from luna.common.dask import configure_dask_client, get_or_create_dask_client
-from luna.common.models import Slide, SlideSchema
+from luna.common.models import Slide
 from luna.common.utils import apply_csv_filter, get_config, timed
 from luna.pathology.common.utils import (
     get_downscaled_thumbnail,
@@ -249,8 +249,7 @@ def __slide_etl(
             output_urlpath,
         )
 
-    df = DataFrame[SlideSchema](pd.json_normalize(slide.__dict__))
-    return df
+    return pd.json_normalize(slide.__dict__)
 
 
 def fire_cli():