We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent e373e44 commit e1661a2Copy full SHA for e1661a2
1 file changed
tests/test_yuanrong_client_zero_copy.py
@@ -45,7 +45,6 @@ def mock_kv_client(self, mocker):
45
46
mocker.patch("yr.datasystem.KVClient", return_value=mock_client)
47
mocker.patch("yr.datasystem.DsTensorClient")
48
- mocker.patch("transfer_queue.storage.clients.yuanrong_client.TORCH_NPU_IMPORTED", False)
49
50
return mock_client
51
0 commit comments