Fix Ubuntu Sanity CI fail due to pylint error

This commit is contained in:
Duncan Riach 2019-11-21 17:46:10 -08:00
parent ed955df943
commit c227f00a33

View File

@ -39,7 +39,7 @@ class BiasAddDeterministicTest(bias_op_base.BiasAddTestBase,
parameterized.TestCase): parameterized.TestCase):
def _makeShapeTuple(self, batch_size, channel_count, data_rank, data_dim, def _makeShapeTuple(self, batch_size, channel_count, data_rank, data_dim,
data_layout): data_layout):
data_dims = data_rank * (data_dim,) data_dims = data_rank * (data_dim,)
if data_layout == 'channels_first': if data_layout == 'channels_first':
shape = (batch_size,) + (channel_count,) + data_dims shape = (batch_size,) + (channel_count,) + data_dims