Skip to content

Commit

Permalink
disable arguments-renamed pylint warning in a few places
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 439880661
  • Loading branch information
iansimon authored and Magenta Team committed Apr 6, 2022
1 parent f73ff0c commit 96666ca
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 2 deletions.
4 changes: 2 additions & 2 deletions magenta/contrib/seq2seq.py
Original file line number Diff line number Diff line change
Expand Up @@ -850,15 +850,15 @@ def __init__(self, inputs, sequence_length, sampling_probability,
def initialize(self, name=None):
return super(ScheduledOutputTrainingHelper, self).initialize(name=name)

def sample(self, time, outputs, state, name=None):
def sample(self, time, outputs, state, name=None): # pylint: disable=arguments-renamed
with tf.name_scope(name, "ScheduledOutputTrainingHelperSample",
[time, outputs, state]):
return bernoulli_sample(
probs=self._sampling_probability,
sample_shape=self.batch_size,
seed=self._seed)

def next_inputs(self, time, outputs, state, sample_ids, name=None):
def next_inputs(self, time, outputs, state, sample_ids, name=None): # pylint: disable=arguments-renamed
with tf.name_scope(name, "ScheduledOutputTrainingHelperNextInputs",
[time, outputs, state, sample_ids]):
(finished, base_next_inputs, state) = (
Expand Down
2 changes: 2 additions & 0 deletions magenta/models/score2perf/transformer_autoencoder.py
Original file line number Diff line number Diff line change
Expand Up @@ -395,13 +395,15 @@ def has_input(self):
else:
return True

# pylint: disable=arguments-renamed
def encode(self, perf_inputs, mel_inputs, target_space, hparams,
features=None, losses=None):
"""Encode transformer inputs, but concatenate mel w perf."""
return mel_perf_transformer_encode(
self._encoder_function, perf_inputs, mel_inputs, target_space, hparams,
attention_weights=self.attention_weights,
prepare_encoder_fn=self._prepare_encoder_fn)
# pylint: enable=arguments-renamed

def body(self, features):
"""Transformer main model_fn.
Expand Down

0 comments on commit 96666ca

Please sign in to comment.