"src/nni_manager/git@developer.sourcefind.cn:OpenDAS/nni.git" did not exist on "66fec2c8fc6156fc33ae908d9994556838d842e2"
Unverified Commit 4fd32a1f authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

Catch `HFValidationError` in `TrainingSummary` (#19252)



* Catch HfValidationError in TrainingSummary
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent f3d2f7a6
...@@ -26,6 +26,7 @@ from typing import Any, Dict, List, Optional, Union ...@@ -26,6 +26,7 @@ from typing import Any, Dict, List, Optional, Union
import requests import requests
import yaml import yaml
from huggingface_hub import model_info from huggingface_hub import model_info
from huggingface_hub.utils import HFValidationError
from . import __version__ from . import __version__
from .models.auto.modeling_auto import ( from .models.auto.modeling_auto import (
...@@ -378,7 +379,7 @@ class TrainingSummary: ...@@ -378,7 +379,7 @@ class TrainingSummary:
for tag in info.tags: for tag in info.tags:
if tag.startswith("license:"): if tag.startswith("license:"):
self.license = tag[8:] self.license = tag[8:]
except requests.exceptions.HTTPError: except (requests.exceptions.HTTPError, HFValidationError):
pass pass
def create_model_index(self, metric_mapping): def create_model_index(self, metric_mapping):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment