Add Axolotl Pooch hash (#347)

* include hash for axolotl * Include hash of individual files used * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * leave orientation as is * refactor to satisfy line lengths --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com> Co-authored-by: Alessandro Felder <[email protected]> Co-authored-by: alessandrofelder <[email protected]>
brainglobe · Aug 6, 2024 · 33f9205 · 33f9205
1 parent 9aa3c72
commit 33f9205
Showing 1 changed file with 17 additions and 6 deletions.
diff --git a/brainglobe_atlasapi/atlas_generation/atlas_scripts/axolotl_atlas.py b/brainglobe_atlasapi/atlas_generation/atlas_scripts/axolotl_atlas.py
@@ -43,12 +43,23 @@ def create_atlas(working_dir, resolution):
 
  # download atlas files
  utils.check_internet_connection()
- pooch.retrieve(
- url=f"{ATLAS_FILE_URL.replace('.org', '.org/api')}/files-archive",
- known_hash=None, # FIXME
- processor=pooch.Unzip(extract_dir=atlas_path),
- progressbar=True,
- )
+ hashes = [
+ "md5:3a9ba5a23c17180981b5678329915226",
+ "md5:66df0da5d7eed10ff59add32741d0bf2",
+ "md5:ab13eb8b8f9324a67fdd162f4e79f3c0",
+ ]
+ list_files = {
+ "axolotl_labels_66rois_40micra.nii.gz": hashes[0],
+ "axolotl_template_40micra.nii.gz": hashes[1],
+ "axolotl_label_names_66rois.csv": hashes[2],
+ }
+ for filename, hash in list_files.items():
+ pooch.retrieve(
+ url=f"{ATLAS_FILE_URL}/files/{filename}",
+ known_hash=hash,
+ path=atlas_path,
+ progressbar=True,
+ )
 
  structures_file = atlas_path / "axolotl_label_names_66rois.csv"
  annotations_file = atlas_path / "axolotl_labels_66rois_40micra.nii.gz"