diff --git a/cm-mlops/script/get-preprocessed-dataset-imagenet/preprocess.py b/cm-mlops/script/get-preprocessed-dataset-imagenet/preprocess.py index 1b9178324b..43e434c073 100644 --- a/cm-mlops/script/get-preprocessed-dataset-imagenet/preprocess.py +++ b/cm-mlops/script/get-preprocessed-dataset-imagenet/preprocess.py @@ -1,7 +1,7 @@ import os import sys -if os.environ.get('CM_IMAGENET_QUANTIZED', False) == "yes": +if os.environ.get('CM_IMAGENET_QUANTIZED', '') == "yes": import generic_preprocess generic_preprocess.preprocess() else: