fix env

Signed-off-by: MengqingCao <[email protected]>

Author: MengqingCao

tests/e2e/multicard/test_data_parallel.py

@@ -33,10 +33,10 @@
                     reason="Data parallel only support on v1")
 @pytest.mark.parametrize("model", MODELS)
 @pytest.mark.parametrize("max_tokens", [32])
+@patch.dict(os.environ, {"ASCEND_RT_VISIBLE_DEVICES": "0,1"})
 def test_data_parallel_inference(model, max_tokens):
     script = "examples/offline_data_parallel.py"
 
-    os.environ["ASCEND_RT_VISIBLE_DEVICES"] = "0,1"
     env = os.environ.copy()
 
     cmd = [