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):
|
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
|
||||||
|
Loading…
x
Reference in New Issue
Block a user