@@ -167,7 +167,8 @@ def test_cpu_only_flow(self, config):
167167
168168 # Put & Verify Meta
169169 meta = client .put (keys , vals )
170- assert all (m == "KVClient" for m in meta )
170+ # b"\x01" is a tag added by YuanrongStorageClient, indicating that it is processed via General KV path.
171+ assert all (m == b"\x02 " for m in meta )
171172
172173 # Get & Verify Values
173174 ret = client .get (keys , shp , dt , meta )
@@ -186,7 +187,8 @@ def test_npu_only_flow(self, config):
186187 client = self .client_cls (config )
187188
188189 meta = client .put (keys , vals )
189- assert all (m == "DsTensorClient" for m in meta )
190+ # b"\x01" is a tag added by YuanrongStorageClient, indicating that it is processed via NPU path.
191+ assert all (m == b"\x01 " for m in meta )
190192
191193 ret = client .get (keys , shp , dt , meta )
192194 for o , r in zip (vals , ret , strict = True ):
@@ -199,7 +201,7 @@ def test_mixed_flow(self, config):
199201 client = self .client_cls (config )
200202
201203 meta = client .put (keys , vals )
202- assert set (meta ) == {"DsTensorClient " , "KVClient " }
204+ assert set (meta ) == {b" \x01 " , b" \x02 " }
203205
204206 ret = client .get (keys , shp , dt , meta )
205207 for o , r in zip (vals , ret , strict = True ):
0 commit comments