fix

Signed-off-by: Xing Liu <xingliu14@gmail.com>

Author: xingliu14

tests/e2e/test_data_parallel.py

@@ -13,7 +13,7 @@
 @pytest.fixture(autouse=True)
 def setup_new_model_design():
     """Automatically set NEW_MODEL_DESIGN=True for all tests."""
-    os.environ['NEW_MODEL_DESIGN'] = 'True'
+    os.environ['NEW_MODEL_DESIGN'] = '1'
 
 
 @pytest.fixture