Fix default values when json

mlexchange · Jun 5, 2024 · a5f59b2 · a5f59b2
1 parent 46fc0ec
commit a5f59b2
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/frontend.py b/src/frontend.py
@@ -46,14 +46,14 @@
 FLOW_TYPE = os.getenv("FLOW_TYPE", "podman")
 
 # Slurm
-PARTITIONS_CPU = json.loads(os.getenv("PARTITIONS_CPU", []))
-RESERVATIONS_CPU = json.loads(os.getenv("RESERVATIONS_CPU", []))
+PARTITIONS_CPU = json.loads(os.getenv("PARTITIONS_CPU", "[]"))
+RESERVATIONS_CPU = json.loads(os.getenv("RESERVATIONS_CPU", "[]"))
 MAX_TIME_CPU = os.getenv("MAX_TIME_CPU", "1:00:00")
-PARTITIONS_GPU = json.loads(os.getenv("PARTITIONS_CPU", []))
-RESERVATIONS_GPU = json.loads(os.getenv("RESERVATIONS_CPU", []))
+PARTITIONS_GPU = json.loads(os.getenv("PARTITIONS_CPU", "[]"))
+RESERVATIONS_GPU = json.loads(os.getenv("RESERVATIONS_CPU", "[]"))
 MAX_TIME_GPU = os.getenv("MAX_TIME_CPU", "1:00:00")
 SUBMISSION_SSH_KEY = os.getenv("SUBMISSION_SSH_KEY", "")
-FORWARD_PORTS = json.loads(os.getenv("FORWARD_PORTS", []))
+FORWARD_PORTS = json.loads(os.getenv("FORWARD_PORTS", "[]"))
 
 # Mlex content api
 CONTENT_API_URL = os.getenv("CONTENT_API_URL", "http://localhost:8000/api/v0/models")