diff --git a/infra/k8s/argo/train-template.yaml b/infra/k8s/argo/train-template.yaml
index 818fd64b3..bb02f1375 100644
--- a/infra/k8s/argo/train-template.yaml
+++ b/infra/k8s/argo/train-template.yaml
@@ -400,6 +400,8 @@ spec:
             value: ":4096:8"
           - name: FOXHUNT_FEATURE_CACHE_DIR
             value: /feature-cache
+          - name: FOXHUNT_NO_GRAPH
+            value: "1"
         resources:
           requests:
             nvidia.com/gpu: "1"
@@ -484,6 +486,8 @@ spec:
             value: ":4096:8"
           - name: FOXHUNT_FEATURE_CACHE_DIR
             value: /feature-cache
+          - name: FOXHUNT_NO_GRAPH
+            value: "1"
         resources:
           requests:
             nvidia.com/gpu: "1"