Spaces:
Sleeping
Sleeping
Commit
•
d977997
1
Parent(s):
14c6f3f
check set membership before function call
Browse files
app.py
CHANGED
@@ -47,12 +47,16 @@ class HubIDCell(Preprocessor):
|
|
47 |
resources.setdefault("model_matches", set())
|
48 |
if match := re.search(hub_id_regex, cell["source"]):
|
49 |
hub_id_match = match.groups(0)[0]
|
50 |
-
if
|
51 |
-
hub_id_match
|
52 |
-
|
53 |
-
|
54 |
-
if
|
55 |
-
|
|
|
|
|
|
|
|
|
56 |
return cell, resources
|
57 |
|
58 |
|
|
|
47 |
resources.setdefault("model_matches", set())
|
48 |
if match := re.search(hub_id_regex, cell["source"]):
|
49 |
hub_id_match = match.groups(0)[0]
|
50 |
+
if (
|
51 |
+
hub_id_match not in resources["model_matches"]
|
52 |
+
or resources["dataset_matches"]
|
53 |
+
):
|
54 |
+
if hub_check := check_hub_item(hub_id_match):
|
55 |
+
hub_id_match, repo_item_type = hub_check
|
56 |
+
if repo_item_type == "model":
|
57 |
+
resources["model_matches"].add(hub_id_match)
|
58 |
+
if repo_item_type == "dataset":
|
59 |
+
resources["dataset_matches"].add(hub_id_match)
|
60 |
return cell, resources
|
61 |
|
62 |
|