From 2a923f2962556aba8c7b71c9ca39ce4fe1bb8af5 Mon Sep 17 00:00:00 2001
From: Phuong Nguyen <phuonguyen@nvidia.com>
Date: Thu, 5 Dec 2024 06:31:17 -0800
Subject: [PATCH] set TE_PATH

Signed-off-by: Phuong Nguyen <phuonguyen@nvidia.com>
---
 qa/L0_jax_distributed_unittest/test.sh | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/qa/L0_jax_distributed_unittest/test.sh b/qa/L0_jax_distributed_unittest/test.sh
index 2b8b2e9feb..8b5710d839 100644
--- a/qa/L0_jax_distributed_unittest/test.sh
+++ b/qa/L0_jax_distributed_unittest/test.sh
@@ -4,6 +4,8 @@
 
 set -xe
 
+: ${TE_PATH:=/opt/transformerengine}
+
 pip install -r $TE_PATH/examples/jax/encoder/requirements.txt
 
 # Make encoder tests to have run-to-run deterministic to have the stable CI results