Skip to content

Commit 2ee77fa

Browse files
committed
Change name of run_test_comparator function in harness.py
1 parent f503b61 commit 2ee77fa

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
@@ -267,7 +267,7 @@ def run_test(
267267
check_dtype,
268268
)
269269

270-
def run_test_comparator(
270+
def run_test_compare_tensor_attributes_only(
271271
self,
272272
mod,
273273
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)