Fix Ubuntu Sanity CI fail due to pylint error
This commit is contained in:
parent
ed955df943
commit
c227f00a33
@ -39,7 +39,7 @@ class BiasAddDeterministicTest(bias_op_base.BiasAddTestBase,
|
||||
parameterized.TestCase):
|
||||
|
||||
def _makeShapeTuple(self, batch_size, channel_count, data_rank, data_dim,
|
||||
data_layout):
|
||||
data_layout):
|
||||
data_dims = data_rank * (data_dim,)
|
||||
if data_layout == 'channels_first':
|
||||
shape = (batch_size,) + (channel_count,) + data_dims
|
||||
|
Loading…
x
Reference in New Issue
Block a user