From 589f4d7eaa28b76b2ddecb8eef75e79df13ad670 Mon Sep 17 00:00:00 2001 From: Matt Watson Date: Tue, 15 Aug 2023 12:59:17 -0700 Subject: [PATCH] Remove file level docstrings Pet peeve. These are inconsistent, usually just a rephrase of the file itself, and keras-core isn't doing them. Who doesn't like less code? We can keep out class/function docstrings as the place to focus on. --- keras_nlp/backend/config.py | 1 + keras_nlp/conftest.py | 1 + keras_nlp/layers/modeling/cached_multi_head_attention.py | 1 - keras_nlp/layers/modeling/cached_multi_head_attention_test.py | 1 - keras_nlp/layers/modeling/f_net_encoder.py | 3 --- keras_nlp/layers/modeling/f_net_encoder_test.py | 1 - keras_nlp/layers/modeling/masked_lm_head.py | 2 -- keras_nlp/layers/modeling/masked_lm_head_test.py | 1 - keras_nlp/layers/modeling/position_embedding.py | 2 -- keras_nlp/layers/modeling/position_embedding_test.py | 1 - keras_nlp/layers/modeling/reversible_embedding.py | 1 + keras_nlp/layers/modeling/reversible_embedding_test.py | 1 + keras_nlp/layers/modeling/rotary_embedding.py | 1 + keras_nlp/layers/modeling/rotary_embedding_test.py | 1 + keras_nlp/layers/modeling/sine_position_encoding.py | 2 -- keras_nlp/layers/modeling/sine_position_encoding_test.py | 1 - keras_nlp/layers/modeling/token_and_position_embedding.py | 2 -- keras_nlp/layers/modeling/token_and_position_embedding_test.py | 1 - keras_nlp/layers/modeling/transformer_decoder.py | 2 -- keras_nlp/layers/modeling/transformer_decoder_test.py | 1 - keras_nlp/layers/modeling/transformer_encoder.py | 2 -- keras_nlp/layers/modeling/transformer_encoder_test.py | 1 - keras_nlp/layers/modeling/transformer_layer_utils.py | 2 -- keras_nlp/layers/preprocessing/multi_segment_packer_test.py | 1 - keras_nlp/layers/preprocessing/random_deletion.py | 1 + keras_nlp/layers/preprocessing/random_deletion_test.py | 1 - keras_nlp/layers/preprocessing/random_swap.py | 1 + keras_nlp/layers/preprocessing/random_swap_test.py | 1 - keras_nlp/layers/preprocessing/start_end_packer_test.py | 3 --- keras_nlp/metrics/bleu.py | 2 -- keras_nlp/metrics/bleu_test.py | 1 - keras_nlp/metrics/edit_distance.py | 2 -- keras_nlp/metrics/edit_distance_test.py | 1 - keras_nlp/metrics/perplexity.py | 2 -- keras_nlp/metrics/perplexity_test.py | 1 - keras_nlp/metrics/rouge_base.py | 3 --- keras_nlp/metrics/rouge_l.py | 2 -- keras_nlp/metrics/rouge_l_test.py | 1 - keras_nlp/metrics/rouge_n.py | 2 -- keras_nlp/metrics/rouge_n_test.py | 1 - keras_nlp/models/albert/albert_backbone.py | 2 -- keras_nlp/models/albert/albert_backbone_test.py | 1 - keras_nlp/models/albert/albert_classifier.py | 1 - keras_nlp/models/albert/albert_classifier_test.py | 1 - keras_nlp/models/albert/albert_masked_lm.py | 2 -- keras_nlp/models/albert/albert_masked_lm_preprocessor.py | 2 -- keras_nlp/models/albert/albert_masked_lm_preprocessor_test.py | 1 - keras_nlp/models/albert/albert_masked_lm_test.py | 1 - keras_nlp/models/albert/albert_preprocessor.py | 1 - keras_nlp/models/albert/albert_preprocessor_test.py | 1 - keras_nlp/models/albert/albert_presets_test.py | 1 - keras_nlp/models/albert/albert_tokenizer.py | 2 -- keras_nlp/models/albert/albert_tokenizer_test.py | 1 - keras_nlp/models/backbone.py | 2 -- keras_nlp/models/bart/bart_backbone.py | 2 -- keras_nlp/models/bart/bart_backbone_test.py | 1 - keras_nlp/models/bart/bart_preprocessor.py | 1 - keras_nlp/models/bart/bart_preprocessor_test.py | 2 -- keras_nlp/models/bart/bart_presets_test.py | 1 + keras_nlp/models/bart/bart_seq_2_seq_lm.py | 1 - keras_nlp/models/bart/bart_seq_2_seq_lm_preprocessor.py | 2 -- keras_nlp/models/bart/bart_seq_2_seq_lm_preprocessor_test.py | 2 -- keras_nlp/models/bart/bart_seq_2_seq_lm_test.py | 1 - keras_nlp/models/bart/bart_tokenizer.py | 2 -- keras_nlp/models/bart/bart_tokenizer_test.py | 3 --- keras_nlp/models/bert/bert_backbone.py | 2 -- keras_nlp/models/bert/bert_backbone_test.py | 1 - keras_nlp/models/bert/bert_classifier.py | 1 - keras_nlp/models/bert/bert_classifier_test.py | 1 - keras_nlp/models/bert/bert_masked_lm.py | 1 - keras_nlp/models/bert/bert_masked_lm_preprocessor.py | 2 -- keras_nlp/models/bert/bert_masked_lm_preprocessor_test.py | 2 -- keras_nlp/models/bert/bert_masked_lm_test.py | 1 - keras_nlp/models/bert/bert_preprocessor.py | 1 - keras_nlp/models/bert/bert_preprocessor_test.py | 2 -- keras_nlp/models/bert/bert_presets_test.py | 1 - keras_nlp/models/bert/bert_tokenizer.py | 1 - keras_nlp/models/bert/bert_tokenizer_test.py | 2 -- keras_nlp/models/deberta_v3/deberta_v3_backbone.py | 2 -- keras_nlp/models/deberta_v3/deberta_v3_backbone_test.py | 1 - keras_nlp/models/deberta_v3/deberta_v3_classifier.py | 1 - keras_nlp/models/deberta_v3/deberta_v3_classifier_test.py | 1 - keras_nlp/models/deberta_v3/deberta_v3_masked_lm.py | 1 - .../models/deberta_v3/deberta_v3_masked_lm_preprocessor.py | 2 -- .../deberta_v3/deberta_v3_masked_lm_preprocessor_test.py | 1 - keras_nlp/models/deberta_v3/deberta_v3_masked_lm_test.py | 1 - keras_nlp/models/deberta_v3/deberta_v3_preprocessor.py | 1 - keras_nlp/models/deberta_v3/deberta_v3_preprocessor_test.py | 1 - keras_nlp/models/deberta_v3/deberta_v3_presets_test.py | 1 - keras_nlp/models/deberta_v3/deberta_v3_tokenizer.py | 2 -- keras_nlp/models/deberta_v3/deberta_v3_tokenizer_test.py | 1 - keras_nlp/models/deberta_v3/disentangled_attention_encoder.py | 2 -- keras_nlp/models/deberta_v3/disentangled_self_attention.py | 2 -- keras_nlp/models/deberta_v3/relative_embedding.py | 2 -- keras_nlp/models/distil_bert/distil_bert_backbone.py | 2 -- keras_nlp/models/distil_bert/distil_bert_backbone_test.py | 1 - keras_nlp/models/distil_bert/distil_bert_classifier.py | 1 - keras_nlp/models/distil_bert/distil_bert_classifier_test.py | 1 - keras_nlp/models/distil_bert/distil_bert_masked_lm.py | 1 - .../models/distil_bert/distil_bert_masked_lm_preprocessor.py | 2 -- .../distil_bert/distil_bert_masked_lm_preprocessor_test.py | 2 -- keras_nlp/models/distil_bert/distil_bert_masked_lm_test.py | 1 - keras_nlp/models/distil_bert/distil_bert_preprocessor.py | 1 - keras_nlp/models/distil_bert/distil_bert_preprocessor_test.py | 2 -- keras_nlp/models/distil_bert/distil_bert_presets_test.py | 1 - keras_nlp/models/distil_bert/distil_bert_tokenizer.py | 1 - keras_nlp/models/distil_bert/distil_bert_tokenizer_test.py | 2 -- keras_nlp/models/f_net/f_net_backbone.py | 2 -- keras_nlp/models/f_net/f_net_backbone_test.py | 1 - keras_nlp/models/f_net/f_net_classifier.py | 2 -- keras_nlp/models/f_net/f_net_classifier_test.py | 1 - keras_nlp/models/f_net/f_net_masked_lm.py | 1 + keras_nlp/models/f_net/f_net_masked_lm_preprocessor.py | 1 + keras_nlp/models/f_net/f_net_masked_lm_preprocessor_test.py | 1 + keras_nlp/models/f_net/f_net_masked_lm_test.py | 1 + keras_nlp/models/f_net/f_net_preprocessor.py | 1 - keras_nlp/models/f_net/f_net_preprocessor_test.py | 1 - keras_nlp/models/f_net/f_net_presets_test.py | 1 - keras_nlp/models/f_net/f_net_tokenizer.py | 2 -- keras_nlp/models/f_net/f_net_tokenizer_test.py | 1 - keras_nlp/models/generative_task.py | 1 - keras_nlp/models/gpt2/gpt2_backbone.py | 2 -- keras_nlp/models/gpt2/gpt2_backbone_test.py | 1 - keras_nlp/models/gpt2/gpt2_causal_lm.py | 1 - keras_nlp/models/gpt2/gpt2_causal_lm_preprocessor.py | 2 -- keras_nlp/models/gpt2/gpt2_causal_lm_preprocessor_test.py | 2 -- keras_nlp/models/gpt2/gpt2_causal_lm_test.py | 1 - keras_nlp/models/gpt2/gpt2_preprocessor.py | 2 -- keras_nlp/models/gpt2/gpt2_preprocessor_test.py | 2 -- keras_nlp/models/gpt2/gpt2_presets_test.py | 1 - keras_nlp/models/gpt2/gpt2_tokenizer.py | 1 - keras_nlp/models/gpt2/gpt2_tokenizer_test.py | 3 --- keras_nlp/models/gpt_neo_x/gpt_neo_x_backbone.py | 2 -- keras_nlp/models/gpt_neo_x/gpt_neo_x_backbone_test.py | 1 - keras_nlp/models/gpt_neo_x/gpt_neo_x_causal_lm.py | 1 - keras_nlp/models/gpt_neo_x/gpt_neo_x_causal_lm_preprocessor.py | 2 -- .../models/gpt_neo_x/gpt_neo_x_causal_lm_preprocessor_test.py | 2 -- keras_nlp/models/gpt_neo_x/gpt_neo_x_causal_lm_test.py | 1 - keras_nlp/models/gpt_neo_x/gpt_neo_x_decoder.py | 1 + keras_nlp/models/gpt_neo_x/gpt_neo_x_preprocessor.py | 1 - keras_nlp/models/gpt_neo_x/gpt_neo_x_preprocessor_test.py | 3 --- keras_nlp/models/gpt_neo_x/gpt_neo_x_tokenizer.py | 1 - keras_nlp/models/gpt_neo_x/gpt_neo_x_tokenizer_test.py | 3 --- keras_nlp/models/opt/opt_backbone.py | 2 -- keras_nlp/models/opt/opt_backbone_test.py | 1 - keras_nlp/models/opt/opt_causal_lm.py | 1 - keras_nlp/models/opt/opt_causal_lm_preprocessor.py | 2 -- keras_nlp/models/opt/opt_causal_lm_preprocessor_test.py | 2 -- keras_nlp/models/opt/opt_causal_lm_test.py | 1 - keras_nlp/models/opt/opt_preprocessor.py | 2 -- keras_nlp/models/opt/opt_preprocessor_test.py | 2 -- keras_nlp/models/opt/opt_presets_test.py | 1 - keras_nlp/models/opt/opt_tokenizer.py | 2 -- keras_nlp/models/opt/opt_tokenizer_test.py | 3 --- keras_nlp/models/roberta/roberta_backbone.py | 2 -- keras_nlp/models/roberta/roberta_backbone_test.py | 1 - keras_nlp/models/roberta/roberta_classifier.py | 1 - keras_nlp/models/roberta/roberta_classifier_test.py | 1 - keras_nlp/models/roberta/roberta_masked_lm.py | 1 - keras_nlp/models/roberta/roberta_masked_lm_preprocessor.py | 2 -- .../models/roberta/roberta_masked_lm_preprocessor_test.py | 2 -- keras_nlp/models/roberta/roberta_masked_lm_test.py | 1 - keras_nlp/models/roberta/roberta_preprocessor.py | 2 -- keras_nlp/models/roberta/roberta_preprocessor_test.py | 2 -- keras_nlp/models/roberta/roberta_presets_test.py | 1 - keras_nlp/models/roberta/roberta_tokenizer.py | 2 -- keras_nlp/models/roberta/roberta_tokenizer_test.py | 3 --- keras_nlp/models/t5/t5_backbone.py | 2 -- keras_nlp/models/t5/t5_backbone_test.py | 1 - keras_nlp/models/t5/t5_layer_norm.py | 1 + keras_nlp/models/t5/t5_multi_head_attention.py | 1 + keras_nlp/models/t5/t5_tokenizer.py | 2 -- keras_nlp/models/t5/t5_tokenizer_test.py | 1 - keras_nlp/models/t5/t5_transformer_layer.py | 1 + keras_nlp/models/task.py | 1 - keras_nlp/models/task_test.py | 1 + .../models/whisper/whisper_audio_feature_extractor_test.py | 2 -- keras_nlp/models/whisper/whisper_backbone.py | 1 - keras_nlp/models/whisper/whisper_backbone_test.py | 1 - keras_nlp/models/whisper/whisper_decoder.py | 1 - keras_nlp/models/whisper/whisper_encoder.py | 1 - keras_nlp/models/whisper/whisper_preprocessor.py | 1 - keras_nlp/models/whisper/whisper_preprocessor_test.py | 3 --- keras_nlp/models/whisper/whisper_presets_test.py | 1 - keras_nlp/models/whisper/whisper_tokenizer.py | 1 - keras_nlp/models/whisper/whisper_tokenizer_test.py | 2 -- keras_nlp/models/xlm_roberta/xlm_roberta_backbone.py | 2 -- keras_nlp/models/xlm_roberta/xlm_roberta_backbone_test.py | 1 - keras_nlp/models/xlm_roberta/xlm_roberta_classifier.py | 1 - keras_nlp/models/xlm_roberta/xlm_roberta_classifier_test.py | 1 - keras_nlp/models/xlm_roberta/xlm_roberta_masked_lm.py | 1 - .../models/xlm_roberta/xlm_roberta_masked_lm_preprocessor.py | 2 -- .../xlm_roberta/xlm_roberta_masked_lm_preprocessor_test.py | 1 - keras_nlp/models/xlm_roberta/xlm_roberta_masked_lm_test.py | 1 - keras_nlp/models/xlm_roberta/xlm_roberta_preprocessor.py | 2 -- keras_nlp/models/xlm_roberta/xlm_roberta_preprocessor_test.py | 1 - keras_nlp/models/xlm_roberta/xlm_roberta_presets_test.py | 1 - keras_nlp/models/xlm_roberta/xlm_roberta_tokenizer.py | 2 -- keras_nlp/models/xlm_roberta/xlm_roberta_tokenizer_test.py | 1 - keras_nlp/models/xlnet/relative_attention.py | 2 -- keras_nlp/models/xlnet/xlnet_backbone.py | 3 --- keras_nlp/models/xlnet/xlnet_backbone_test.py | 3 --- keras_nlp/models/xlnet/xlnet_content_and_query_embedding.py | 3 --- keras_nlp/models/xlnet/xlnet_encoder.py | 3 --- keras_nlp/samplers/beam_sampler.py | 1 - keras_nlp/samplers/beam_sampler_test.py | 1 - keras_nlp/samplers/contrastive_sampler.py | 1 - keras_nlp/samplers/contrastive_sampler_test.py | 1 - keras_nlp/samplers/greedy_sampler.py | 1 - keras_nlp/samplers/greedy_sampler_test.py | 1 - keras_nlp/samplers/random_sampler.py | 1 - keras_nlp/samplers/random_sampler_test.py | 1 - keras_nlp/samplers/sampler.py | 1 - keras_nlp/samplers/serialization_test.py | 1 - keras_nlp/samplers/top_k_sampler.py | 1 - keras_nlp/samplers/top_k_sampler_test.py | 1 - keras_nlp/samplers/top_p_sampler.py | 1 - keras_nlp/samplers/top_p_sampler_test.py | 1 - keras_nlp/tests/doc_tests/docstring_lib.py | 1 - keras_nlp/tests/doc_tests/fenced_docstring_lib.py | 2 -- keras_nlp/tests/test_case.py | 1 + keras_nlp/tokenizers/byte_pair_tokenizer_test.py | 1 - keras_nlp/tokenizers/byte_tokenizer.py | 2 -- keras_nlp/tokenizers/byte_tokenizer_test.py | 1 - keras_nlp/tokenizers/sentence_piece_tokenizer_trainer.py | 2 +- keras_nlp/tokenizers/sentence_piece_tokenizer_trainer_test.py | 1 - keras_nlp/tokenizers/unicode_codepoint_tokenizer_test.py | 1 - keras_nlp/tokenizers/word_piece_tokenizer_trainer.py | 1 - keras_nlp/tokenizers/word_piece_tokenizer_trainer_test.py | 1 - keras_nlp/utils/keras_utils.py | 1 - keras_nlp/utils/pipeline_model.py | 2 -- keras_nlp/utils/pipeline_model_test.py | 1 + 232 files changed, 21 insertions(+), 313 deletions(-) diff --git a/keras_nlp/backend/config.py b/keras_nlp/backend/config.py index 9e80984cef..578e1746b7 100644 --- a/keras_nlp/backend/config.py +++ b/keras_nlp/backend/config.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + import json import os diff --git a/keras_nlp/conftest.py b/keras_nlp/conftest.py index 3a29b038c5..6bcda7ed0e 100644 --- a/keras_nlp/conftest.py +++ b/keras_nlp/conftest.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + import os import pytest diff --git a/keras_nlp/layers/modeling/cached_multi_head_attention.py b/keras_nlp/layers/modeling/cached_multi_head_attention.py index 3d8f5d6ab7..eb73ec4a25 100644 --- a/keras_nlp/layers/modeling/cached_multi_head_attention.py +++ b/keras_nlp/layers/modeling/cached_multi_head_attention.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Cached MHA layer based on `keras.layers.MultiHeadAttention`.""" from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import keras diff --git a/keras_nlp/layers/modeling/cached_multi_head_attention_test.py b/keras_nlp/layers/modeling/cached_multi_head_attention_test.py index cf23a4ff11..4d19750485 100644 --- a/keras_nlp/layers/modeling/cached_multi_head_attention_test.py +++ b/keras_nlp/layers/modeling/cached_multi_head_attention_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for CachedMultiHeadAttention.""" from keras_nlp.backend import ops from keras_nlp.layers.modeling.cached_multi_head_attention import ( diff --git a/keras_nlp/layers/modeling/f_net_encoder.py b/keras_nlp/layers/modeling/f_net_encoder.py index 806a3dac8c..24b4f18aab 100644 --- a/keras_nlp/layers/modeling/f_net_encoder.py +++ b/keras_nlp/layers/modeling/f_net_encoder.py @@ -12,9 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""FNet encoder block implementation based on `keras.layers.Layer`.""" - - from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import keras from keras_nlp.backend import ops diff --git a/keras_nlp/layers/modeling/f_net_encoder_test.py b/keras_nlp/layers/modeling/f_net_encoder_test.py index bba3cfbae8..d62545166b 100644 --- a/keras_nlp/layers/modeling/f_net_encoder_test.py +++ b/keras_nlp/layers/modeling/f_net_encoder_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for FNet Encoder.""" import os diff --git a/keras_nlp/layers/modeling/masked_lm_head.py b/keras_nlp/layers/modeling/masked_lm_head.py index 46c15f0280..10dbd3eb2c 100644 --- a/keras_nlp/layers/modeling/masked_lm_head.py +++ b/keras_nlp/layers/modeling/masked_lm_head.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Masked Language Model (MaskedLM) head.""" - from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import keras from keras_nlp.backend import ops diff --git a/keras_nlp/layers/modeling/masked_lm_head_test.py b/keras_nlp/layers/modeling/masked_lm_head_test.py index 62532967dc..7f08f29224 100644 --- a/keras_nlp/layers/modeling/masked_lm_head_test.py +++ b/keras_nlp/layers/modeling/masked_lm_head_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for Transformer Encoder.""" import os diff --git a/keras_nlp/layers/modeling/position_embedding.py b/keras_nlp/layers/modeling/position_embedding.py index 3b4dc6387d..6effdc6f0f 100644 --- a/keras_nlp/layers/modeling/position_embedding.py +++ b/keras_nlp/layers/modeling/position_embedding.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Position embedding implementation based on `keras.layers.Layer`.""" - from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import keras from keras_nlp.backend import ops diff --git a/keras_nlp/layers/modeling/position_embedding_test.py b/keras_nlp/layers/modeling/position_embedding_test.py index a953d8147e..f3010d801d 100644 --- a/keras_nlp/layers/modeling/position_embedding_test.py +++ b/keras_nlp/layers/modeling/position_embedding_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for position embedding layer.""" import os diff --git a/keras_nlp/layers/modeling/reversible_embedding.py b/keras_nlp/layers/modeling/reversible_embedding.py index fd9e9e3a01..51de8dfdda 100644 --- a/keras_nlp/layers/modeling/reversible_embedding.py +++ b/keras_nlp/layers/modeling/reversible_embedding.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + import numpy as np from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/layers/modeling/reversible_embedding_test.py b/keras_nlp/layers/modeling/reversible_embedding_test.py index e1de16d185..4dd7026a92 100644 --- a/keras_nlp/layers/modeling/reversible_embedding_test.py +++ b/keras_nlp/layers/modeling/reversible_embedding_test.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + import os import numpy as np diff --git a/keras_nlp/layers/modeling/rotary_embedding.py b/keras_nlp/layers/modeling/rotary_embedding.py index e5dcff1a48..2fbc6bae1f 100644 --- a/keras_nlp/layers/modeling/rotary_embedding.py +++ b/keras_nlp/layers/modeling/rotary_embedding.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import keras from keras_nlp.backend import ops diff --git a/keras_nlp/layers/modeling/rotary_embedding_test.py b/keras_nlp/layers/modeling/rotary_embedding_test.py index 577d4a2447..20fb6562f7 100644 --- a/keras_nlp/layers/modeling/rotary_embedding_test.py +++ b/keras_nlp/layers/modeling/rotary_embedding_test.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + from keras_nlp.backend import keras from keras_nlp.backend import ops from keras_nlp.layers.modeling.rotary_embedding import RotaryEmbedding diff --git a/keras_nlp/layers/modeling/sine_position_encoding.py b/keras_nlp/layers/modeling/sine_position_encoding.py index 8d39e2452b..a5663e3bd3 100644 --- a/keras_nlp/layers/modeling/sine_position_encoding.py +++ b/keras_nlp/layers/modeling/sine_position_encoding.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Sinusoidal position embedding layer.""" - from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import keras from keras_nlp.backend import ops diff --git a/keras_nlp/layers/modeling/sine_position_encoding_test.py b/keras_nlp/layers/modeling/sine_position_encoding_test.py index 7bcc665323..5e76f3fdab 100644 --- a/keras_nlp/layers/modeling/sine_position_encoding_test.py +++ b/keras_nlp/layers/modeling/sine_position_encoding_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for Sinusoidal Positional encoding.""" from keras_nlp.backend import keras from keras_nlp.backend import ops diff --git a/keras_nlp/layers/modeling/token_and_position_embedding.py b/keras_nlp/layers/modeling/token_and_position_embedding.py index aebf909074..86b519b72a 100644 --- a/keras_nlp/layers/modeling/token_and_position_embedding.py +++ b/keras_nlp/layers/modeling/token_and_position_embedding.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Creates an Embedding Layer and adds Positional Embeddings""" - from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import keras from keras_nlp.layers.modeling.position_embedding import PositionEmbedding diff --git a/keras_nlp/layers/modeling/token_and_position_embedding_test.py b/keras_nlp/layers/modeling/token_and_position_embedding_test.py index b85c919500..415d845e93 100644 --- a/keras_nlp/layers/modeling/token_and_position_embedding_test.py +++ b/keras_nlp/layers/modeling/token_and_position_embedding_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for TokenAndPositionEmbedding""" import os diff --git a/keras_nlp/layers/modeling/transformer_decoder.py b/keras_nlp/layers/modeling/transformer_decoder.py index b608f6f482..24d6abc614 100644 --- a/keras_nlp/layers/modeling/transformer_decoder.py +++ b/keras_nlp/layers/modeling/transformer_decoder.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Transformer decoder block implementation based on `keras.layers.Layer`.""" - from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import keras from keras_nlp.backend import ops diff --git a/keras_nlp/layers/modeling/transformer_decoder_test.py b/keras_nlp/layers/modeling/transformer_decoder_test.py index a8bb026d79..e24fd1ab94 100644 --- a/keras_nlp/layers/modeling/transformer_decoder_test.py +++ b/keras_nlp/layers/modeling/transformer_decoder_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for Transformer Decoder.""" import os diff --git a/keras_nlp/layers/modeling/transformer_encoder.py b/keras_nlp/layers/modeling/transformer_encoder.py index 92d933e9e3..678b4215b6 100644 --- a/keras_nlp/layers/modeling/transformer_encoder.py +++ b/keras_nlp/layers/modeling/transformer_encoder.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Transformer encoder block implementation based on `keras.layers.Layer`.""" - from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import keras from keras_nlp.utils.keras_utils import clone_initializer diff --git a/keras_nlp/layers/modeling/transformer_encoder_test.py b/keras_nlp/layers/modeling/transformer_encoder_test.py index 6f880935d6..b8ea452388 100644 --- a/keras_nlp/layers/modeling/transformer_encoder_test.py +++ b/keras_nlp/layers/modeling/transformer_encoder_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for Transformer Encoder.""" import os diff --git a/keras_nlp/layers/modeling/transformer_layer_utils.py b/keras_nlp/layers/modeling/transformer_layer_utils.py index 8a593ae171..863da59a36 100644 --- a/keras_nlp/layers/modeling/transformer_layer_utils.py +++ b/keras_nlp/layers/modeling/transformer_layer_utils.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -""" Utility functions for `TransformerEncoder` and `TransformerDecoder`.""" - from absl import logging from keras_nlp.backend import ops diff --git a/keras_nlp/layers/preprocessing/multi_segment_packer_test.py b/keras_nlp/layers/preprocessing/multi_segment_packer_test.py index 72e30ed72b..6c17495066 100644 --- a/keras_nlp/layers/preprocessing/multi_segment_packer_test.py +++ b/keras_nlp/layers/preprocessing/multi_segment_packer_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for multi-segment packing.""" import numpy as np diff --git a/keras_nlp/layers/preprocessing/random_deletion.py b/keras_nlp/layers/preprocessing/random_deletion.py index 74ede12443..41289c0054 100644 --- a/keras_nlp/layers/preprocessing/random_deletion.py +++ b/keras_nlp/layers/preprocessing/random_deletion.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + import random import tensorflow as tf diff --git a/keras_nlp/layers/preprocessing/random_deletion_test.py b/keras_nlp/layers/preprocessing/random_deletion_test.py index 3328120149..af1e946eb6 100644 --- a/keras_nlp/layers/preprocessing/random_deletion_test.py +++ b/keras_nlp/layers/preprocessing/random_deletion_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for RandomDeletion Layer.""" import tensorflow as tf diff --git a/keras_nlp/layers/preprocessing/random_swap.py b/keras_nlp/layers/preprocessing/random_swap.py index 16d44a573a..ad6f1c0980 100644 --- a/keras_nlp/layers/preprocessing/random_swap.py +++ b/keras_nlp/layers/preprocessing/random_swap.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + import random import tensorflow as tf diff --git a/keras_nlp/layers/preprocessing/random_swap_test.py b/keras_nlp/layers/preprocessing/random_swap_test.py index 544cb5f8a1..8d4efe2a69 100644 --- a/keras_nlp/layers/preprocessing/random_swap_test.py +++ b/keras_nlp/layers/preprocessing/random_swap_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for RandomSwaps Layer.""" import tensorflow as tf diff --git a/keras_nlp/layers/preprocessing/start_end_packer_test.py b/keras_nlp/layers/preprocessing/start_end_packer_test.py index 1d0e98c760..8f70a9a384 100644 --- a/keras_nlp/layers/preprocessing/start_end_packer_test.py +++ b/keras_nlp/layers/preprocessing/start_end_packer_test.py @@ -12,9 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for Start End Packer layer.""" - - import tensorflow as tf from keras_nlp.layers.preprocessing.start_end_packer import StartEndPacker diff --git a/keras_nlp/metrics/bleu.py b/keras_nlp/metrics/bleu.py index 3ee35ddaee..51d224c676 100644 --- a/keras_nlp/metrics/bleu.py +++ b/keras_nlp/metrics/bleu.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""BLEU metric implementation.""" - import collections import math diff --git a/keras_nlp/metrics/bleu_test.py b/keras_nlp/metrics/bleu_test.py index b1d8dc4969..63b8c8b5c0 100644 --- a/keras_nlp/metrics/bleu_test.py +++ b/keras_nlp/metrics/bleu_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for Bleu.""" import pytest import tensorflow as tf diff --git a/keras_nlp/metrics/edit_distance.py b/keras_nlp/metrics/edit_distance.py index fd41bde87b..899f7f1f2e 100644 --- a/keras_nlp/metrics/edit_distance.py +++ b/keras_nlp/metrics/edit_distance.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Edit Distance metric.""" - import tensorflow as tf from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/metrics/edit_distance_test.py b/keras_nlp/metrics/edit_distance_test.py index c463fc55c2..c2051cebbb 100644 --- a/keras_nlp/metrics/edit_distance_test.py +++ b/keras_nlp/metrics/edit_distance_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for EditDistance.""" import pytest import tensorflow as tf diff --git a/keras_nlp/metrics/perplexity.py b/keras_nlp/metrics/perplexity.py index 989d868e1c..eb742fc31c 100644 --- a/keras_nlp/metrics/perplexity.py +++ b/keras_nlp/metrics/perplexity.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Perplexity metric.""" - from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import keras from keras_nlp.backend import ops diff --git a/keras_nlp/metrics/perplexity_test.py b/keras_nlp/metrics/perplexity_test.py index cdf69c9ba6..7b54e8d260 100644 --- a/keras_nlp/metrics/perplexity_test.py +++ b/keras_nlp/metrics/perplexity_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for Perplexity.""" from keras_nlp.backend import ops from keras_nlp.metrics.perplexity import Perplexity from keras_nlp.tests.test_case import TestCase diff --git a/keras_nlp/metrics/rouge_base.py b/keras_nlp/metrics/rouge_base.py index deec9536ac..f84e718080 100644 --- a/keras_nlp/metrics/rouge_base.py +++ b/keras_nlp/metrics/rouge_base.py @@ -12,9 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""ROUGE metric.""" - - import tensorflow as tf from keras_nlp.backend import keras diff --git a/keras_nlp/metrics/rouge_l.py b/keras_nlp/metrics/rouge_l.py index 216cd6ed2a..82fcdedade 100644 --- a/keras_nlp/metrics/rouge_l.py +++ b/keras_nlp/metrics/rouge_l.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""ROUGE-L metric.""" - from keras_nlp.api_export import keras_nlp_export from keras_nlp.metrics.rouge_base import RougeBase diff --git a/keras_nlp/metrics/rouge_l_test.py b/keras_nlp/metrics/rouge_l_test.py index 4b1206cb4e..edf729dca7 100644 --- a/keras_nlp/metrics/rouge_l_test.py +++ b/keras_nlp/metrics/rouge_l_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for RougeL.""" import tensorflow as tf from keras_nlp.metrics.rouge_l import RougeL diff --git a/keras_nlp/metrics/rouge_n.py b/keras_nlp/metrics/rouge_n.py index 77bfd7021e..8b135d1dd0 100644 --- a/keras_nlp/metrics/rouge_n.py +++ b/keras_nlp/metrics/rouge_n.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""ROUGE-N metric.""" - from keras_nlp.api_export import keras_nlp_export from keras_nlp.metrics.rouge_base import RougeBase diff --git a/keras_nlp/metrics/rouge_n_test.py b/keras_nlp/metrics/rouge_n_test.py index 21effb0be8..74c86dbe75 100644 --- a/keras_nlp/metrics/rouge_n_test.py +++ b/keras_nlp/metrics/rouge_n_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for RougeN.""" import pytest import tensorflow as tf diff --git a/keras_nlp/models/albert/albert_backbone.py b/keras_nlp/models/albert/albert_backbone.py index 45da2d3086..51da1f49a7 100644 --- a/keras_nlp/models/albert/albert_backbone.py +++ b/keras_nlp/models/albert/albert_backbone.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""ALBERT backbone model.""" - import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/albert/albert_backbone_test.py b/keras_nlp/models/albert/albert_backbone_test.py index eb2d513d99..5f93c5dc12 100644 --- a/keras_nlp/models/albert/albert_backbone_test.py +++ b/keras_nlp/models/albert/albert_backbone_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Test for ALBERT backbone model.""" import os diff --git a/keras_nlp/models/albert/albert_classifier.py b/keras_nlp/models/albert/albert_classifier.py index e884d05d88..8cadb03b49 100644 --- a/keras_nlp/models/albert/albert_classifier.py +++ b/keras_nlp/models/albert/albert_classifier.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""ALBERT classification model.""" import copy diff --git a/keras_nlp/models/albert/albert_classifier_test.py b/keras_nlp/models/albert/albert_classifier_test.py index eddfd05204..71fef68ce1 100644 --- a/keras_nlp/models/albert/albert_classifier_test.py +++ b/keras_nlp/models/albert/albert_classifier_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for ALBERT classification model.""" import io import os diff --git a/keras_nlp/models/albert/albert_masked_lm.py b/keras_nlp/models/albert/albert_masked_lm.py index 8084340aa9..db483bb623 100644 --- a/keras_nlp/models/albert/albert_masked_lm.py +++ b/keras_nlp/models/albert/albert_masked_lm.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""ALBERT masked LM model.""" - import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/albert/albert_masked_lm_preprocessor.py b/keras_nlp/models/albert/albert_masked_lm_preprocessor.py index cd875acb24..9a52d28a2c 100644 --- a/keras_nlp/models/albert/albert_masked_lm_preprocessor.py +++ b/keras_nlp/models/albert/albert_masked_lm_preprocessor.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""ALBERT masked language model preprocessor layer.""" - from absl import logging from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/albert/albert_masked_lm_preprocessor_test.py b/keras_nlp/models/albert/albert_masked_lm_preprocessor_test.py index 68bf0e60bf..d95b072108 100644 --- a/keras_nlp/models/albert/albert_masked_lm_preprocessor_test.py +++ b/keras_nlp/models/albert/albert_masked_lm_preprocessor_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for ALBERT masked language model preprocessor layer.""" import io diff --git a/keras_nlp/models/albert/albert_masked_lm_test.py b/keras_nlp/models/albert/albert_masked_lm_test.py index 096c071767..9d9ea2478d 100644 --- a/keras_nlp/models/albert/albert_masked_lm_test.py +++ b/keras_nlp/models/albert/albert_masked_lm_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for ALBERT masked language model.""" import io import os diff --git a/keras_nlp/models/albert/albert_preprocessor.py b/keras_nlp/models/albert/albert_preprocessor.py index ef7a717e71..4849aab392 100644 --- a/keras_nlp/models/albert/albert_preprocessor.py +++ b/keras_nlp/models/albert/albert_preprocessor.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""ALBERT preprocessor layer.""" import copy diff --git a/keras_nlp/models/albert/albert_preprocessor_test.py b/keras_nlp/models/albert/albert_preprocessor_test.py index f2f9326288..2830b85073 100644 --- a/keras_nlp/models/albert/albert_preprocessor_test.py +++ b/keras_nlp/models/albert/albert_preprocessor_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for ALBERT preprocessor layer.""" import io import sentencepiece diff --git a/keras_nlp/models/albert/albert_presets_test.py b/keras_nlp/models/albert/albert_presets_test.py index 09904246a2..86265a1abd 100644 --- a/keras_nlp/models/albert/albert_presets_test.py +++ b/keras_nlp/models/albert/albert_presets_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for loading pretrained model presets.""" import pytest from absl.testing import parameterized diff --git a/keras_nlp/models/albert/albert_tokenizer.py b/keras_nlp/models/albert/albert_tokenizer.py index ab60583889..1b85be3a99 100644 --- a/keras_nlp/models/albert/albert_tokenizer.py +++ b/keras_nlp/models/albert/albert_tokenizer.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""ALBERT tokenizer.""" - import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/albert/albert_tokenizer_test.py b/keras_nlp/models/albert/albert_tokenizer_test.py index 75f374b903..fb7c145518 100644 --- a/keras_nlp/models/albert/albert_tokenizer_test.py +++ b/keras_nlp/models/albert/albert_tokenizer_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for ALBERT tokenizer.""" import io import sentencepiece diff --git a/keras_nlp/models/backbone.py b/keras_nlp/models/backbone.py index 26cd4bf86c..7fafab514c 100644 --- a/keras_nlp/models/backbone.py +++ b/keras_nlp/models/backbone.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Base class for Backbone models.""" - import os from keras_nlp.backend import keras diff --git a/keras_nlp/models/bart/bart_backbone.py b/keras_nlp/models/bart/bart_backbone.py index ae26a9196e..203d3ab2d7 100644 --- a/keras_nlp/models/bart/bart_backbone.py +++ b/keras_nlp/models/bart/bart_backbone.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""BART backbone model.""" - import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/bart/bart_backbone_test.py b/keras_nlp/models/bart/bart_backbone_test.py index 35923fcdd1..feb643a4de 100644 --- a/keras_nlp/models/bart/bart_backbone_test.py +++ b/keras_nlp/models/bart/bart_backbone_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Test for BART backbone models.""" import os diff --git a/keras_nlp/models/bart/bart_preprocessor.py b/keras_nlp/models/bart/bart_preprocessor.py index 04626ddb9c..e3eca171fe 100644 --- a/keras_nlp/models/bart/bart_preprocessor.py +++ b/keras_nlp/models/bart/bart_preprocessor.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""BART preprocessor layer.""" import copy diff --git a/keras_nlp/models/bart/bart_preprocessor_test.py b/keras_nlp/models/bart/bart_preprocessor_test.py index df569c5de1..f6bc96d52f 100644 --- a/keras_nlp/models/bart/bart_preprocessor_test.py +++ b/keras_nlp/models/bart/bart_preprocessor_test.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for BART preprocessor layer.""" - import tensorflow as tf from keras_nlp.backend import keras diff --git a/keras_nlp/models/bart/bart_presets_test.py b/keras_nlp/models/bart/bart_presets_test.py index 324e840178..89518d845b 100644 --- a/keras_nlp/models/bart/bart_presets_test.py +++ b/keras_nlp/models/bart/bart_presets_test.py @@ -13,6 +13,7 @@ # limitations under the License. # Copyright 2023 The KerasNLP Authors # + from keras_nlp.backend import ops from keras_nlp.tests.test_case import TestCase diff --git a/keras_nlp/models/bart/bart_seq_2_seq_lm.py b/keras_nlp/models/bart/bart_seq_2_seq_lm.py index a4f98d213a..2131519ce3 100644 --- a/keras_nlp/models/bart/bart_seq_2_seq_lm.py +++ b/keras_nlp/models/bart/bart_seq_2_seq_lm.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""BART Seq2Seq LM (Language Model).""" import copy diff --git a/keras_nlp/models/bart/bart_seq_2_seq_lm_preprocessor.py b/keras_nlp/models/bart/bart_seq_2_seq_lm_preprocessor.py index 0eb9a6cb44..beab96ef24 100644 --- a/keras_nlp/models/bart/bart_seq_2_seq_lm_preprocessor.py +++ b/keras_nlp/models/bart/bart_seq_2_seq_lm_preprocessor.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""BART Seq2Seq LM preprocessor layer.""" - import copy import tensorflow as tf diff --git a/keras_nlp/models/bart/bart_seq_2_seq_lm_preprocessor_test.py b/keras_nlp/models/bart/bart_seq_2_seq_lm_preprocessor_test.py index afaceac0ed..41a3c3dab2 100644 --- a/keras_nlp/models/bart/bart_seq_2_seq_lm_preprocessor_test.py +++ b/keras_nlp/models/bart/bart_seq_2_seq_lm_preprocessor_test.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for BART preprocessor layer.""" - import tensorflow as tf from keras_nlp.backend import keras diff --git a/keras_nlp/models/bart/bart_seq_2_seq_lm_test.py b/keras_nlp/models/bart/bart_seq_2_seq_lm_test.py index a3b9db4bd0..7c72580daf 100644 --- a/keras_nlp/models/bart/bart_seq_2_seq_lm_test.py +++ b/keras_nlp/models/bart/bart_seq_2_seq_lm_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for BART causal LM model.""" import os from unittest.mock import patch diff --git a/keras_nlp/models/bart/bart_tokenizer.py b/keras_nlp/models/bart/bart_tokenizer.py index 0737ef03b3..0f8728d7dd 100644 --- a/keras_nlp/models/bart/bart_tokenizer.py +++ b/keras_nlp/models/bart/bart_tokenizer.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""BART tokenizer.""" - import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/bart/bart_tokenizer_test.py b/keras_nlp/models/bart/bart_tokenizer_test.py index 1c86b7c81d..d3e77958e8 100644 --- a/keras_nlp/models/bart/bart_tokenizer_test.py +++ b/keras_nlp/models/bart/bart_tokenizer_test.py @@ -12,9 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for BART tokenizer.""" - - from keras_nlp.backend import keras from keras_nlp.models.bart.bart_tokenizer import BartTokenizer from keras_nlp.tests.test_case import TestCase diff --git a/keras_nlp/models/bert/bert_backbone.py b/keras_nlp/models/bert/bert_backbone.py index 155da97205..381f1f8cb4 100644 --- a/keras_nlp/models/bert/bert_backbone.py +++ b/keras_nlp/models/bert/bert_backbone.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""BERT backbone model.""" - import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/bert/bert_backbone_test.py b/keras_nlp/models/bert/bert_backbone_test.py index da5f6c4e5d..6374a6848f 100644 --- a/keras_nlp/models/bert/bert_backbone_test.py +++ b/keras_nlp/models/bert/bert_backbone_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Test for BERT backbone models.""" import os diff --git a/keras_nlp/models/bert/bert_classifier.py b/keras_nlp/models/bert/bert_classifier.py index 719dab5dfc..b6aa3f1898 100644 --- a/keras_nlp/models/bert/bert_classifier.py +++ b/keras_nlp/models/bert/bert_classifier.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""BERT classification model.""" import copy diff --git a/keras_nlp/models/bert/bert_classifier_test.py b/keras_nlp/models/bert/bert_classifier_test.py index a5dd9fefa8..92122335b5 100644 --- a/keras_nlp/models/bert/bert_classifier_test.py +++ b/keras_nlp/models/bert/bert_classifier_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for BERT classification model.""" import os diff --git a/keras_nlp/models/bert/bert_masked_lm.py b/keras_nlp/models/bert/bert_masked_lm.py index d696f77fdd..c06257e8d1 100644 --- a/keras_nlp/models/bert/bert_masked_lm.py +++ b/keras_nlp/models/bert/bert_masked_lm.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""BERT masked LM model.""" import copy diff --git a/keras_nlp/models/bert/bert_masked_lm_preprocessor.py b/keras_nlp/models/bert/bert_masked_lm_preprocessor.py index 4b79c65078..685c65c70b 100644 --- a/keras_nlp/models/bert/bert_masked_lm_preprocessor.py +++ b/keras_nlp/models/bert/bert_masked_lm_preprocessor.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""BERT masked language model preprocessor layer.""" - from absl import logging from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/bert/bert_masked_lm_preprocessor_test.py b/keras_nlp/models/bert/bert_masked_lm_preprocessor_test.py index 205fa965bf..ad208c24a3 100644 --- a/keras_nlp/models/bert/bert_masked_lm_preprocessor_test.py +++ b/keras_nlp/models/bert/bert_masked_lm_preprocessor_test.py @@ -11,8 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for BERT masked language model preprocessor layer.""" - import tensorflow as tf diff --git a/keras_nlp/models/bert/bert_masked_lm_test.py b/keras_nlp/models/bert/bert_masked_lm_test.py index 316c0b0989..8659de2474 100644 --- a/keras_nlp/models/bert/bert_masked_lm_test.py +++ b/keras_nlp/models/bert/bert_masked_lm_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for BERT masked language model.""" import os diff --git a/keras_nlp/models/bert/bert_preprocessor.py b/keras_nlp/models/bert/bert_preprocessor.py index fe84d8f62e..214193753d 100644 --- a/keras_nlp/models/bert/bert_preprocessor.py +++ b/keras_nlp/models/bert/bert_preprocessor.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""BERT preprocessor layer.""" import copy diff --git a/keras_nlp/models/bert/bert_preprocessor_test.py b/keras_nlp/models/bert/bert_preprocessor_test.py index 954646c93f..efedb2d550 100644 --- a/keras_nlp/models/bert/bert_preprocessor_test.py +++ b/keras_nlp/models/bert/bert_preprocessor_test.py @@ -11,8 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for BERT preprocessor layer.""" - import tensorflow as tf diff --git a/keras_nlp/models/bert/bert_presets_test.py b/keras_nlp/models/bert/bert_presets_test.py index 6728a7ded5..71e739bbfc 100644 --- a/keras_nlp/models/bert/bert_presets_test.py +++ b/keras_nlp/models/bert/bert_presets_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for loading pretrained model presets.""" import pytest from absl.testing import parameterized diff --git a/keras_nlp/models/bert/bert_tokenizer.py b/keras_nlp/models/bert/bert_tokenizer.py index 2f6eb2c28d..5c01bf24e1 100644 --- a/keras_nlp/models/bert/bert_tokenizer.py +++ b/keras_nlp/models/bert/bert_tokenizer.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""BERT tokenizer.""" import copy diff --git a/keras_nlp/models/bert/bert_tokenizer_test.py b/keras_nlp/models/bert/bert_tokenizer_test.py index a67a64d629..2cd1baa490 100644 --- a/keras_nlp/models/bert/bert_tokenizer_test.py +++ b/keras_nlp/models/bert/bert_tokenizer_test.py @@ -11,8 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for BERT tokenizer.""" - from keras_nlp.backend import keras from keras_nlp.models.bert.bert_tokenizer import BertTokenizer diff --git a/keras_nlp/models/deberta_v3/deberta_v3_backbone.py b/keras_nlp/models/deberta_v3/deberta_v3_backbone.py index 43a5a34c0a..b963d35433 100644 --- a/keras_nlp/models/deberta_v3/deberta_v3_backbone.py +++ b/keras_nlp/models/deberta_v3/deberta_v3_backbone.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""DeBERTa backbone model.""" - import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/deberta_v3/deberta_v3_backbone_test.py b/keras_nlp/models/deberta_v3/deberta_v3_backbone_test.py index 58ffe65d6a..cd38d63f48 100644 --- a/keras_nlp/models/deberta_v3/deberta_v3_backbone_test.py +++ b/keras_nlp/models/deberta_v3/deberta_v3_backbone_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Test for DeBERTa backbone models.""" import os diff --git a/keras_nlp/models/deberta_v3/deberta_v3_classifier.py b/keras_nlp/models/deberta_v3/deberta_v3_classifier.py index dc2e8b1db8..5012e3cd08 100644 --- a/keras_nlp/models/deberta_v3/deberta_v3_classifier.py +++ b/keras_nlp/models/deberta_v3/deberta_v3_classifier.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""DeBERTa classification model.""" import copy diff --git a/keras_nlp/models/deberta_v3/deberta_v3_classifier_test.py b/keras_nlp/models/deberta_v3/deberta_v3_classifier_test.py index 3e5cbd0abe..c53b0fd143 100644 --- a/keras_nlp/models/deberta_v3/deberta_v3_classifier_test.py +++ b/keras_nlp/models/deberta_v3/deberta_v3_classifier_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for DeBERTa classification model.""" import io import os diff --git a/keras_nlp/models/deberta_v3/deberta_v3_masked_lm.py b/keras_nlp/models/deberta_v3/deberta_v3_masked_lm.py index 2fcb8bf4d3..81f15ceda7 100644 --- a/keras_nlp/models/deberta_v3/deberta_v3_masked_lm.py +++ b/keras_nlp/models/deberta_v3/deberta_v3_masked_lm.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""DeBERTaV3 masked lm model.""" import copy diff --git a/keras_nlp/models/deberta_v3/deberta_v3_masked_lm_preprocessor.py b/keras_nlp/models/deberta_v3/deberta_v3_masked_lm_preprocessor.py index 9de16e9cb3..1644c13823 100644 --- a/keras_nlp/models/deberta_v3/deberta_v3_masked_lm_preprocessor.py +++ b/keras_nlp/models/deberta_v3/deberta_v3_masked_lm_preprocessor.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""DeBERTa masked language model preprocessor layer.""" - from absl import logging from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/deberta_v3/deberta_v3_masked_lm_preprocessor_test.py b/keras_nlp/models/deberta_v3/deberta_v3_masked_lm_preprocessor_test.py index 78454511a0..3c1f671297 100644 --- a/keras_nlp/models/deberta_v3/deberta_v3_masked_lm_preprocessor_test.py +++ b/keras_nlp/models/deberta_v3/deberta_v3_masked_lm_preprocessor_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for DeBERTa preprocessor layer.""" import io import sentencepiece diff --git a/keras_nlp/models/deberta_v3/deberta_v3_masked_lm_test.py b/keras_nlp/models/deberta_v3/deberta_v3_masked_lm_test.py index 5394295fe7..3dead2c80d 100644 --- a/keras_nlp/models/deberta_v3/deberta_v3_masked_lm_test.py +++ b/keras_nlp/models/deberta_v3/deberta_v3_masked_lm_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for DeBERTa masked language model.""" import io import os diff --git a/keras_nlp/models/deberta_v3/deberta_v3_preprocessor.py b/keras_nlp/models/deberta_v3/deberta_v3_preprocessor.py index 0996ad9405..dee91dcffa 100644 --- a/keras_nlp/models/deberta_v3/deberta_v3_preprocessor.py +++ b/keras_nlp/models/deberta_v3/deberta_v3_preprocessor.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""DeBERTa preprocessor layer.""" import copy diff --git a/keras_nlp/models/deberta_v3/deberta_v3_preprocessor_test.py b/keras_nlp/models/deberta_v3/deberta_v3_preprocessor_test.py index 70ef8f4ad7..1e95e7988c 100644 --- a/keras_nlp/models/deberta_v3/deberta_v3_preprocessor_test.py +++ b/keras_nlp/models/deberta_v3/deberta_v3_preprocessor_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for DeBERTa preprocessor layer.""" import io import sentencepiece diff --git a/keras_nlp/models/deberta_v3/deberta_v3_presets_test.py b/keras_nlp/models/deberta_v3/deberta_v3_presets_test.py index 51298b0250..a033825dad 100644 --- a/keras_nlp/models/deberta_v3/deberta_v3_presets_test.py +++ b/keras_nlp/models/deberta_v3/deberta_v3_presets_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for loading pretrained model presets.""" import pytest from absl.testing import parameterized diff --git a/keras_nlp/models/deberta_v3/deberta_v3_tokenizer.py b/keras_nlp/models/deberta_v3/deberta_v3_tokenizer.py index f04b78aa0f..03c9cd5821 100644 --- a/keras_nlp/models/deberta_v3/deberta_v3_tokenizer.py +++ b/keras_nlp/models/deberta_v3/deberta_v3_tokenizer.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""DeBERTa tokenizer.""" - import copy import tensorflow as tf diff --git a/keras_nlp/models/deberta_v3/deberta_v3_tokenizer_test.py b/keras_nlp/models/deberta_v3/deberta_v3_tokenizer_test.py index 720a30dabf..d9a0708b9d 100644 --- a/keras_nlp/models/deberta_v3/deberta_v3_tokenizer_test.py +++ b/keras_nlp/models/deberta_v3/deberta_v3_tokenizer_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for DeBERTa tokenizer.""" import io import sentencepiece diff --git a/keras_nlp/models/deberta_v3/disentangled_attention_encoder.py b/keras_nlp/models/deberta_v3/disentangled_attention_encoder.py index 594e7188ca..38e9d2623b 100644 --- a/keras_nlp/models/deberta_v3/disentangled_attention_encoder.py +++ b/keras_nlp/models/deberta_v3/disentangled_attention_encoder.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Disentangled attention encoder block implementation based on `keras.layers.Layer`.""" - from keras_nlp.backend import keras from keras_nlp.models.deberta_v3.disentangled_self_attention import ( DisentangledSelfAttention, diff --git a/keras_nlp/models/deberta_v3/disentangled_self_attention.py b/keras_nlp/models/deberta_v3/disentangled_self_attention.py index 62a504d077..421ba91d8e 100644 --- a/keras_nlp/models/deberta_v3/disentangled_self_attention.py +++ b/keras_nlp/models/deberta_v3/disentangled_self_attention.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Disentangled self-attention layer.""" - import math from keras_nlp.backend import keras diff --git a/keras_nlp/models/deberta_v3/relative_embedding.py b/keras_nlp/models/deberta_v3/relative_embedding.py index 6c08424f3c..6ae29a5fd7 100644 --- a/keras_nlp/models/deberta_v3/relative_embedding.py +++ b/keras_nlp/models/deberta_v3/relative_embedding.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Relative embedding layer.""" - from keras_nlp.backend import keras from keras_nlp.backend import ops diff --git a/keras_nlp/models/distil_bert/distil_bert_backbone.py b/keras_nlp/models/distil_bert/distil_bert_backbone.py index a00b15da22..a3634215fa 100644 --- a/keras_nlp/models/distil_bert/distil_bert_backbone.py +++ b/keras_nlp/models/distil_bert/distil_bert_backbone.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""DistilBERT backbone model.""" - import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/distil_bert/distil_bert_backbone_test.py b/keras_nlp/models/distil_bert/distil_bert_backbone_test.py index c713c3759d..897e7ffc26 100644 --- a/keras_nlp/models/distil_bert/distil_bert_backbone_test.py +++ b/keras_nlp/models/distil_bert/distil_bert_backbone_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Test for DistilBERT backbone models.""" import os diff --git a/keras_nlp/models/distil_bert/distil_bert_classifier.py b/keras_nlp/models/distil_bert/distil_bert_classifier.py index 22a68dda8a..f8ae318538 100644 --- a/keras_nlp/models/distil_bert/distil_bert_classifier.py +++ b/keras_nlp/models/distil_bert/distil_bert_classifier.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""DistilBERT classification model.""" import copy diff --git a/keras_nlp/models/distil_bert/distil_bert_classifier_test.py b/keras_nlp/models/distil_bert/distil_bert_classifier_test.py index 1319849e0b..6ca36d1692 100644 --- a/keras_nlp/models/distil_bert/distil_bert_classifier_test.py +++ b/keras_nlp/models/distil_bert/distil_bert_classifier_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for DistilBERT classification model.""" import os diff --git a/keras_nlp/models/distil_bert/distil_bert_masked_lm.py b/keras_nlp/models/distil_bert/distil_bert_masked_lm.py index 0f5299a589..13d0cf5f45 100644 --- a/keras_nlp/models/distil_bert/distil_bert_masked_lm.py +++ b/keras_nlp/models/distil_bert/distil_bert_masked_lm.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""DistilBERT masked lm model.""" import copy diff --git a/keras_nlp/models/distil_bert/distil_bert_masked_lm_preprocessor.py b/keras_nlp/models/distil_bert/distil_bert_masked_lm_preprocessor.py index 9d4774b324..3fcf9bced1 100644 --- a/keras_nlp/models/distil_bert/distil_bert_masked_lm_preprocessor.py +++ b/keras_nlp/models/distil_bert/distil_bert_masked_lm_preprocessor.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""DistilBERT masked language model preprocessor layer.""" - from absl import logging from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/distil_bert/distil_bert_masked_lm_preprocessor_test.py b/keras_nlp/models/distil_bert/distil_bert_masked_lm_preprocessor_test.py index 614ee153ea..091ae77262 100644 --- a/keras_nlp/models/distil_bert/distil_bert_masked_lm_preprocessor_test.py +++ b/keras_nlp/models/distil_bert/distil_bert_masked_lm_preprocessor_test.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for DistilBERT masked language model preprocessor layer.""" - import tensorflow as tf from keras_nlp.backend import keras diff --git a/keras_nlp/models/distil_bert/distil_bert_masked_lm_test.py b/keras_nlp/models/distil_bert/distil_bert_masked_lm_test.py index 50c36dd6a7..e43feb32f1 100644 --- a/keras_nlp/models/distil_bert/distil_bert_masked_lm_test.py +++ b/keras_nlp/models/distil_bert/distil_bert_masked_lm_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for DistilBERT masked language model.""" import os diff --git a/keras_nlp/models/distil_bert/distil_bert_preprocessor.py b/keras_nlp/models/distil_bert/distil_bert_preprocessor.py index 1d62ca448f..f2c4326234 100644 --- a/keras_nlp/models/distil_bert/distil_bert_preprocessor.py +++ b/keras_nlp/models/distil_bert/distil_bert_preprocessor.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""DistilBERT preprocessor layers.""" import copy diff --git a/keras_nlp/models/distil_bert/distil_bert_preprocessor_test.py b/keras_nlp/models/distil_bert/distil_bert_preprocessor_test.py index fc5421bb6a..77176e4c15 100644 --- a/keras_nlp/models/distil_bert/distil_bert_preprocessor_test.py +++ b/keras_nlp/models/distil_bert/distil_bert_preprocessor_test.py @@ -11,8 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for DistilBERT preprocessor layer.""" - import tensorflow as tf diff --git a/keras_nlp/models/distil_bert/distil_bert_presets_test.py b/keras_nlp/models/distil_bert/distil_bert_presets_test.py index fcef57e0c0..a974f54aee 100644 --- a/keras_nlp/models/distil_bert/distil_bert_presets_test.py +++ b/keras_nlp/models/distil_bert/distil_bert_presets_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for loading pretrained model presets.""" import pytest from absl.testing import parameterized diff --git a/keras_nlp/models/distil_bert/distil_bert_tokenizer.py b/keras_nlp/models/distil_bert/distil_bert_tokenizer.py index 450681d540..9e344a378b 100644 --- a/keras_nlp/models/distil_bert/distil_bert_tokenizer.py +++ b/keras_nlp/models/distil_bert/distil_bert_tokenizer.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""DistilBERT tokenizer.""" import copy diff --git a/keras_nlp/models/distil_bert/distil_bert_tokenizer_test.py b/keras_nlp/models/distil_bert/distil_bert_tokenizer_test.py index f658c98348..db85435f1c 100644 --- a/keras_nlp/models/distil_bert/distil_bert_tokenizer_test.py +++ b/keras_nlp/models/distil_bert/distil_bert_tokenizer_test.py @@ -11,8 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for DistilBERT tokenizer.""" - from keras_nlp.backend import keras from keras_nlp.models.distil_bert.distil_bert_tokenizer import ( diff --git a/keras_nlp/models/f_net/f_net_backbone.py b/keras_nlp/models/f_net/f_net_backbone.py index 5053a55d87..a2e6e8ce95 100644 --- a/keras_nlp/models/f_net/f_net_backbone.py +++ b/keras_nlp/models/f_net/f_net_backbone.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""FNet backbone model.""" - import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/f_net/f_net_backbone_test.py b/keras_nlp/models/f_net/f_net_backbone_test.py index 8d1f6030de..39236fa5f1 100644 --- a/keras_nlp/models/f_net/f_net_backbone_test.py +++ b/keras_nlp/models/f_net/f_net_backbone_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Test for FNet backbone model.""" import os diff --git a/keras_nlp/models/f_net/f_net_classifier.py b/keras_nlp/models/f_net/f_net_classifier.py index 1a84a83f48..838b43197f 100644 --- a/keras_nlp/models/f_net/f_net_classifier.py +++ b/keras_nlp/models/f_net/f_net_classifier.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""FNet classification model.""" - import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/f_net/f_net_classifier_test.py b/keras_nlp/models/f_net/f_net_classifier_test.py index f30e043da1..66ec838470 100644 --- a/keras_nlp/models/f_net/f_net_classifier_test.py +++ b/keras_nlp/models/f_net/f_net_classifier_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for FNet classification model.""" import io import os diff --git a/keras_nlp/models/f_net/f_net_masked_lm.py b/keras_nlp/models/f_net/f_net_masked_lm.py index 0355a235f2..2dab73323b 100644 --- a/keras_nlp/models/f_net/f_net_masked_lm.py +++ b/keras_nlp/models/f_net/f_net_masked_lm.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/f_net/f_net_masked_lm_preprocessor.py b/keras_nlp/models/f_net/f_net_masked_lm_preprocessor.py index 003d242ab4..87fa1a316d 100644 --- a/keras_nlp/models/f_net/f_net_masked_lm_preprocessor.py +++ b/keras_nlp/models/f_net/f_net_masked_lm_preprocessor.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + from absl import logging from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/f_net/f_net_masked_lm_preprocessor_test.py b/keras_nlp/models/f_net/f_net_masked_lm_preprocessor_test.py index d87999580f..d926f66566 100644 --- a/keras_nlp/models/f_net/f_net_masked_lm_preprocessor_test.py +++ b/keras_nlp/models/f_net/f_net_masked_lm_preprocessor_test.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + import io import sentencepiece diff --git a/keras_nlp/models/f_net/f_net_masked_lm_test.py b/keras_nlp/models/f_net/f_net_masked_lm_test.py index 1ae14d0dcd..d4e9e548f1 100644 --- a/keras_nlp/models/f_net/f_net_masked_lm_test.py +++ b/keras_nlp/models/f_net/f_net_masked_lm_test.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + import io import os diff --git a/keras_nlp/models/f_net/f_net_preprocessor.py b/keras_nlp/models/f_net/f_net_preprocessor.py index 3a31a5ed8b..5ebd5d1645 100644 --- a/keras_nlp/models/f_net/f_net_preprocessor.py +++ b/keras_nlp/models/f_net/f_net_preprocessor.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""FNet preprocessor layer.""" import copy diff --git a/keras_nlp/models/f_net/f_net_preprocessor_test.py b/keras_nlp/models/f_net/f_net_preprocessor_test.py index 31a4e8b3b3..638df89123 100644 --- a/keras_nlp/models/f_net/f_net_preprocessor_test.py +++ b/keras_nlp/models/f_net/f_net_preprocessor_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for FNet preprocessor layer.""" import io import sentencepiece diff --git a/keras_nlp/models/f_net/f_net_presets_test.py b/keras_nlp/models/f_net/f_net_presets_test.py index 3eedbbf7de..1f8f79c447 100644 --- a/keras_nlp/models/f_net/f_net_presets_test.py +++ b/keras_nlp/models/f_net/f_net_presets_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for loading pretrained model presets.""" import pytest from absl.testing import parameterized diff --git a/keras_nlp/models/f_net/f_net_tokenizer.py b/keras_nlp/models/f_net/f_net_tokenizer.py index 29d5b3913b..294e02e4db 100644 --- a/keras_nlp/models/f_net/f_net_tokenizer.py +++ b/keras_nlp/models/f_net/f_net_tokenizer.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""FNet tokenizer.""" - import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/f_net/f_net_tokenizer_test.py b/keras_nlp/models/f_net/f_net_tokenizer_test.py index dfbb4aae26..259bba8ef4 100644 --- a/keras_nlp/models/f_net/f_net_tokenizer_test.py +++ b/keras_nlp/models/f_net/f_net_tokenizer_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for FNet tokenizer.""" import io import pytest diff --git a/keras_nlp/models/generative_task.py b/keras_nlp/models/generative_task.py index 141775dc84..27a733d0c6 100644 --- a/keras_nlp/models/generative_task.py +++ b/keras_nlp/models/generative_task.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Base class for Generative Task models.""" import itertools diff --git a/keras_nlp/models/gpt2/gpt2_backbone.py b/keras_nlp/models/gpt2/gpt2_backbone.py index faf12c730a..cb734e3db7 100644 --- a/keras_nlp/models/gpt2/gpt2_backbone.py +++ b/keras_nlp/models/gpt2/gpt2_backbone.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""GPT-2 backbone model.""" - import copy from tensorflow.experimental import dtensor diff --git a/keras_nlp/models/gpt2/gpt2_backbone_test.py b/keras_nlp/models/gpt2/gpt2_backbone_test.py index ccf5e09055..1a1cedcfe4 100644 --- a/keras_nlp/models/gpt2/gpt2_backbone_test.py +++ b/keras_nlp/models/gpt2/gpt2_backbone_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Test for GPT-2 backbone models.""" import os diff --git a/keras_nlp/models/gpt2/gpt2_causal_lm.py b/keras_nlp/models/gpt2/gpt2_causal_lm.py index 6e508c43f8..627d76e517 100644 --- a/keras_nlp/models/gpt2/gpt2_causal_lm.py +++ b/keras_nlp/models/gpt2/gpt2_causal_lm.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""GPT2 Causal LM (Language Model).""" import copy diff --git a/keras_nlp/models/gpt2/gpt2_causal_lm_preprocessor.py b/keras_nlp/models/gpt2/gpt2_causal_lm_preprocessor.py index ddc8a58919..3a8561e4f1 100644 --- a/keras_nlp/models/gpt2/gpt2_causal_lm_preprocessor.py +++ b/keras_nlp/models/gpt2/gpt2_causal_lm_preprocessor.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""GPT2 Causal LM preprocessor layer.""" - import tensorflow as tf from absl import logging diff --git a/keras_nlp/models/gpt2/gpt2_causal_lm_preprocessor_test.py b/keras_nlp/models/gpt2/gpt2_causal_lm_preprocessor_test.py index 4f2ff8c2a5..63ff66b194 100644 --- a/keras_nlp/models/gpt2/gpt2_causal_lm_preprocessor_test.py +++ b/keras_nlp/models/gpt2/gpt2_causal_lm_preprocessor_test.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for GPT2 causal LM preprocessor layer.""" - import tensorflow as tf from keras_nlp.backend import keras diff --git a/keras_nlp/models/gpt2/gpt2_causal_lm_test.py b/keras_nlp/models/gpt2/gpt2_causal_lm_test.py index e52274da7f..c50b6c5cf4 100644 --- a/keras_nlp/models/gpt2/gpt2_causal_lm_test.py +++ b/keras_nlp/models/gpt2/gpt2_causal_lm_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for GPT2 causal LM model.""" import os from unittest.mock import patch diff --git a/keras_nlp/models/gpt2/gpt2_preprocessor.py b/keras_nlp/models/gpt2/gpt2_preprocessor.py index 6469f5c354..bb37364364 100644 --- a/keras_nlp/models/gpt2/gpt2_preprocessor.py +++ b/keras_nlp/models/gpt2/gpt2_preprocessor.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""GPT2 preprocessor layer.""" - import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/gpt2/gpt2_preprocessor_test.py b/keras_nlp/models/gpt2/gpt2_preprocessor_test.py index 31e86c60f0..db221c7279 100644 --- a/keras_nlp/models/gpt2/gpt2_preprocessor_test.py +++ b/keras_nlp/models/gpt2/gpt2_preprocessor_test.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for GPT2 preprocessor layer.""" - import tensorflow as tf from keras_nlp.backend import keras diff --git a/keras_nlp/models/gpt2/gpt2_presets_test.py b/keras_nlp/models/gpt2/gpt2_presets_test.py index e9cdd707d7..37e7e53e87 100644 --- a/keras_nlp/models/gpt2/gpt2_presets_test.py +++ b/keras_nlp/models/gpt2/gpt2_presets_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for loading pretrained model presets.""" import pytest from absl.testing import parameterized diff --git a/keras_nlp/models/gpt2/gpt2_tokenizer.py b/keras_nlp/models/gpt2/gpt2_tokenizer.py index d16dd3f210..9401b385d8 100644 --- a/keras_nlp/models/gpt2/gpt2_tokenizer.py +++ b/keras_nlp/models/gpt2/gpt2_tokenizer.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""GPT-2 preprocessing layers.""" import copy diff --git a/keras_nlp/models/gpt2/gpt2_tokenizer_test.py b/keras_nlp/models/gpt2/gpt2_tokenizer_test.py index 787f5060be..38ed89f121 100644 --- a/keras_nlp/models/gpt2/gpt2_tokenizer_test.py +++ b/keras_nlp/models/gpt2/gpt2_tokenizer_test.py @@ -12,9 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for GPT-2 preprocessing layers.""" - - from keras_nlp.backend import keras from keras_nlp.models.gpt2.gpt2_tokenizer import GPT2Tokenizer from keras_nlp.tests.test_case import TestCase diff --git a/keras_nlp/models/gpt_neo_x/gpt_neo_x_backbone.py b/keras_nlp/models/gpt_neo_x/gpt_neo_x_backbone.py index 2919976da6..62dba1dd2d 100644 --- a/keras_nlp/models/gpt_neo_x/gpt_neo_x_backbone.py +++ b/keras_nlp/models/gpt_neo_x/gpt_neo_x_backbone.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""GPT-NeoX backbone model.""" - from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import keras from keras_nlp.layers.modeling.reversible_embedding import ReversibleEmbedding diff --git a/keras_nlp/models/gpt_neo_x/gpt_neo_x_backbone_test.py b/keras_nlp/models/gpt_neo_x/gpt_neo_x_backbone_test.py index 0f4e34b1f5..803006a06a 100644 --- a/keras_nlp/models/gpt_neo_x/gpt_neo_x_backbone_test.py +++ b/keras_nlp/models/gpt_neo_x/gpt_neo_x_backbone_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Test for GPTNeoX backbone models.""" import os diff --git a/keras_nlp/models/gpt_neo_x/gpt_neo_x_causal_lm.py b/keras_nlp/models/gpt_neo_x/gpt_neo_x_causal_lm.py index 6462a6a8db..3d361b3d38 100644 --- a/keras_nlp/models/gpt_neo_x/gpt_neo_x_causal_lm.py +++ b/keras_nlp/models/gpt_neo_x/gpt_neo_x_causal_lm.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""GPTNeoX Causal LM (Language Model).""" from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import keras diff --git a/keras_nlp/models/gpt_neo_x/gpt_neo_x_causal_lm_preprocessor.py b/keras_nlp/models/gpt_neo_x/gpt_neo_x_causal_lm_preprocessor.py index 123c8582ff..029be909f1 100644 --- a/keras_nlp/models/gpt_neo_x/gpt_neo_x_causal_lm_preprocessor.py +++ b/keras_nlp/models/gpt_neo_x/gpt_neo_x_causal_lm_preprocessor.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""GPTNeoX Causal LM preprocessor layer.""" - import tensorflow as tf from absl import logging diff --git a/keras_nlp/models/gpt_neo_x/gpt_neo_x_causal_lm_preprocessor_test.py b/keras_nlp/models/gpt_neo_x/gpt_neo_x_causal_lm_preprocessor_test.py index da4ae14aad..e494b12b63 100644 --- a/keras_nlp/models/gpt_neo_x/gpt_neo_x_causal_lm_preprocessor_test.py +++ b/keras_nlp/models/gpt_neo_x/gpt_neo_x_causal_lm_preprocessor_test.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for GPTNeoX causal LM preprocessor layer.""" - import tensorflow as tf from keras_nlp.backend import keras diff --git a/keras_nlp/models/gpt_neo_x/gpt_neo_x_causal_lm_test.py b/keras_nlp/models/gpt_neo_x/gpt_neo_x_causal_lm_test.py index 1106e184a1..dda9f51bdd 100644 --- a/keras_nlp/models/gpt_neo_x/gpt_neo_x_causal_lm_test.py +++ b/keras_nlp/models/gpt_neo_x/gpt_neo_x_causal_lm_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for GPTNeoX causal LM model.""" import os from unittest.mock import patch diff --git a/keras_nlp/models/gpt_neo_x/gpt_neo_x_decoder.py b/keras_nlp/models/gpt_neo_x/gpt_neo_x_decoder.py index 1ad2fd263c..bbbacd35ad 100644 --- a/keras_nlp/models/gpt_neo_x/gpt_neo_x_decoder.py +++ b/keras_nlp/models/gpt_neo_x/gpt_neo_x_decoder.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + from keras_nlp.backend import keras from keras_nlp.backend import ops from keras_nlp.layers.modeling.transformer_layer_utils import ( diff --git a/keras_nlp/models/gpt_neo_x/gpt_neo_x_preprocessor.py b/keras_nlp/models/gpt_neo_x/gpt_neo_x_preprocessor.py index b20fb8157d..0dad9a053a 100644 --- a/keras_nlp/models/gpt_neo_x/gpt_neo_x_preprocessor.py +++ b/keras_nlp/models/gpt_neo_x/gpt_neo_x_preprocessor.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""GPTNeoX preprocessor layer.""" from keras_nlp.api_export import keras_nlp_export from keras_nlp.layers.preprocessing.start_end_packer import StartEndPacker from keras_nlp.models.gpt_neo_x.gpt_neo_x_tokenizer import GPTNeoXTokenizer diff --git a/keras_nlp/models/gpt_neo_x/gpt_neo_x_preprocessor_test.py b/keras_nlp/models/gpt_neo_x/gpt_neo_x_preprocessor_test.py index 974d33ce3b..53655822a0 100644 --- a/keras_nlp/models/gpt_neo_x/gpt_neo_x_preprocessor_test.py +++ b/keras_nlp/models/gpt_neo_x/gpt_neo_x_preprocessor_test.py @@ -12,9 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for GPTNeoX preprocessor layer.""" - - import tensorflow as tf from keras_nlp.backend import keras diff --git a/keras_nlp/models/gpt_neo_x/gpt_neo_x_tokenizer.py b/keras_nlp/models/gpt_neo_x/gpt_neo_x_tokenizer.py index e34ade14fe..3935d85a65 100644 --- a/keras_nlp/models/gpt_neo_x/gpt_neo_x_tokenizer.py +++ b/keras_nlp/models/gpt_neo_x/gpt_neo_x_tokenizer.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""GptNeoX tokenizer""" from keras_nlp.api_export import keras_nlp_export from keras_nlp.tokenizers.byte_pair_tokenizer import BytePairTokenizer diff --git a/keras_nlp/models/gpt_neo_x/gpt_neo_x_tokenizer_test.py b/keras_nlp/models/gpt_neo_x/gpt_neo_x_tokenizer_test.py index 7cbb01456d..da6daa2c24 100644 --- a/keras_nlp/models/gpt_neo_x/gpt_neo_x_tokenizer_test.py +++ b/keras_nlp/models/gpt_neo_x/gpt_neo_x_tokenizer_test.py @@ -12,9 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for GPT-NeoX preprocessing layers.""" - - from keras_nlp.backend import keras from keras_nlp.models.gpt_neo_x.gpt_neo_x_tokenizer import GPTNeoXTokenizer from keras_nlp.tests.test_case import TestCase diff --git a/keras_nlp/models/opt/opt_backbone.py b/keras_nlp/models/opt/opt_backbone.py index 4052644c00..8fe37472a0 100644 --- a/keras_nlp/models/opt/opt_backbone.py +++ b/keras_nlp/models/opt/opt_backbone.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""OPT backbone model.""" - import copy from tensorflow.experimental import dtensor diff --git a/keras_nlp/models/opt/opt_backbone_test.py b/keras_nlp/models/opt/opt_backbone_test.py index d59f24fc8d..012c99c8a6 100644 --- a/keras_nlp/models/opt/opt_backbone_test.py +++ b/keras_nlp/models/opt/opt_backbone_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Test for OPT backbone models.""" import os diff --git a/keras_nlp/models/opt/opt_causal_lm.py b/keras_nlp/models/opt/opt_causal_lm.py index c649da8552..f5033d197e 100644 --- a/keras_nlp/models/opt/opt_causal_lm.py +++ b/keras_nlp/models/opt/opt_causal_lm.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""OPT Causal LM (Language Model).""" import copy diff --git a/keras_nlp/models/opt/opt_causal_lm_preprocessor.py b/keras_nlp/models/opt/opt_causal_lm_preprocessor.py index 099cacd19c..c27294989f 100644 --- a/keras_nlp/models/opt/opt_causal_lm_preprocessor.py +++ b/keras_nlp/models/opt/opt_causal_lm_preprocessor.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""OPT Causal LM preprocessor layer.""" - import tensorflow as tf from absl import logging diff --git a/keras_nlp/models/opt/opt_causal_lm_preprocessor_test.py b/keras_nlp/models/opt/opt_causal_lm_preprocessor_test.py index accd88baf8..eb54a94196 100644 --- a/keras_nlp/models/opt/opt_causal_lm_preprocessor_test.py +++ b/keras_nlp/models/opt/opt_causal_lm_preprocessor_test.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for OPT causal LM preprocessor layer.""" - import tensorflow as tf from keras_nlp.backend import keras diff --git a/keras_nlp/models/opt/opt_causal_lm_test.py b/keras_nlp/models/opt/opt_causal_lm_test.py index b4bff2c7d3..19325c7b0a 100644 --- a/keras_nlp/models/opt/opt_causal_lm_test.py +++ b/keras_nlp/models/opt/opt_causal_lm_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for OPT causal LM model.""" import os from unittest.mock import patch diff --git a/keras_nlp/models/opt/opt_preprocessor.py b/keras_nlp/models/opt/opt_preprocessor.py index 71c93e68ef..6a6b5537bf 100644 --- a/keras_nlp/models/opt/opt_preprocessor.py +++ b/keras_nlp/models/opt/opt_preprocessor.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""OPT preprocessor layer.""" - import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/opt/opt_preprocessor_test.py b/keras_nlp/models/opt/opt_preprocessor_test.py index 060963796d..ae68cf8089 100644 --- a/keras_nlp/models/opt/opt_preprocessor_test.py +++ b/keras_nlp/models/opt/opt_preprocessor_test.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for OPT preprocessor layer.""" - import tensorflow as tf from keras_nlp.backend import keras diff --git a/keras_nlp/models/opt/opt_presets_test.py b/keras_nlp/models/opt/opt_presets_test.py index 744d9d11ef..a9426d29e7 100644 --- a/keras_nlp/models/opt/opt_presets_test.py +++ b/keras_nlp/models/opt/opt_presets_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for loading pretrained model presets.""" import pytest from absl.testing import parameterized diff --git a/keras_nlp/models/opt/opt_tokenizer.py b/keras_nlp/models/opt/opt_tokenizer.py index e0a5d70a00..b15aa94842 100644 --- a/keras_nlp/models/opt/opt_tokenizer.py +++ b/keras_nlp/models/opt/opt_tokenizer.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""OPT tokenizer.""" - import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/opt/opt_tokenizer_test.py b/keras_nlp/models/opt/opt_tokenizer_test.py index ca8839fb5e..af460db409 100644 --- a/keras_nlp/models/opt/opt_tokenizer_test.py +++ b/keras_nlp/models/opt/opt_tokenizer_test.py @@ -12,9 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for OPT tokenizer layer.""" - - from keras_nlp.backend import keras from keras_nlp.models.opt.opt_tokenizer import OPTTokenizer from keras_nlp.tests.test_case import TestCase diff --git a/keras_nlp/models/roberta/roberta_backbone.py b/keras_nlp/models/roberta/roberta_backbone.py index 29bc140589..8495b5cb69 100644 --- a/keras_nlp/models/roberta/roberta_backbone.py +++ b/keras_nlp/models/roberta/roberta_backbone.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""RoBERTa backbone model.""" - import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/roberta/roberta_backbone_test.py b/keras_nlp/models/roberta/roberta_backbone_test.py index f60a17004c..9a466b527f 100644 --- a/keras_nlp/models/roberta/roberta_backbone_test.py +++ b/keras_nlp/models/roberta/roberta_backbone_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Test for RoBERTa backbone models.""" import os import numpy as np diff --git a/keras_nlp/models/roberta/roberta_classifier.py b/keras_nlp/models/roberta/roberta_classifier.py index c5e9f53de1..e62b89918b 100644 --- a/keras_nlp/models/roberta/roberta_classifier.py +++ b/keras_nlp/models/roberta/roberta_classifier.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""RoBERTa classification model.""" import copy diff --git a/keras_nlp/models/roberta/roberta_classifier_test.py b/keras_nlp/models/roberta/roberta_classifier_test.py index 68da040e1e..6636768bdc 100644 --- a/keras_nlp/models/roberta/roberta_classifier_test.py +++ b/keras_nlp/models/roberta/roberta_classifier_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for RoBERTa classification model.""" import os diff --git a/keras_nlp/models/roberta/roberta_masked_lm.py b/keras_nlp/models/roberta/roberta_masked_lm.py index f80566c30d..aad1e5c382 100644 --- a/keras_nlp/models/roberta/roberta_masked_lm.py +++ b/keras_nlp/models/roberta/roberta_masked_lm.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""RoBERTa masked lm model.""" import copy diff --git a/keras_nlp/models/roberta/roberta_masked_lm_preprocessor.py b/keras_nlp/models/roberta/roberta_masked_lm_preprocessor.py index 881bb3b9a4..0d5a24e129 100644 --- a/keras_nlp/models/roberta/roberta_masked_lm_preprocessor.py +++ b/keras_nlp/models/roberta/roberta_masked_lm_preprocessor.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""RoBERTa masked language model preprocessor layer.""" - from absl import logging from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/roberta/roberta_masked_lm_preprocessor_test.py b/keras_nlp/models/roberta/roberta_masked_lm_preprocessor_test.py index 1bde1fa616..92abcf3011 100644 --- a/keras_nlp/models/roberta/roberta_masked_lm_preprocessor_test.py +++ b/keras_nlp/models/roberta/roberta_masked_lm_preprocessor_test.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for RoBERTa masked language model preprocessor layer.""" - import tensorflow as tf from keras_nlp.backend import keras diff --git a/keras_nlp/models/roberta/roberta_masked_lm_test.py b/keras_nlp/models/roberta/roberta_masked_lm_test.py index 82aaea1a84..d2a5a27011 100644 --- a/keras_nlp/models/roberta/roberta_masked_lm_test.py +++ b/keras_nlp/models/roberta/roberta_masked_lm_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for RoBERTa masked language model.""" import os diff --git a/keras_nlp/models/roberta/roberta_preprocessor.py b/keras_nlp/models/roberta/roberta_preprocessor.py index a83fe892e0..7e641a93a8 100644 --- a/keras_nlp/models/roberta/roberta_preprocessor.py +++ b/keras_nlp/models/roberta/roberta_preprocessor.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""RoBERTa preprocessor layer.""" - import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/roberta/roberta_preprocessor_test.py b/keras_nlp/models/roberta/roberta_preprocessor_test.py index 0f857d8fd3..3471fd372d 100644 --- a/keras_nlp/models/roberta/roberta_preprocessor_test.py +++ b/keras_nlp/models/roberta/roberta_preprocessor_test.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for RoBERTa preprocessor layer.""" - import tensorflow as tf from keras_nlp.backend import keras diff --git a/keras_nlp/models/roberta/roberta_presets_test.py b/keras_nlp/models/roberta/roberta_presets_test.py index c3f5f63aaf..22c1f2e8a6 100644 --- a/keras_nlp/models/roberta/roberta_presets_test.py +++ b/keras_nlp/models/roberta/roberta_presets_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for loading pretrained model presets.""" import pytest from absl.testing import parameterized diff --git a/keras_nlp/models/roberta/roberta_tokenizer.py b/keras_nlp/models/roberta/roberta_tokenizer.py index 5eee04bc5e..838f32a8ac 100644 --- a/keras_nlp/models/roberta/roberta_tokenizer.py +++ b/keras_nlp/models/roberta/roberta_tokenizer.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""RoBERTa tokenizer.""" - import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/roberta/roberta_tokenizer_test.py b/keras_nlp/models/roberta/roberta_tokenizer_test.py index e4b90e4e37..ef47f204eb 100644 --- a/keras_nlp/models/roberta/roberta_tokenizer_test.py +++ b/keras_nlp/models/roberta/roberta_tokenizer_test.py @@ -12,9 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for RoBERTa tokenizer.""" - - from keras_nlp.backend import keras from keras_nlp.models.roberta.roberta_tokenizer import RobertaTokenizer from keras_nlp.tests.test_case import TestCase diff --git a/keras_nlp/models/t5/t5_backbone.py b/keras_nlp/models/t5/t5_backbone.py index 071405f73d..7514cc51ae 100644 --- a/keras_nlp/models/t5/t5_backbone.py +++ b/keras_nlp/models/t5/t5_backbone.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""T5 backbone model.""" - from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import keras from keras_nlp.layers.modeling.reversible_embedding import ReversibleEmbedding diff --git a/keras_nlp/models/t5/t5_backbone_test.py b/keras_nlp/models/t5/t5_backbone_test.py index 4b7affddb3..ab4270d9d9 100644 --- a/keras_nlp/models/t5/t5_backbone_test.py +++ b/keras_nlp/models/t5/t5_backbone_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Test for T5 backbone model.""" import os diff --git a/keras_nlp/models/t5/t5_layer_norm.py b/keras_nlp/models/t5/t5_layer_norm.py index dab361e279..7cfdb2315e 100644 --- a/keras_nlp/models/t5/t5_layer_norm.py +++ b/keras_nlp/models/t5/t5_layer_norm.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + import tensorflow as tf from keras_nlp.backend import keras diff --git a/keras_nlp/models/t5/t5_multi_head_attention.py b/keras_nlp/models/t5/t5_multi_head_attention.py index 298cf67b18..7c11382f13 100644 --- a/keras_nlp/models/t5/t5_multi_head_attention.py +++ b/keras_nlp/models/t5/t5_multi_head_attention.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + import tensorflow as tf from tensorflow.compiler.tf2xla.python.xla import dynamic_slice diff --git a/keras_nlp/models/t5/t5_tokenizer.py b/keras_nlp/models/t5/t5_tokenizer.py index 5eb2437b5d..ae9facb318 100644 --- a/keras_nlp/models/t5/t5_tokenizer.py +++ b/keras_nlp/models/t5/t5_tokenizer.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""T5 tokenizer.""" - from keras_nlp.api_export import keras_nlp_export from keras_nlp.tokenizers.sentence_piece_tokenizer import SentencePieceTokenizer diff --git a/keras_nlp/models/t5/t5_tokenizer_test.py b/keras_nlp/models/t5/t5_tokenizer_test.py index e6b5a8d690..ce16492acd 100644 --- a/keras_nlp/models/t5/t5_tokenizer_test.py +++ b/keras_nlp/models/t5/t5_tokenizer_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for T5 tokenizer.""" import io import sentencepiece diff --git a/keras_nlp/models/t5/t5_transformer_layer.py b/keras_nlp/models/t5/t5_transformer_layer.py index 8e516b8bb2..c789f13dea 100644 --- a/keras_nlp/models/t5/t5_transformer_layer.py +++ b/keras_nlp/models/t5/t5_transformer_layer.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + import tensorflow as tf from keras_nlp.backend import keras diff --git a/keras_nlp/models/task.py b/keras_nlp/models/task.py index 583c107253..6f65a78f33 100644 --- a/keras_nlp/models/task.py +++ b/keras_nlp/models/task.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Base class for Task models.""" import os diff --git a/keras_nlp/models/task_test.py b/keras_nlp/models/task_test.py index 4155eb2e63..09fe1b0086 100644 --- a/keras_nlp/models/task_test.py +++ b/keras_nlp/models/task_test.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + from keras_nlp.backend import keras from keras_nlp.models.preprocessor import Preprocessor from keras_nlp.models.task import Task diff --git a/keras_nlp/models/whisper/whisper_audio_feature_extractor_test.py b/keras_nlp/models/whisper/whisper_audio_feature_extractor_test.py index 9dfeed8705..8eca8f9c79 100644 --- a/keras_nlp/models/whisper/whisper_audio_feature_extractor_test.py +++ b/keras_nlp/models/whisper/whisper_audio_feature_extractor_test.py @@ -11,8 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for Whisper audio feature extractor.""" - import tensorflow as tf diff --git a/keras_nlp/models/whisper/whisper_backbone.py b/keras_nlp/models/whisper/whisper_backbone.py index 5bdb219ed2..87c94ae7fd 100644 --- a/keras_nlp/models/whisper/whisper_backbone.py +++ b/keras_nlp/models/whisper/whisper_backbone.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Whisper backbone model.""" import copy diff --git a/keras_nlp/models/whisper/whisper_backbone_test.py b/keras_nlp/models/whisper/whisper_backbone_test.py index c9005fa17f..e3266ea9d5 100644 --- a/keras_nlp/models/whisper/whisper_backbone_test.py +++ b/keras_nlp/models/whisper/whisper_backbone_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Test for Whisper backbone models.""" import os diff --git a/keras_nlp/models/whisper/whisper_decoder.py b/keras_nlp/models/whisper/whisper_decoder.py index 0d38332d89..7f5d834741 100644 --- a/keras_nlp/models/whisper/whisper_decoder.py +++ b/keras_nlp/models/whisper/whisper_decoder.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Whisper decoder block.""" from keras_nlp.backend import keras from keras_nlp.layers.modeling.transformer_decoder import TransformerDecoder diff --git a/keras_nlp/models/whisper/whisper_encoder.py b/keras_nlp/models/whisper/whisper_encoder.py index 4527e3fa51..31267cbf78 100644 --- a/keras_nlp/models/whisper/whisper_encoder.py +++ b/keras_nlp/models/whisper/whisper_encoder.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Whisper encoder block.""" from keras_nlp.backend import keras from keras_nlp.layers.modeling.transformer_encoder import TransformerEncoder diff --git a/keras_nlp/models/whisper/whisper_preprocessor.py b/keras_nlp/models/whisper/whisper_preprocessor.py index 62d853698a..88fd60cae2 100644 --- a/keras_nlp/models/whisper/whisper_preprocessor.py +++ b/keras_nlp/models/whisper/whisper_preprocessor.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Whisper preprocessor layer.""" import copy diff --git a/keras_nlp/models/whisper/whisper_preprocessor_test.py b/keras_nlp/models/whisper/whisper_preprocessor_test.py index 9c14e726ae..3f07ef618f 100644 --- a/keras_nlp/models/whisper/whisper_preprocessor_test.py +++ b/keras_nlp/models/whisper/whisper_preprocessor_test.py @@ -12,9 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for Whisper preprocessor layer.""" - - import tensorflow as tf from keras_nlp.backend import keras diff --git a/keras_nlp/models/whisper/whisper_presets_test.py b/keras_nlp/models/whisper/whisper_presets_test.py index 445db674e5..09529c66f3 100644 --- a/keras_nlp/models/whisper/whisper_presets_test.py +++ b/keras_nlp/models/whisper/whisper_presets_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for loading pretrained model presets.""" import pytest import tensorflow as tf diff --git a/keras_nlp/models/whisper/whisper_tokenizer.py b/keras_nlp/models/whisper/whisper_tokenizer.py index 37a5e6078a..b1406b0a04 100644 --- a/keras_nlp/models/whisper/whisper_tokenizer.py +++ b/keras_nlp/models/whisper/whisper_tokenizer.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Whisper tokenizer.""" import copy import json diff --git a/keras_nlp/models/whisper/whisper_tokenizer_test.py b/keras_nlp/models/whisper/whisper_tokenizer_test.py index bb1697b32a..5ebcb12e55 100644 --- a/keras_nlp/models/whisper/whisper_tokenizer_test.py +++ b/keras_nlp/models/whisper/whisper_tokenizer_test.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for Whisper preprocessing layers.""" - import pytest from keras_nlp.models.whisper.whisper_tokenizer import WhisperTokenizer diff --git a/keras_nlp/models/xlm_roberta/xlm_roberta_backbone.py b/keras_nlp/models/xlm_roberta/xlm_roberta_backbone.py index 81a069cb18..b83b4596b2 100644 --- a/keras_nlp/models/xlm_roberta/xlm_roberta_backbone.py +++ b/keras_nlp/models/xlm_roberta/xlm_roberta_backbone.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""XLM-RoBERTa backbone model.""" - import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/xlm_roberta/xlm_roberta_backbone_test.py b/keras_nlp/models/xlm_roberta/xlm_roberta_backbone_test.py index ae0f5d1670..26559f990d 100644 --- a/keras_nlp/models/xlm_roberta/xlm_roberta_backbone_test.py +++ b/keras_nlp/models/xlm_roberta/xlm_roberta_backbone_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for XLM-RoBERTa backbone models.""" import os import numpy as np diff --git a/keras_nlp/models/xlm_roberta/xlm_roberta_classifier.py b/keras_nlp/models/xlm_roberta/xlm_roberta_classifier.py index 1eec0588e7..67a9dd5bef 100644 --- a/keras_nlp/models/xlm_roberta/xlm_roberta_classifier.py +++ b/keras_nlp/models/xlm_roberta/xlm_roberta_classifier.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""XLM-RoBERTa classification model.""" import copy diff --git a/keras_nlp/models/xlm_roberta/xlm_roberta_classifier_test.py b/keras_nlp/models/xlm_roberta/xlm_roberta_classifier_test.py index 65092ce591..f96e40b5b4 100644 --- a/keras_nlp/models/xlm_roberta/xlm_roberta_classifier_test.py +++ b/keras_nlp/models/xlm_roberta/xlm_roberta_classifier_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for XLM-RoBERTa classification model.""" import io import os diff --git a/keras_nlp/models/xlm_roberta/xlm_roberta_masked_lm.py b/keras_nlp/models/xlm_roberta/xlm_roberta_masked_lm.py index b0d9b7fc34..eb21b3d099 100644 --- a/keras_nlp/models/xlm_roberta/xlm_roberta_masked_lm.py +++ b/keras_nlp/models/xlm_roberta/xlm_roberta_masked_lm.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""XLM-RoBERTa masked lm model.""" import copy diff --git a/keras_nlp/models/xlm_roberta/xlm_roberta_masked_lm_preprocessor.py b/keras_nlp/models/xlm_roberta/xlm_roberta_masked_lm_preprocessor.py index 208df561d1..2a8750c583 100644 --- a/keras_nlp/models/xlm_roberta/xlm_roberta_masked_lm_preprocessor.py +++ b/keras_nlp/models/xlm_roberta/xlm_roberta_masked_lm_preprocessor.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""XLM-RoBERTa masked language model preprocessor layer.""" - from absl import logging from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/xlm_roberta/xlm_roberta_masked_lm_preprocessor_test.py b/keras_nlp/models/xlm_roberta/xlm_roberta_masked_lm_preprocessor_test.py index 72114a3f31..cbb74c7722 100644 --- a/keras_nlp/models/xlm_roberta/xlm_roberta_masked_lm_preprocessor_test.py +++ b/keras_nlp/models/xlm_roberta/xlm_roberta_masked_lm_preprocessor_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for XLM-RoBERTa masked language model preprocessor layer.""" import io import sentencepiece diff --git a/keras_nlp/models/xlm_roberta/xlm_roberta_masked_lm_test.py b/keras_nlp/models/xlm_roberta/xlm_roberta_masked_lm_test.py index 445a473c06..94c06ea7be 100644 --- a/keras_nlp/models/xlm_roberta/xlm_roberta_masked_lm_test.py +++ b/keras_nlp/models/xlm_roberta/xlm_roberta_masked_lm_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for XLM-RoBERTa masked language model.""" import io import os diff --git a/keras_nlp/models/xlm_roberta/xlm_roberta_preprocessor.py b/keras_nlp/models/xlm_roberta/xlm_roberta_preprocessor.py index beff54d2ca..e557bd0635 100644 --- a/keras_nlp/models/xlm_roberta/xlm_roberta_preprocessor.py +++ b/keras_nlp/models/xlm_roberta/xlm_roberta_preprocessor.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""XLM-RoBERTa preprocessor layer.""" - import copy from keras_nlp.api_export import keras_nlp_export diff --git a/keras_nlp/models/xlm_roberta/xlm_roberta_preprocessor_test.py b/keras_nlp/models/xlm_roberta/xlm_roberta_preprocessor_test.py index 7fa3d29431..711b52e264 100644 --- a/keras_nlp/models/xlm_roberta/xlm_roberta_preprocessor_test.py +++ b/keras_nlp/models/xlm_roberta/xlm_roberta_preprocessor_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for XLM-RoBERTa preprocessor layer.""" import io import sentencepiece diff --git a/keras_nlp/models/xlm_roberta/xlm_roberta_presets_test.py b/keras_nlp/models/xlm_roberta/xlm_roberta_presets_test.py index 468b3f8fb2..b74de26b01 100644 --- a/keras_nlp/models/xlm_roberta/xlm_roberta_presets_test.py +++ b/keras_nlp/models/xlm_roberta/xlm_roberta_presets_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for loading pretrained model presets.""" import pytest from absl.testing import parameterized diff --git a/keras_nlp/models/xlm_roberta/xlm_roberta_tokenizer.py b/keras_nlp/models/xlm_roberta/xlm_roberta_tokenizer.py index 340d42d1ae..c0b91abb3c 100644 --- a/keras_nlp/models/xlm_roberta/xlm_roberta_tokenizer.py +++ b/keras_nlp/models/xlm_roberta/xlm_roberta_tokenizer.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""XLM-RoBERTa tokenizer.""" - import copy import tensorflow as tf diff --git a/keras_nlp/models/xlm_roberta/xlm_roberta_tokenizer_test.py b/keras_nlp/models/xlm_roberta/xlm_roberta_tokenizer_test.py index 965078e682..7800e3913c 100644 --- a/keras_nlp/models/xlm_roberta/xlm_roberta_tokenizer_test.py +++ b/keras_nlp/models/xlm_roberta/xlm_roberta_tokenizer_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for XLM-RoBERTa tokenizer.""" import io import sentencepiece diff --git a/keras_nlp/models/xlnet/relative_attention.py b/keras_nlp/models/xlnet/relative_attention.py index 771e748eee..d11dc4bd5a 100644 --- a/keras_nlp/models/xlnet/relative_attention.py +++ b/keras_nlp/models/xlnet/relative_attention.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Keras-based relative attention layers.""" - import math import string diff --git a/keras_nlp/models/xlnet/xlnet_backbone.py b/keras_nlp/models/xlnet/xlnet_backbone.py index dd6abd5773..3f3a8b771e 100644 --- a/keras_nlp/models/xlnet/xlnet_backbone.py +++ b/keras_nlp/models/xlnet/xlnet_backbone.py @@ -12,9 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""XLNet backbone model.""" - - from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import keras from keras_nlp.models.backbone import Backbone diff --git a/keras_nlp/models/xlnet/xlnet_backbone_test.py b/keras_nlp/models/xlnet/xlnet_backbone_test.py index d59b88e46c..f8cdc3e7be 100644 --- a/keras_nlp/models/xlnet/xlnet_backbone_test.py +++ b/keras_nlp/models/xlnet/xlnet_backbone_test.py @@ -12,9 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Test for XLNet backbone models.""" - - import os import numpy as np diff --git a/keras_nlp/models/xlnet/xlnet_content_and_query_embedding.py b/keras_nlp/models/xlnet/xlnet_content_and_query_embedding.py index d4749fc4cc..1e7a63ed8b 100644 --- a/keras_nlp/models/xlnet/xlnet_content_and_query_embedding.py +++ b/keras_nlp/models/xlnet/xlnet_content_and_query_embedding.py @@ -12,9 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""XLNet Content and Query Embedding implementation""" - - from keras_nlp.backend import keras from keras_nlp.backend import ops diff --git a/keras_nlp/models/xlnet/xlnet_encoder.py b/keras_nlp/models/xlnet/xlnet_encoder.py index ed001df8eb..13f5a953ee 100644 --- a/keras_nlp/models/xlnet/xlnet_encoder.py +++ b/keras_nlp/models/xlnet/xlnet_encoder.py @@ -12,9 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""XLNet Encoder block implementation based on `keras.layers.Layer`.""" - - from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import keras from keras_nlp.backend import ops diff --git a/keras_nlp/samplers/beam_sampler.py b/keras_nlp/samplers/beam_sampler.py index bc7a33e994..1c17342ced 100644 --- a/keras_nlp/samplers/beam_sampler.py +++ b/keras_nlp/samplers/beam_sampler.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Beam Sampler.""" import tensorflow as tf import tree diff --git a/keras_nlp/samplers/beam_sampler_test.py b/keras_nlp/samplers/beam_sampler_test.py index 737fe8aa84..fca37cd85b 100644 --- a/keras_nlp/samplers/beam_sampler_test.py +++ b/keras_nlp/samplers/beam_sampler_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for Beam sampler.""" import pytest import tensorflow as tf diff --git a/keras_nlp/samplers/contrastive_sampler.py b/keras_nlp/samplers/contrastive_sampler.py index 4e69e72a84..ab88858bbb 100644 --- a/keras_nlp/samplers/contrastive_sampler.py +++ b/keras_nlp/samplers/contrastive_sampler.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Contrastive Sampler.""" import tree diff --git a/keras_nlp/samplers/contrastive_sampler_test.py b/keras_nlp/samplers/contrastive_sampler_test.py index 51d1d53800..2e6f19da19 100644 --- a/keras_nlp/samplers/contrastive_sampler_test.py +++ b/keras_nlp/samplers/contrastive_sampler_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for Contrastive Sampler.""" import pytest import tensorflow as tf diff --git a/keras_nlp/samplers/greedy_sampler.py b/keras_nlp/samplers/greedy_sampler.py index 62bbb06ed3..8e178b7468 100644 --- a/keras_nlp/samplers/greedy_sampler.py +++ b/keras_nlp/samplers/greedy_sampler.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Greedy Sampler.""" from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import ops diff --git a/keras_nlp/samplers/greedy_sampler_test.py b/keras_nlp/samplers/greedy_sampler_test.py index fcc331c07e..618c94d118 100644 --- a/keras_nlp/samplers/greedy_sampler_test.py +++ b/keras_nlp/samplers/greedy_sampler_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for Greedy sampler.""" import pytest import tensorflow as tf diff --git a/keras_nlp/samplers/random_sampler.py b/keras_nlp/samplers/random_sampler.py index e18de34f2d..1255336b29 100644 --- a/keras_nlp/samplers/random_sampler.py +++ b/keras_nlp/samplers/random_sampler.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Random Sampler.""" from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import ops diff --git a/keras_nlp/samplers/random_sampler_test.py b/keras_nlp/samplers/random_sampler_test.py index 726b80ebc7..3149d11dba 100644 --- a/keras_nlp/samplers/random_sampler_test.py +++ b/keras_nlp/samplers/random_sampler_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for Random sampler.""" import numpy as np import pytest diff --git a/keras_nlp/samplers/sampler.py b/keras_nlp/samplers/sampler.py index 7ca4c9b10f..f7f083221f 100644 --- a/keras_nlp/samplers/sampler.py +++ b/keras_nlp/samplers/sampler.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Base sampler class.""" from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import config diff --git a/keras_nlp/samplers/serialization_test.py b/keras_nlp/samplers/serialization_test.py index a0c5491f53..07dc328790 100644 --- a/keras_nlp/samplers/serialization_test.py +++ b/keras_nlp/samplers/serialization_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for Sampler classes.""" from keras_nlp.samplers.serialization import deserialize from keras_nlp.samplers.serialization import get diff --git a/keras_nlp/samplers/top_k_sampler.py b/keras_nlp/samplers/top_k_sampler.py index f6d88aa840..3456694848 100644 --- a/keras_nlp/samplers/top_k_sampler.py +++ b/keras_nlp/samplers/top_k_sampler.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Top-k Sampler.""" from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import ops diff --git a/keras_nlp/samplers/top_k_sampler_test.py b/keras_nlp/samplers/top_k_sampler_test.py index bd2d56506f..c0f516a214 100644 --- a/keras_nlp/samplers/top_k_sampler_test.py +++ b/keras_nlp/samplers/top_k_sampler_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for Top-K sampler.""" import pytest import tensorflow as tf diff --git a/keras_nlp/samplers/top_p_sampler.py b/keras_nlp/samplers/top_p_sampler.py index 608f227121..a04b39aa2b 100644 --- a/keras_nlp/samplers/top_p_sampler.py +++ b/keras_nlp/samplers/top_p_sampler.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Top-p Sampler.""" from keras_nlp.api_export import keras_nlp_export from keras_nlp.backend import ops diff --git a/keras_nlp/samplers/top_p_sampler_test.py b/keras_nlp/samplers/top_p_sampler_test.py index 8a2c25e4da..fea5ca4110 100644 --- a/keras_nlp/samplers/top_p_sampler_test.py +++ b/keras_nlp/samplers/top_p_sampler_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for Top-P sampler.""" import numpy as np import pytest diff --git a/keras_nlp/tests/doc_tests/docstring_lib.py b/keras_nlp/tests/doc_tests/docstring_lib.py index 8e13a708f9..5784a7231f 100644 --- a/keras_nlp/tests/doc_tests/docstring_lib.py +++ b/keras_nlp/tests/doc_tests/docstring_lib.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Doctests lib for KerasNLP.""" import doctest import re diff --git a/keras_nlp/tests/doc_tests/fenced_docstring_lib.py b/keras_nlp/tests/doc_tests/fenced_docstring_lib.py index 38819429d5..2b4c7f1de0 100644 --- a/keras_nlp/tests/doc_tests/fenced_docstring_lib.py +++ b/keras_nlp/tests/doc_tests/fenced_docstring_lib.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Fenced docstring docstest lib for KerasNLP.""" - import ast import doctest import re diff --git a/keras_nlp/tests/test_case.py b/keras_nlp/tests/test_case.py index fc5ff723e8..67721ff812 100644 --- a/keras_nlp/tests/test_case.py +++ b/keras_nlp/tests/test_case.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + import tensorflow as tf import tree from absl.testing import parameterized diff --git a/keras_nlp/tokenizers/byte_pair_tokenizer_test.py b/keras_nlp/tokenizers/byte_pair_tokenizer_test.py index f3f28fa509..d5f7b3762a 100644 --- a/keras_nlp/tokenizers/byte_pair_tokenizer_test.py +++ b/keras_nlp/tokenizers/byte_pair_tokenizer_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. - import pytest import tensorflow as tf diff --git a/keras_nlp/tokenizers/byte_tokenizer.py b/keras_nlp/tokenizers/byte_tokenizer.py index 317d9588e7..f6b60bfeaf 100644 --- a/keras_nlp/tokenizers/byte_tokenizer.py +++ b/keras_nlp/tokenizers/byte_tokenizer.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""Byte Tokenizer.""" - import numpy as np import tensorflow as tf diff --git a/keras_nlp/tokenizers/byte_tokenizer_test.py b/keras_nlp/tokenizers/byte_tokenizer_test.py index 9b520fd087..c09f1072f4 100644 --- a/keras_nlp/tokenizers/byte_tokenizer_test.py +++ b/keras_nlp/tokenizers/byte_tokenizer_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. - import tensorflow as tf from keras_nlp.tests.test_case import TestCase diff --git a/keras_nlp/tokenizers/sentence_piece_tokenizer_trainer.py b/keras_nlp/tokenizers/sentence_piece_tokenizer_trainer.py index 48ce4b1e36..af9f2624b2 100644 --- a/keras_nlp/tokenizers/sentence_piece_tokenizer_trainer.py +++ b/keras_nlp/tokenizers/sentence_piece_tokenizer_trainer.py @@ -11,7 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Trainer for SentencePiece Tokenizer.""" + import io import tensorflow as tf diff --git a/keras_nlp/tokenizers/sentence_piece_tokenizer_trainer_test.py b/keras_nlp/tokenizers/sentence_piece_tokenizer_trainer_test.py index 872bff2d9c..f16d38b72e 100644 --- a/keras_nlp/tokenizers/sentence_piece_tokenizer_trainer_test.py +++ b/keras_nlp/tokenizers/sentence_piece_tokenizer_trainer_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for SentencePiece Tokenizer Trainer.""" import os import re diff --git a/keras_nlp/tokenizers/unicode_codepoint_tokenizer_test.py b/keras_nlp/tokenizers/unicode_codepoint_tokenizer_test.py index 7cb5e714af..c1dcb79ee1 100644 --- a/keras_nlp/tokenizers/unicode_codepoint_tokenizer_test.py +++ b/keras_nlp/tokenizers/unicode_codepoint_tokenizer_test.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. - import tensorflow as tf from keras_nlp.tests.test_case import TestCase diff --git a/keras_nlp/tokenizers/word_piece_tokenizer_trainer.py b/keras_nlp/tokenizers/word_piece_tokenizer_trainer.py index dffba4e7ea..8571097e06 100644 --- a/keras_nlp/tokenizers/word_piece_tokenizer_trainer.py +++ b/keras_nlp/tokenizers/word_piece_tokenizer_trainer.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Trainer for WordPiece Tokenizer.""" import tensorflow as tf diff --git a/keras_nlp/tokenizers/word_piece_tokenizer_trainer_test.py b/keras_nlp/tokenizers/word_piece_tokenizer_trainer_test.py index bcdf6c9a2b..62d0f4adf1 100644 --- a/keras_nlp/tokenizers/word_piece_tokenizer_trainer_test.py +++ b/keras_nlp/tokenizers/word_piece_tokenizer_trainer_test.py @@ -11,7 +11,6 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. -"""Tests for WordPiece Tokenizer Trainer.""" import os diff --git a/keras_nlp/utils/keras_utils.py b/keras_nlp/utils/keras_utils.py index f81977c57e..6e4d43193c 100644 --- a/keras_nlp/utils/keras_utils.py +++ b/keras_nlp/utils/keras_utils.py @@ -12,7 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. - import sys import tensorflow as tf diff --git a/keras_nlp/utils/pipeline_model.py b/keras_nlp/utils/pipeline_model.py index b2cba03ef8..fa08aaf929 100644 --- a/keras_nlp/utils/pipeline_model.py +++ b/keras_nlp/utils/pipeline_model.py @@ -12,8 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -"""A base class for models including preprocessing.""" - import functools import math diff --git a/keras_nlp/utils/pipeline_model_test.py b/keras_nlp/utils/pipeline_model_test.py index 586e6ef8c9..4c7c7f1964 100644 --- a/keras_nlp/utils/pipeline_model_test.py +++ b/keras_nlp/utils/pipeline_model_test.py @@ -11,6 +11,7 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. + import os import numpy as np