Skip to content

Commit b2bd47a

Browse files
committed
Change name of run_test_comparator function in harness.py
1 parent 9ae69e8 commit b2bd47a

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

tests/py/dynamo/conversion/harness.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -251,7 +251,7 @@ def run_test(
251251
check_dtype,
252252
)
253253

254-
def run_test_comparator(
254+
def run_test_compare_tensor_attributes_only(
255255
self,
256256
mod,
257257
inputs,

tests/py/dynamo/conversion/test_rand_aten.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@ def forward(self, x):
8686
x.dtype == y.dtype if check_dtype else True
8787
)
8888
expected_ops = []
89-
self.run_test_comparator(
89+
self.run_test_compare_tensor_attributes_only(
9090
rand_model,
9191
inputs,
9292
expected_ops,
@@ -122,7 +122,7 @@ def forward(self, x):
122122
)
123123
expected_ops = []
124124
# TRT-TRT returns int32 while torch returns int64
125-
self.run_test_comparator(
125+
self.run_test_compare_tensor_attributes_only(
126126
rand_model,
127127
inputs,
128128
expected_ops,

0 commit comments

Comments
 (0)