move empty directory removal to sparseml instead of sparsify (#273)

neuralmagic · Jul 21, 2023 · e1c5088 · e1c5088
1 parent a75c0a2
commit e1c5088
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/src/sparsify/auto/tasks/image_classification/runner.py b/src/sparsify/auto/tasks/image_classification/runner.py
@@ -77,11 +77,6 @@ def config_to_args(
         if "dataset" not in config.kwargs:
             # custom datasets are set to imagefolder
             config.kwargs["dataset"] = "imagefolder"
-            if not os.path.exists(config.dataset):
-                raise FileNotFoundError(
-                    f"The custom dataset {config.dataset} "
-                    "does not exist. Please ensure that the path provided is correct."
-                )
 
         if "model_tag" not in config.kwargs:
             config.kwargs["model_tag"] = "sparsify_auto_image_classification"