Commit
·
269eb04
1
Parent(s):
7b2f58e
Update app.py
Browse files
app.py
CHANGED
|
@@ -243,16 +243,15 @@ def certification(hf_username, first_name, last_name):
|
|
| 243 |
]
|
| 244 |
for unit in results_certification:
|
| 245 |
if unit["unit"] == "Unit 6":
|
| 246 |
-
|
| 247 |
-
|
| 248 |
-
|
| 249 |
-
|
| 250 |
-
|
| 251 |
elif unit["unit"] != "Unit 8 PII":
|
| 252 |
-
|
| 253 |
-
|
| 254 |
-
|
| 255 |
-
|
| 256 |
else:
|
| 257 |
user_models = get_user_sf_models(hf_username, unit['env'], unit['library'])
|
| 258 |
|
|
|
|
| 243 |
]
|
| 244 |
for unit in results_certification:
|
| 245 |
if unit["unit"] == "Unit 6":
|
| 246 |
+
# Since Unit 6 can use PandaReachDense-v2 or v3
|
| 247 |
+
user_models = get_user_models(hf_username, "PandaReachDense-v3", unit["library"])
|
| 248 |
+
if len(user_models) == 0:
|
| 249 |
+
print("Empty")
|
| 250 |
+
user_models = get_user_models(hf_username, "PandaReachDense-v2", unit["library"])
|
| 251 |
elif unit["unit"] != "Unit 8 PII":
|
| 252 |
+
# Get user model
|
| 253 |
+
user_models = get_user_models(hf_username, unit['env'], unit['library'])
|
| 254 |
+
# For sample factory vizdoom we don't have env tag for now
|
|
|
|
| 255 |
else:
|
| 256 |
user_models = get_user_sf_models(hf_username, unit['env'], unit['library'])
|
| 257 |
|