FIX: Fix resource request names

skipci
octue · Feb 5, 2025 · 9c757a6 · 9c757a6
1 parent 44277cf
commit 9c757a6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/functions/event_handler/main.py b/functions/event_handler/main.py
@@ -110,8 +110,8 @@ def _dispatch_question_as_kueue_job(event, attributes):
 
     resources = {
         "requests": {
-            "cpus": attributes.get("resources_cpu", int(os.environ["QUESTION_DEFAULT_CPUS"])),
-            "memory": attributes.get("resource_memory", os.environ["QUESTION_DEFAULT_MEMORY"]),
+            "cpu": attributes.get("cpus", int(os.environ["QUESTION_DEFAULT_CPUS"])),
+            "memory": attributes.get("memory", os.environ["QUESTION_DEFAULT_MEMORY"]),
             "ephemeral_storage": attributes.get("ephemeral_storage", os.environ["QUESTION_DEFAULT_EPHEMERAL_STORAGE"]),
         }
     }