From 6b3aef6416524c49c9f604ea0475ae93fdf984e2 Mon Sep 17 00:00:00 2001
From: Arjun Suresh <arjunsuresh1987@gmail.com>
Date: Fri, 25 Nov 2022 23:02:01 +0000
Subject: [PATCH] Fix

---
 cm-mlops/script/get-preprocessed-dataset-imagenet/preprocess.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cm-mlops/script/get-preprocessed-dataset-imagenet/preprocess.py b/cm-mlops/script/get-preprocessed-dataset-imagenet/preprocess.py
index 33e5472fa2..1b9178324b 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):
+if os.environ.get('CM_IMAGENET_QUANTIZED', False) == "yes":
     import generic_preprocess
     generic_preprocess.preprocess()
 else: