camera-ready

Wuyxin · May 1, 2023 · 4c955d8 · 4c955d8
1 parent c1cabcb
commit 4c955d8
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/README.md b/README.md
@@ -53,8 +53,6 @@ DISC is an algorithm on image classification tasks which adaptively discovers an
 
 ### **How does DISC do it?**
 
-
-
 - Build a concept bank with multiple categories. Examples from the concept bank:
 
  <p align="center">
@@ -93,7 +91,7 @@ pip install tarfile zipfile gdown # Used for data download
 
 ```shell
 cd disc
-python download_all.py
+python download_datasets.py
 ```
 <details>
   <summary>Manual download (If auto download fails)</summary>

diff --git a/concept_bank/download.py b/concept_bank/download.py
@@ -45,5 +45,5 @@
         dst_path = os.path.join(concept_root, f)
         shutil.move(src_path, dst_path)
 
-os.remove(tmp_dir)
+shutil.rmtree(tmp_dir)
 
diff --git a/disc/download_all.py → disc/download_datasets.py b/disc/download_all.py → disc/download_datasets.py