fix(ci): fixup checksum scanning pipeline (mudler#3631)

Signed-off-by: Ettore Di Giacinto <[email protected]>
kp-forks · Sep 23, 2024 · 3e8e71f · 3e8e71f
1 parent 4edd8c8
commit 3e8e71f
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/.github/check_and_update.py b/.github/check_and_update.py
@@ -29,9 +29,14 @@ def calculate_sha256(file_path):
 def manual_safety_check_hf(repo_id):
     scanResponse = requests.get('https://huggingface.co/api/models/' + repo_id + "/scan")
     scan = scanResponse.json()
-    if scan['hasUnsafeFile']:
-        return scan
-    return None
+    # Check if 'hasUnsafeFile' exists in the response
+    if 'hasUnsafeFile' in scan:
+        if scan['hasUnsafeFile']:
+            return scan
+        else:
+            return None
+    else:
+        return None
 
 download_type, repo_id_or_url = parse_uri(uri)