From f211a2770be28f56188baa9eee7c91e16f7f505b Mon Sep 17 00:00:00 2001 From: niekdejonge Date: Tue, 2 Jul 2024 10:52:57 +0200 Subject: [PATCH] Update zenodo version and ms2deepscore model type. --- ms2query/run_ms2query.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/ms2query/run_ms2query.py b/ms2query/run_ms2query.py index fbe26a04..2b77142d 100644 --- a/ms2query/run_ms2query.py +++ b/ms2query/run_ms2query.py @@ -10,8 +10,8 @@ def zenodo_dois(ionisation_mode): "Returns the most up to date url for Zenodo" - zenodo_DOIs = {"positive": 10527997, - "negative": 10528030} + zenodo_DOIs = {"positive": 12622446, + "negative": 12606475} assert ionisation_mode in zenodo_DOIs, "Expected 'positive' or 'negative' as input" zenodo_doi = zenodo_DOIs[ionisation_mode] zenodo_metadata_url = "https://zenodo.org/api/records/" + str(zenodo_doi) @@ -25,12 +25,11 @@ def available_zenodo_files(zenodo_metadata_url, with urlopen(zenodo_metadata_url) as zenodo_metadata_file: file_names_metadata_json: dict = json.loads(zenodo_metadata_file.read()) files = file_names_metadata_json["files"] - file_names_and_sizes = {} for file in files: file_name = file["key"] if only_models: - model_extensions = [".model", ".hdf5", ".onnx", ".npy"] + model_extensions = [".model", ".pt", ".onnx", ".npy"] if any(file_name.endswith(e) for e in model_extensions): file_names_and_sizes[file_name] = file["size"] else: