From eb47dc5c63403aca1d7b5467a2528a1080a09ae8 Mon Sep 17 00:00:00 2001 From: Ethan Harris Date: Fri, 7 Jan 2022 19:03:01 +0000 Subject: [PATCH] Some fixes for compatibility with PL master --- flash/core/data/io/input_transform.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/flash/core/data/io/input_transform.py b/flash/core/data/io/input_transform.py index f61c16dd7a..46be001a1a 100644 --- a/flash/core/data/io/input_transform.py +++ b/flash/core/data/io/input_transform.py @@ -921,7 +921,7 @@ def _resolve_transforms(self, running_stage: RunningStage) -> Optional[Dict[str, from flash.core.data.data_pipeline import DataPipeline transforms_out = {} - stage = _STAGES_PREFIX[running_stage.value] + stage = _STAGES_PREFIX[running_stage] # iterate over all transforms hook name for transform_name in InputTransformPlacement: @@ -997,7 +997,7 @@ def _check_transforms( if transform is None: return transform - keys_diff = set(transform.keys()).difference([v for v in InputTransformPlacement]) + keys_diff = set(transform.keys()).difference([v.value for v in InputTransformPlacement]) if len(keys_diff) > 0: raise MisconfigurationException(