Abhishek Thakur
commited on
Commit
·
472a04b
1
Parent(s):
527614c
use competition task
Browse files- competitions/submissions.py +12 -12
competitions/submissions.py
CHANGED
|
@@ -217,7 +217,7 @@ class Submissions:
|
|
| 217 |
raise AuthenticationError("Please verify your email on Hugging Face Hub")
|
| 218 |
return user_info
|
| 219 |
|
| 220 |
-
def _create_autotrain_project(self,
|
| 221 |
project_config = {}
|
| 222 |
project_config["dataset_name"] = "lewtun/imdb-dummy"
|
| 223 |
project_config["dataset_config"] = "lewtun--imdb-dummy"
|
|
@@ -226,16 +226,16 @@ class Submissions:
|
|
| 226 |
|
| 227 |
payload = {
|
| 228 |
"username": self.autotrain_username,
|
| 229 |
-
"proj_name":
|
| 230 |
"task": 1,
|
| 231 |
"config": {
|
| 232 |
"language": "en",
|
| 233 |
"max_models": 5,
|
| 234 |
-
"
|
| 235 |
-
"
|
| 236 |
-
"
|
| 237 |
-
"
|
| 238 |
-
"
|
| 239 |
},
|
| 240 |
},
|
| 241 |
}
|
|
@@ -285,7 +285,7 @@ class Submissions:
|
|
| 285 |
if project_status["status"] == 3:
|
| 286 |
is_data_processing_success = True
|
| 287 |
logger.info("✅ Data processing complete!")
|
| 288 |
-
time.sleep(
|
| 289 |
|
| 290 |
# Approve training job
|
| 291 |
_ = http_post(
|
|
@@ -337,10 +337,10 @@ class Submissions:
|
|
| 337 |
)
|
| 338 |
# schedule submission for evaluation
|
| 339 |
self._create_autotrain_project(
|
| 340 |
-
|
| 341 |
-
|
| 342 |
-
|
| 343 |
-
|
| 344 |
)
|
| 345 |
remaining_submissions = self.submission_limit - submissions_made
|
| 346 |
return remaining_submissions
|
|
|
|
| 217 |
raise AuthenticationError("Please verify your email on Hugging Face Hub")
|
| 218 |
return user_info
|
| 219 |
|
| 220 |
+
def _create_autotrain_project(self, submission_id, competition_id, user_id, competition_type):
|
| 221 |
project_config = {}
|
| 222 |
project_config["dataset_name"] = "lewtun/imdb-dummy"
|
| 223 |
project_config["dataset_config"] = "lewtun--imdb-dummy"
|
|
|
|
| 226 |
|
| 227 |
payload = {
|
| 228 |
"username": self.autotrain_username,
|
| 229 |
+
"proj_name": submission_id,
|
| 230 |
"task": 1,
|
| 231 |
"config": {
|
| 232 |
"language": "en",
|
| 233 |
"max_models": 5,
|
| 234 |
+
"competition": {
|
| 235 |
+
"submission_id": submission_id,
|
| 236 |
+
"competition_id": competition_id,
|
| 237 |
+
"user_id": user_id,
|
| 238 |
+
"competition_type": "generic",
|
| 239 |
},
|
| 240 |
},
|
| 241 |
}
|
|
|
|
| 285 |
if project_status["status"] == 3:
|
| 286 |
is_data_processing_success = True
|
| 287 |
logger.info("✅ Data processing complete!")
|
| 288 |
+
time.sleep(3)
|
| 289 |
|
| 290 |
# Approve training job
|
| 291 |
_ = http_post(
|
|
|
|
| 337 |
)
|
| 338 |
# schedule submission for evaluation
|
| 339 |
self._create_autotrain_project(
|
| 340 |
+
submission_id=f"{submission_id}",
|
| 341 |
+
competition_id=f"{self.competition_id}",
|
| 342 |
+
user_id=user_id,
|
| 343 |
+
competition_type="generic",
|
| 344 |
)
|
| 345 |
remaining_submissions = self.submission_limit - submissions_made
|
| 346 |
return remaining_submissions
|