From afc13cc68ec16157236ef86819724e17cd21beb5 Mon Sep 17 00:00:00 2001 From: Raymond Lim Date: Wed, 15 Nov 2023 13:41:32 -0500 Subject: [PATCH] fix: slide_etl pandera schema dropping columns --- src/luna/pathology/cli/slide_etl.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/src/luna/pathology/cli/slide_etl.py b/src/luna/pathology/cli/slide_etl.py index c1fb3f97..c11a128f 100644 --- 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():