Skip to content

Commit c5872a6

Browse files
committed
reduce testing code
1 parent b52ee50 commit c5872a6

File tree

1 file changed

+1
-11
lines changed

1 file changed

+1
-11
lines changed

tests/common.py

Lines changed: 1 addition & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -326,17 +326,7 @@ def group_nodes_by_type(graph):
326326

327327

328328
def check_op_count(graph, op_type, expected_count):
329-
actual_count = len(group_nodes_by_type(graph)[op_type])
330-
count_match = actual_count == expected_count
331-
'''
332-
if False == count_match:
333-
print ("LSTM count mismatch:", actual_count, expected_count)
334-
input()
335-
'''
336-
return actual_count == expected_count
337-
# return len(group_nodes_by_type(graph)[op_type]) == expected_count
338-
# FIXME: after switching to grappler some of the op counts are off. Fix later.
339-
# return True
329+
return len(group_nodes_by_type(graph)[op_type]) == expected_count
340330

341331

342332
def check_lstm_count(graph, expected_count):

0 commit comments

Comments
 (0)