From de9e7eebd73d11941afca6f269570e6530e0f390 Mon Sep 17 00:00:00 2001 From: Celina Hanouti Date: Tue, 22 Oct 2024 16:26:04 +0200 Subject: [PATCH] Update how file's security metadata is retrieved following changes in the API response (#2621) * Update security key name and BlobSecurityInfo object * typos * small typo * add comment * fix typo again --- src/huggingface_hub/hf_api.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/src/huggingface_hub/hf_api.py b/src/huggingface_hub/hf_api.py index f0dee92a3a..9ffd761f56 100644 --- a/src/huggingface_hub/hf_api.py +++ b/src/huggingface_hub/hf_api.py @@ -336,7 +336,8 @@ def __post_init__(self): # hack to make BlobLfsInfo backward compatible @dataclass class BlobSecurityInfo(dict): - safe: bool + safe: bool # duplicate information with "status" field, keeping it for backward compatibility + status: str av_scan: Optional[Dict] pickle_import_scan: Optional[Dict] @@ -656,10 +657,14 @@ def __init__(self, **kwargs): oid=last_commit["id"], title=last_commit["title"], date=parse_datetime(last_commit["date"]) ) self.last_commit = last_commit - security = kwargs.pop("security", None) + security = kwargs.pop("securityFileStatus", None) if security is not None: + safe = security["status"] == "safe" security = BlobSecurityInfo( - safe=security["safe"], av_scan=security["avScan"], pickle_import_scan=security["pickleImportScan"] + safe=safe, + status=security["status"], + av_scan=security["avScan"], + pickle_import_scan=security["pickleImportScan"], ) self.security = security