resolve pylint tensorflow issue

This commit is contained in:
erogol 2020-07-09 14:39:38 +02:00
parent 67735cbdac
commit 830954a6a5
1 changed files with 2 additions and 1 deletions

View File

@ -102,6 +102,7 @@ class Tacotron2(keras.models.Model):
return decoder_frames, output_frames, attentions, stop_tokens
def build_inference(self, ):
input_ids = tf.random.uniform([1, 4], 10, tf.int32)
# TODO: issue https://github.com/PyCQA/pylint/issues/3613
input_ids = tf.random.uniform(shape=[1, 4], maxval=10, dtype=tf.int32) #pylint: disable=unexpected-keyword-arg
self(input_ids)