Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Dec 9, 2024
1 parent 1eebd23 commit 20f1dd1
Showing 1 changed file with 8 additions and 5 deletions.
13 changes: 8 additions & 5 deletions src/anemoi/inference/checkpoint.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,8 +11,8 @@
import datetime
import logging
from collections import defaultdict
from pathlib import Path
from functools import cached_property
from pathlib import Path

from anemoi.utils.checkpoints import load_metadata
from earthkit.data.utils.dates import to_datetime
Expand All @@ -25,19 +25,22 @@
def _download_huggingfacehub(huggingface_config):
"""Download model from huggingface"""
try:
from huggingface_hub import hf_hub_download, snapshot_download
from huggingface_hub import hf_hub_download
from huggingface_hub import snapshot_download
except ImportError as e:
raise ImportError("Could not import `huggingface_hub`, please run `pip install huggingface_hub`.") from e

if 'filename' in huggingface_config:
if "filename" in huggingface_config:
config_path = hf_hub_download(**huggingface_config)
else:
repo_path = Path(snapshot_download(**huggingface_config))
ckpt_files = list(repo_path.glob('*.ckpt'))
ckpt_files = list(repo_path.glob("*.ckpt"))
if len(ckpt_files) == 1:
return str(ckpt_files[0])
else:
ValueError(f"Multiple ckpt files found in repo, {ckpt_files}.\nCannot pick one to load, please specify `filename`.")
ValueError(
f"Multiple ckpt files found in repo, {ckpt_files}.\nCannot pick one to load, please specify `filename`."
)
return config_path


Expand Down

0 comments on commit 20f1dd1

Please sign in to comment.