diff --git a/docker/docker-compose-training.yml b/docker/docker-compose-training.yml
index 02bb62db..0848f465 100644
--- a/docker/docker-compose-training.yml
+++ b/docker/docker-compose-training.yml
@@ -24,6 +24,8 @@ services:
       - MODELMETADATA_FILE_S3_KEY=${DR_LOCAL_S3_MODEL_METADATA_KEY}
       - CUDA_VISIBLE_DEVICES=${DR_SAGEMAKER_CUDA_DEVICES:-}
       - MAX_MEMORY_STEPS=${DR_TRAIN_MAX_STEPS_PER_ITERATION:-}
+      - TELEGRAF_HOST=${DR_TELEGRAF_HOST:-}
+      - TELEGRAF_PORT=${DR_TELEGRAF_PORT:-}      
     volumes:
       - "/var/run/docker.sock:/var/run/docker.sock"
       - "/tmp/sagemaker:/tmp/sagemaker"