Update src/submission/check_validity.py
Browse files
src/submission/check_validity.py
CHANGED
@@ -51,6 +51,7 @@ def is_model_on_hub(model_name: str, revision: str, token: str = None, trust_rem
|
|
51 |
f"uses {tokenizer_class_candidate}, which is not in a transformers release, therefore not supported at the moment.",
|
52 |
None
|
53 |
)
|
|
|
54 |
return True, None, config
|
55 |
|
56 |
except ValueError:
|
@@ -61,6 +62,7 @@ def is_model_on_hub(model_name: str, revision: str, token: str = None, trust_rem
|
|
61 |
)
|
62 |
|
63 |
except Exception as e:
|
|
|
64 |
return False, "was not found on hub!", None
|
65 |
|
66 |
|
@@ -100,4 +102,4 @@ def already_submitted_models(requested_models_dir: str) -> set[str]:
|
|
100 |
organisation, _ = info["model"].split("/")
|
101 |
users_to_submission_dates[organisation].append(info["submitted_time"])
|
102 |
|
103 |
-
return set(file_names), users_to_submission_dates
|
|
|
51 |
f"uses {tokenizer_class_candidate}, which is not in a transformers release, therefore not supported at the moment.",
|
52 |
None
|
53 |
)
|
54 |
+
print(config)
|
55 |
return True, None, config
|
56 |
|
57 |
except ValueError:
|
|
|
62 |
)
|
63 |
|
64 |
except Exception as e:
|
65 |
+
print("exception is", e)
|
66 |
return False, "was not found on hub!", None
|
67 |
|
68 |
|
|
|
102 |
organisation, _ = info["model"].split("/")
|
103 |
users_to_submission_dates[organisation].append(info["submitted_time"])
|
104 |
|
105 |
+
return set(file_names), users_to_submission_dates
|