From 4d3914841932065f19a61ad46e178f94dedeff5a Mon Sep 17 00:00:00 2001 From: Stas Bekman Date: Mon, 14 Sep 2020 04:51:19 -0700 Subject: [PATCH] fix deprecation warnings (#7033) * fix deprecation warnings * remove tests/test_tokenization_common.py's test_padding_to_max_length * revert test_padding_to_max_length --- src/transformers/modeling_funnel.py | 2 +- src/transformers/modeling_tf_utils.py | 2 +- tests/test_tokenization_common.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/transformers/modeling_funnel.py b/src/transformers/modeling_funnel.py index fea08e3351..9e4e40c5db 100644 --- a/src/transformers/modeling_funnel.py +++ b/src/transformers/modeling_funnel.py @@ -127,7 +127,7 @@ def load_tf_weights_in_funnel(model, config, tf_checkpoint_path): skipped = False for m_name in name[1:]: if not isinstance(pointer, FunnelPositionwiseFFN) and re.fullmatch(r"layer_\d+", m_name): - layer_index = int(re.search("layer_(\d+)", m_name).groups()[0]) + layer_index = int(re.search(r"layer_(\d+)", m_name).groups()[0]) if layer_index < config.num_hidden_layers: block_idx = 0 while layer_index >= config.block_sizes[block_idx]: diff --git a/src/transformers/modeling_tf_utils.py b/src/transformers/modeling_tf_utils.py index 4ec504810e..278fde5482 100644 --- a/src/transformers/modeling_tf_utils.py +++ b/src/transformers/modeling_tf_utils.py @@ -699,7 +699,7 @@ class TFConv1D(tf.keras.layers.Layer): class TFSharedEmbeddings(tf.keras.layers.Layer): - """ + r""" Construct shared token embeddings. The weights of the embedding layer is usually shared with the weights of the linear decoder when doing diff --git a/tests/test_tokenization_common.py b/tests/test_tokenization_common.py index 26c48b475d..f502428ce8 100644 --- a/tests/test_tokenization_common.py +++ b/tests/test_tokenization_common.py @@ -156,7 +156,7 @@ class TokenizerTesterMixin: tokenizers = self.get_tokenizers() for tokenizer in tokenizers: with self.subTest(f"{tokenizer.__class__.__name__}"): - self.assertNotEqual(tokenizer.max_len, 42) + self.assertNotEqual(tokenizer.model_max_length, 42) # Now let's start the test tokenizers = self.get_tokenizers()