Skip to content

Commit 23f7286

Browse files
committed
Update
[ghstack-poisoned]
1 parent c1c4ccd commit 23f7286

File tree

1 file changed

+5
-2
lines changed

1 file changed

+5
-2
lines changed

backends/test/suite/tests/test_reporting.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import json
12
import unittest
23

34
from csv import DictReader
@@ -102,14 +103,16 @@ def test_csv_report_simple(self):
102103
self.assertEqual(records[2]["Test Case"], "test2")
103104
self.assertEqual(records[2]["Flow"], "flow1")
104105
self.assertEqual(records[2]["Result"], "Pass")
105-
self.assertEqual(records[2]["Params"], str({"dtype": torch.float32}))
106+
self.assertEqual(records[2]["Params"], json.dumps({"dtype": "torch.float32"}))
106107

107108
# Validate fourth record: test2, backend2, EXPORT_FAIL with use_dynamic_shapes param
108109
self.assertEqual(records[3]["Test ID"], "test2_backend2_flow1")
109110
self.assertEqual(records[3]["Test Case"], "test2")
110111
self.assertEqual(records[3]["Flow"], "flow1")
111112
self.assertEqual(records[3]["Result"], "Skip")
112-
self.assertEqual(records[3]["Params"], str({"use_dynamic_shapes": True}))
113+
self.assertEqual(
114+
records[3]["Params"], json.dumps({"use_dynamic_shapes": "True"})
115+
)
113116

114117
def test_count_ops(self):
115118
"""

0 commit comments

Comments
 (0)