fix crop_mode

playgroundai · Nov 23, 2023 · a8a8b5d · a8a8b5d
1 parent 1023e49
commit a8a8b5d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/img2dataset/main.py b/img2dataset/main.py
@@ -200,7 +200,7 @@ def signal_handler(signal_arg, frame):  # pylint: disable=unused-argument
         number_sample_per_shard,
         done_shards,
         temp_download_folder,
-        crop_mode,
+        crop_mode if crop_mode != "no" else None,
     )
 
     if output_format == "webdataset":