From fe72093203e568c91bf73974bf362d195984fb7c Mon Sep 17 00:00:00 2001
From: Shangyan Zhou <sy.zhou@deepseek.com>
Date: Wed, 25 Jun 2025 18:18:12 +0800
Subject: [PATCH] Use env.

---
 tests/test_internode.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_internode.py b/tests/test_internode.py
index 59ad9ae..bb5cd58 100644
--- a/tests/test_internode.py
+++ b/tests/test_internode.py
@@ -222,7 +222,7 @@ def test_main(num_sms: int, local_rank: int, num_local_ranks: int, num_ranks: in
 def test_loop(local_rank: int, num_local_ranks: int):
     num_nodes = int(os.getenv('WORLD_SIZE', 1))
     rank, num_ranks, group = init_dist(local_rank, num_local_ranks)
-    test_ll_compatibility = True
+    test_ll_compatibility = os.getenv('EP_TEST_LL_COMPATIBILITY', False)
     if test_ll_compatibility:
         ll_num_tokens, ll_hidden, ll_num_experts, ll_num_topk = 16, 5120, 256, 9