Skip to content

Commit ed8e966

Browse files
vivianp17scottclowe
authored andcommitted
MNT:CanInv: Fix file download path and add HF link
Ensured metadata and files are saved in the correct root directory Added Hugging Face download URL
1 parent 3692c67 commit ed8e966

1 file changed

Lines changed: 7 additions & 3 deletions

File tree

bioscan_dataset/CanadianInvertebrates.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -342,9 +342,11 @@ class CanadianInvertebrates(Dataset):
342342
"""
343343

344344
base_folder = "CanadianInvertebrates"
345-
346345
meta = {
347-
"urls": ["https://vault.cs.uwaterloo.ca/s/9bnzWdb5fCpdRwQ/download/CanInv_metadata.csv"],
346+
"urls": [
347+
"https://vault.cs.uwaterloo.ca/s/9bnzWdb5fCpdRwQ/download/CanInv_metadata.csv",
348+
"https://huggingface.co/datasets/bioscan-ml/CanadianInvertebrates-ML/blob/main/CanInv_metadata.csv",
349+
],
348350
"filename": "CanInv_metadata.csv",
349351
"csv_md5": "38c1cbd95e59458b65b39f9021dcb7ff",
350352
}
@@ -646,7 +648,9 @@ def _download_metadata(self, verbose=1) -> None:
646648
if verbose >= 1:
647649
print("Metadata CSV file already downloaded and verified")
648650
return
649-
download_url(self.meta["urls"][0], self.root, filename="CanInv_metadata.csv")
651+
target_dir = os.path.join(self.root, self.base_folder)
652+
os.makedirs(target_dir, exist_ok=True)
653+
download_url(self.meta["urls"][0], target_dir, md5=self.meta["csv_md5"])
650654

651655
def download(self) -> None:
652656
r"""

0 commit comments

Comments
 (0)