Avoid error if transform key does not exist

luisas · May 6, 2024 · a2b65a5 · a2b65a5
1 parent 67c1b62
commit a2b65a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bin/launch_transform_csv.py b/bin/launch_transform_csv.py
@@ -38,7 +38,7 @@ def main(data_csv, config_json, out_path):
 
     # Transform the data according to what defined in the experiment class and the specifics of the user in the Json
     # in case of no transformation specification so when the config has "augmentation" : None  just save a copy of the original csv file
-    if config["transform"]: 
+    if config.get("transform") is not None:
         csv_obj.transform(config["transform"])
 
     # save the modified csv