Update app.py

app.py

@@ -103,9 +103,9 @@ def check_if_passed(model):
   :param model: user model
   """
   if model["best_result"] >= model["min_result"]:
-    model["passed"] = True
+    model["passed_"] = True
 
-def test_(hf_username):
+def certification(hf_username):
   results_certification = [
       {
           "unit": "Unit 1",
@@ -114,17 +114,8 @@ def test_(hf_username):
           "min_result": 200,
           "best_result": 0,
           "best_model_id": "",
-          "passed": False
+          "passed_": False
       },
-    {
-          "unit": "Bonus Unit 1",
-          "env": "ML-Agents-Huggy",
-          "library": "ml-agents",
-          "min_result": -100,
-          "best_result": 0,
-          "best_model_id": "",
-          "passed": False
-    },
   {
           "unit": "Unit 2",
           "env": "Taxi-v3",
@@ -132,7 +123,7 @@ def test_(hf_username):
           "min_result": 4,
           "best_result": 0,
           "best_model_id": "",
-          "passed": False
+          "passed_": False
   },
   {
           "unit": "Unit 3",
@@ -141,7 +132,7 @@ def test_(hf_username):
           "min_result": 200,
           "best_result": 0,
           "best_model_id": "",
-          "passed": False
+          "passed_": False
   },
   {
           "unit": "Unit 4",
@@ -150,7 +141,7 @@ def test_(hf_username):
           "min_result": 350,
           "best_result": 0,
           "best_model_id": "",
-          "passed": False
+          "passed_": False
   },
     {
           "unit": "Unit 4",
@@ -159,7 +150,7 @@ def test_(hf_username):
           "min_result": 5,
           "best_result": 0,
           "best_model_id": "",
-          "passed": False
+          "passed_": False
     },
       {
           "unit": "Unit 5",
@@ -168,7 +159,7 @@ def test_(hf_username):
           "min_result": -100,
           "best_result": 0,
           "best_model_id": "",
-          "passed": False
+          "passed_": False
     },
       {
           "unit": "Unit 5",
@@ -177,7 +168,7 @@ def test_(hf_username):
           "min_result": -100,
           "best_result": 0,
           "best_model_id": "",
-          "passed": False
+          "passed_": False
     },
       {
           "unit": "Unit 6",
@@ -186,7 +177,7 @@ def test_(hf_username):
           "min_result": 650,
           "best_result": 0,
           "best_model_id": "",
-          "passed": False
+          "passed_": False
     },
       {
           "unit": "Unit 6",
@@ -195,7 +186,7 @@ def test_(hf_username):
           "min_result": -3.5,
           "best_result": 0,
           "best_model_id": "",
-          "passed": False
+          "passed_": False
     },
       {
           "unit": "Unit 7",
@@ -204,13 +195,31 @@ def test_(hf_username):
           "min_result": -100,
           "best_result": 0,
           "best_model_id": "",
-          "passed": False
+          "passed_": False
+    },
+      {
+          "unit": "Unit 8 PI",
+          "env": "GodotRL-JumperHard",
+          "library": "cleanrl",
+          "min_result": 100,
+          "best_result": 0,
+          "best_model_id": "",
+          "passed_": False
+    },
+      {
+          "unit": "Unit 8 PII",
+          "env": "Vizdoom-Battle",
+          "library": "cleanrl",
+          "min_result": 100,
+          "best_result": 0,
+          "best_model_id": "",
+          "passed_": False
     },
   ] 
   for unit in results_certification:
     # Get user model
     user_models = get_user_models(hf_username, unit['env'], unit['library'])
-    print(user_models)
+
     # Calculate the best result and get the best_model_id
     best_result, best_model_id = calculate_best_result(user_models)
 
@@ -220,12 +229,12 @@ def test_(hf_username):
 
     # Based on best_result do we pass the unit?
     check_if_passed(unit)
-    #pass_emoji(unit["passed"])
+    unit["passed"] = pass_emoji(unit["passed_"])
     
   print(results_certification)
  
-  df = pd.DataFrame (results_certification)
-
+  df = pd.DataFrame(results_certification)
+  df = df[['passed', 'unit', 'env', 'min_result', 'best_result', 'best_model_id']]
   return df
 
 
@@ -247,7 +256,7 @@ with gr.Blocks() as demo:
     hf_username = gr.Textbox(placeholder="ThomasSimonini", label="Your Hugging Face Username")
     #email = gr.Textbox(placeholder="[email protected]", label="Your Email (to receive your certificate)")
     check_progress_button = gr.Button(value="Check my progress")
-    output = gr.components.Dataframe(value= test_(hf_username), headers=["Unit", "Environment", "Library", "Baseline", "Your best result", "Your best model id", "Pass?"], datatype=["markdown", "markdown", "markdown", "number", "number", "markdown", "bool"])
-    check_progress_button.click(fn=test_, inputs=hf_username, outputs=output)
+    output = gr.components.Dataframe(value= certification(hf_username), headers=["Pass?", "Unit", "Environment", "Baseline", "Your best result", "Your best model id"], datatype=["markdown", "markdown", "markdown", "number", "number", "markdown", "bool"])
+    check_progress_button.click(fn=certification, inputs=hf_username, outputs=output)
 
 demo.launch()