File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed
backends/test/suite/tests Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change 1+ import json
12import unittest
23
34from 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 """
You can’t perform that action at this time.
0 commit comments