Updated integration tests after update to client_stats in `work`.
@@ -72,19 +72,16 @@ class CanonicalFormTest(test.TestCase):
' y=int32[?,1]\n'
' >*,\n'
' <\n'
- ' <\n'
' trainable=<\n'
' float32[2,1],\n'
' float32[1]\n'
' >,\n'
' non_trainable=<>\n'
' >\n'
- ' >\n'
'> -> <\n'
' <\n'
' <\n'
' <\n'
- ' <\n'
' float32[2,1],\n'
' float32[1]\n'
' >,\n'
@@ -102,8 +99,6 @@ class CanonicalFormTest(test.TestCase):
' >\n'
' >,\n'
' <>\n'
- ' >,\n'
- ' <>\n'
'>)')
# pyformat: enable
@@ -140,7 +135,7 @@ class CanonicalFormTest(test.TestCase):
]
server_output_1_arrays = anonymous_tuple.flatten(server_output_1)
state_2 = ip_2.initialize()
- server_state_2, server_output_2, _ = ip_2.next(state_2, [client_data])
+ server_state_2, server_output_2 = ip_2.next(state_2, [client_data])
server_state_2_names = anonymous_tuple.name_list(server_state_2)
server_state_2_arrays = anonymous_tuple.flatten(server_state_2)
server_output_2_names = [
