Skip to content

Commit 7847233

Browse files
authored
fix:https://nvbugs/5324252 (#4925)
Signed-off-by: nv-guomingz <[email protected]>
1 parent 8731f5f commit 7847233

File tree

1 file changed

+8
-8
lines changed

1 file changed

+8
-8
lines changed

tests/unittest/_torch/test_resource_manager.py

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -406,11 +406,11 @@ def test_put_get(self):
406406
self.assertEqual(len(peft_table), len(expected_values))
407407

408408
for i, entry in enumerate(peft_table):
409-
self.assertEqual(entry.pageId, expected_values[i][0])
410-
self.assertEqual(entry.slotIdx, expected_values[i][1])
411-
self.assertEqual(entry.inSize, expected_values[i][2])
412-
self.assertEqual(entry.outSize, expected_values[i][3])
413-
self.assertEqual(entry.moduleId, expected_values[i][4])
414-
self.assertEqual(entry.layerId, expected_values[i][5])
415-
self.assertEqual(entry.adapterSize, expected_values[i][6])
416-
self.assertEqual(entry.numSlots, expected_values[i][7])
409+
self.assertEqual(entry.page_id, expected_values[i][0])
410+
self.assertEqual(entry.slot_idx, expected_values[i][1])
411+
self.assertEqual(entry.in_size, expected_values[i][2])
412+
self.assertEqual(entry.out_size, expected_values[i][3])
413+
self.assertEqual(entry.module_id, expected_values[i][4])
414+
self.assertEqual(entry.layer_id, expected_values[i][5])
415+
self.assertEqual(entry.adapter_size, expected_values[i][6])
416+
self.assertEqual(entry.num_slots, expected_values[i][7])

0 commit comments

Comments
 (0)