fix extension

devsoc-unsw · Sep 25, 2023 · 67a89de · 67a89de
1 parent 49b5837
commit 67a89de
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/backend/server/routers/ctf.py b/backend/server/routers/ctf.py
@@ -73,20 +73,21 @@ def hard_requirements(data: PlannerData) -> bool:
     # including validity of the program
     return (
         data.program == "3778"
-        # and "COMPA1" in data.specialisations
-        # and "MATHC2" in data.specialisations
-        # and len(data.plan) == 3
+        and "COMPA1" in data.specialisations
+        and "MATHC2" in data.specialisations
+        and len(data.plan) == 3
     )
 
 
 def extended_courses(data: PlannerData) -> bool:
-    return len({
+    extended_courses = {
         "COMP3821",
         "COMP3891",
         "COMP6841",
         "COMP6843"
         "COMP6845"
-    }.intersection(all_courses(data))) >= 4
+    }
+    return len(extended_courses & all_courses(data)) >= 3
 
 
 def summer_course(data: PlannerData) -> bool: