Abhishek Thakur commited on
Commit
472a04b
·
1 Parent(s): 527614c

use competition task

Browse files
Files changed (1) hide show
  1. 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, project_id, submission_dataset, model, dataset):
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": project_id,
230
  "task": 1,
231
  "config": {
232
  "language": "en",
233
  "max_models": 5,
234
- "benchmark": {
235
- "dataset": dataset,
236
- "model": model,
237
- "submission_dataset": submission_dataset,
238
- "create_prediction_repo": False,
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(5)
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
- project_id=f"{submission_id}",
341
- dataset=f"{self.competition_id}",
342
- submission_dataset=user_id,
343
- model="generic_competition",
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