fix device requests when not selecting any gpu

PHT-Medic · Nov 15, 2022 · b2906e9 · b2906e9
1 parent 9c6ee93
commit b2906e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/airflow/dags/DAG_PHT_run_train.py b/airflow/dags/DAG_PHT_run_train.py
@@ -235,7 +235,7 @@ def execute_container(train_state):
                 network_disabled=True,
                 stderr=True,
                 stdout=True,
-                device_requests=[device_request]
+                device_requests=[device_request] if device_request else []
             )
         # If the container is already in use remove it
         except APIError as e: