ThomasSimonini HF staff commited on
Commit
df6fc31
1 Parent(s): b74b163

Update app with PandaReach-v3

Browse files
Files changed (1) hide show
  1. app.py +8 -11
app.py CHANGED
@@ -190,16 +190,7 @@ def certification(hf_username):
190
  },
191
  {
192
  "unit": "Unit 6",
193
- "env": "AntBulletEnv-v0",
194
- "library": "stable-baselines3",
195
- "min_result": 650,
196
- "best_result": 0,
197
- "best_model_id": "",
198
- "passed_": False
199
- },
200
- {
201
- "unit": "Unit 6",
202
- "env": "PandaReachDense-v2",
203
  "library": "stable-baselines3",
204
  "min_result": -3.5,
205
  "best_result": 0,
@@ -234,14 +225,20 @@ def certification(hf_username):
234
  "passed_": False
235
  },
236
  ]
 
237
  for unit in results_certification:
 
 
 
 
 
 
238
  if unit["unit"] != "Unit 8 PII":
239
  # Get user model
240
  user_models = get_user_models(hf_username, unit['env'], unit['library'])
241
  # For sample factory vizdoom we don't have env tag for now
242
  else:
243
  user_models = get_user_sf_models(hf_username, unit['env'], unit['library'])
244
-
245
 
246
  # Calculate the best result and get the best_model_id
247
  best_result, best_model_id = calculate_best_result(user_models)
 
190
  },
191
  {
192
  "unit": "Unit 6",
193
+ "env": "PandaReachDense",
 
 
 
 
 
 
 
 
 
194
  "library": "stable-baselines3",
195
  "min_result": -3.5,
196
  "best_result": 0,
 
225
  "passed_": False
226
  },
227
  ]
228
+
229
  for unit in results_certification:
230
+ if unit["unit"] == "Unit 6":
231
+ # Since Unit 6 can use PandaReachDense-v2 or v3
232
+ user_models = get_user_models(hf_username, "PandaReachDense-v3", unit["library"])
233
+ if len(user_models) == 0:
234
+ print("Empty")
235
+ user_models = get_user_models(hf_username, "PandaReachDense-v2", unit["library"])
236
  if unit["unit"] != "Unit 8 PII":
237
  # Get user model
238
  user_models = get_user_models(hf_username, unit['env'], unit['library'])
239
  # For sample factory vizdoom we don't have env tag for now
240
  else:
241
  user_models = get_user_sf_models(hf_username, unit['env'], unit['library'])
 
242
 
243
  # Calculate the best result and get the best_model_id
244
  best_result, best_model_id = calculate_best_result(user_models)