Skip to content

Commit 34dee5e

Browse files
author
DvirDukhan
committed
re-export of v3 rdb
1 parent 92084f9 commit 34dee5e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/flow/test_serializations.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -246,7 +246,7 @@ def test_v3_torch_model(self):
246246
def test_v3_troch_script(self):
247247
key_name = "torch_script{1}"
248248
con = self.env.getConnection()
249-
script_rdb =b'\x07\x81\x00\x8f\xd2\t\x12\x0fL\x03\x05\x04CPU\x00\x05\x0fTORCH_SCRIPT_V3\x05\xc3@|A\x01\x16def bar(tensors: List[T`\r\x05], key\xc0\x13\x01st@\x10\x02arg\xe0\x03\x10\x03):\n \x00\x03a = \xa0A\x02[0]`\x12\x00b\xe0\x02\x12\x001\x80\x12\x05return ,\x05+ b\n\nd\x80y\x08_variadic\xe0I\x82\x00l\xe0\x03\x82\x00:\xe0\x08\x83\x00l \xaa\x01\n\x00\x02\x02\x05\x03bar\x05\x0cbar_variadic\x00\t\x00\xaap\xbfWl\xe6\x9e~'
249+
script_rdb =b'\x07\x81\x00\x8f\xd2\t\x12\x0fL\x03\x05\x04CPU\x00\x05\x0fTORCH_SCRIPT_V3\x05\xc3@|A\x01\x16def bar(tensors: List[T`\r\x05], key\xc0\x13\x01st@\x10\x02arg\xe0\x03\x10\x03):\n \x00\x03a = \xa0A\x02[0]`\x12\x00b\xe0\x02\x12\x001\x80\x12\x05return ,\x05+ b\n\nd\x80y\x08_variadic\xe0I\x82\x00l\xe0\x03\x82\x00:\xe0\x08\x83\x00l \xaa\x01\n\x00\x02\x02\x05\x04bar\x00\x05\rbar_variadic\x00\x00\t\x00\xd3\x88D\xce3ek\x01'
250250
con.restore(key_name, 0, script_rdb, True)
251251
_, device, _, tag, _, entry_points = con.execute_command("AI.SCRIPTGET", key_name, "META")
252252
self.env.assertEqual([device, tag, entry_points], [b"CPU", b"TORCH_SCRIPT_V3", [b'bar', b'bar_variadic']])

0 commit comments

Comments
 (0)