diff --git a/infra/k8s/argo/train-dqn-template.yaml b/infra/k8s/argo/train-dqn-template.yaml
index 97ea91175..64c62a5a6 100644
--- a/infra/k8s/argo/train-dqn-template.yaml
+++ b/infra/k8s/argo/train-dqn-template.yaml
@@ -52,6 +52,8 @@ spec:
         value: /data/futures-baseline-mbp10
       - name: trades-data-dir
         value: /data/futures-baseline-trades
+      - name: feature-cache-dir
+        value: /data/feature-cache
       - name: tx-cost-bps
         value: "0.1"
       - name: tick-size