Remove file level docstrings

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
 

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

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

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 (

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

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
 

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

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
 

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

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
 

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

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

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

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

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

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

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

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
 

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

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
 

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

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
 

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

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
 

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

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
 

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

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
 

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

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
 

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
 

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

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
 

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

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

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

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
 

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

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
 

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
 

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

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
 

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
 

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

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

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

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