From 830954a6a5644581ff6f6547c78d011411f7f7a1 Mon Sep 17 00:00:00 2001 From: erogol Date: Thu, 9 Jul 2020 14:39:38 +0200 Subject: [PATCH] resolve pylint tensorflow issue --- tf/models/tacotron2.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tf/models/tacotron2.py b/tf/models/tacotron2.py index 242e54c1..70d725e2 100644 --- a/tf/models/tacotron2.py +++ b/tf/models/tacotron2.py @@ -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)