diff --git a/airflow/dags/DAG_PHT_run_train.py b/airflow/dags/DAG_PHT_run_train.py
index c55d940..ed7336d 100644
--- 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: