Merge branch 'main' into checksum

jet-net · Oct 24, 2023 · 4ac9b5c · 4ac9b5c
2 parents 882aae2 + 436091b
commit 4ac9b5c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -27,7 +27,7 @@ repos:
     args: ["--profile", "black"]
 
 - repo: https://github.com/psf/black-pre-commit-mirror
-  rev: 23.9.1
+  rev: 23.10.0
   hooks:
   - id: black-jupyter
     language_version: python3

diff --git a/jetnet/datasets/utils.py b/jetnet/datasets/utils.py
@@ -79,9 +79,9 @@ def _getZenodoFileURL(record_id: int, file_name: str) -> str:
 
     records_url = f"https://zenodo.org/api/records/{record_id}"
     r = requests.get(records_url).json()
-    file = next(item for item in r["files"] if item["key"] == file_name)
-    file_url = file["links"]["self"]
-    md5 = file["checksum"].split("md5:")[1]
+    file = next(item for item in r["files"] if item["filename"] == file_name)
+    file_url = file["links"]["download"]
+    md5 = file["checksum"]
     return file_url, md5