504 error from UMA test

[ElliottKasoar]

Summary

From #695

=================================== ERRORS ====================================
_______________ ERROR collecting tests/test_mlip_calculators.py _______________
.venv\Lib\site-packages\huggingface_hub\utils\_http.py:761: in hf_raise_for_status
    response.raise_for_status()
.venv\Lib\site-packages\httpx\_models.py:829: in raise_for_status
    raise HTTPStatusError(message, request=request, response=self)
E   httpx.HTTPStatusError: Server error '504 Gateway Time-out' for url 'https://huggingface.co/facebook/UMA/resolve/main/checkpoints/uma-s-1p1.pt'
E   For more information check: https://developer.mozilla.org/en-US/docs/Web/HTTP/Status/504

The above exception was the direct cause of the following exception:
.venv\Lib\site-packages\huggingface_hub\file_download.py:1669: in _get_metadata_or_catch_error
    metadata = get_hf_file_metadata(
.venv\Lib\site-packages\huggingface_hub\utils\_validators.py:88: in _inner_fn
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
.venv\Lib\site-packages\huggingface_hub\file_download.py:1591: in get_hf_file_metadata
    response = _httpx_follow_relative_redirects_with_backoff(
.venv\Lib\site-packages\huggingface_hub\utils\_http.py:685: in _httpx_follow_relative_redirects_with_backoff
    response = http_backoff(
.venv\Lib\site-packages\huggingface_hub\utils\_http.py:559: in http_backoff
    return next(
.venv\Lib\site-packages\huggingface_hub\utils\_http.py:468: in _http_backoff_base
    if not _should_retry(response):
           ^^^^^^^^^^^^^^^^^^^^^^^
.venv\Lib\site-packages\huggingface_hub\utils\_http.py:448: in _should_retry
    hf_raise_for_status(response)  # Will raise uncaught exception
    ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.venv\Lib\site-packages\huggingface_hub\utils\_http.py:877: in hf_raise_for_status
    raise _format(HfHubHTTPError, str(e), response) from e
E   huggingface_hub.errors.HfHubHTTPError: Server error '504 Gateway Time-out' for url 'https://huggingface.co/facebook/UMA/resolve/main/checkpoints/uma-s-1p1.pt' (Amz CF ID: BHGy4s-LgV-InkjECYHMdnkfw6qlVu66lo5LUjlMn4kdRCdGJGJrQw==)
E   For more information check: https://developer.mozilla.org/en-US/docs/Web/HTTP/Status/504

The above exception was the direct cause of the following exception:
tests\test_mlip_calculators.py:51: in <module>
    UMA_PREDICT_UNIT = pretrained_mlip.get_predict_unit(
.venv\Lib\site-packages\fairchem\core\calculate\pretrained_mlip.py:96: in get_predict_unit
    checkpoint_path = pretrained_checkpoint_path_from_name(model_name)
                      ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
.venv\Lib\site-packages\fairchem\core\calculate\pretrained_mlip.py:58: in pretrained_checkpoint_path_from_name
    checkpoint_path = hf_hub_download(
.venv\Lib\site-packages\huggingface_hub\utils\_validators.py:88: in _inner_fn
    return fn(*args, **kwargs)
           ^^^^^^^^^^^^^^^^^^^
.venv\Lib\site-packages\huggingface_hub\file_download.py:997: in hf_hub_download
    return _hf_hub_download_to_cache_dir(
.venv\Lib\site-packages\huggingface_hub\file_download.py:1148: in _hf_hub_download_to_cache_dir
    _raise_on_head_call_error(head_call_error, force_download, local_files_only)
.venv\Lib\site-packages\huggingface_hub\file_download.py:1785: in _raise_on_head_call_error
    raise LocalEntryNotFoundError(
E   huggingface_hub.errors.LocalEntryNotFoundError: An error happened while trying to locate the file on the Hub and we cannot find the requested files in the local cache. Please check your connection and try again or make sure your Internet connection is on.

Some of these tests passed, so it may just be a new one to catch, or there may be something else wrong

Platform

Windows & Linux (CI)

Version

0.9.1

Python version

No response