fix typo in export (#1024)

keras-team · May 5, 2023 · 1aa01c8 · 1aa01c8
1 parent ee2015a
commit 1aa01c8
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 3 deletions.
diff --git a/keras_nlp/metrics/rouge_n.py b/keras_nlp/metrics/rouge_n.py
@@ -18,7 +18,7 @@
 from keras_nlp.metrics.rouge_base import RougeBase
 
 
-@keras_nlp_export("keras_nlp.metrics.RogueN")
+@keras_nlp_export("keras_nlp.metrics.RougeN")
 class RougeN(RougeBase):
     """ROUGE-N metric.
 

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
@@ -24,7 +24,7 @@
 from keras_nlp.utils.keras_utils import pack_x_y_sample_weight
 
 
-@keras_nlp_export("keras_nlp.models.DistilBertMaskedLMPrerprocessor")
+@keras_nlp_export("keras_nlp.models.DistilBertMaskedLMPreprocessor")
 class DistilBertMaskedLMPreprocessor(DistilBertPreprocessor):
     """DistilBERT preprocessing for the masked language modeling task.
 

diff --git a/keras_nlp/models/roberta/roberta_backbone.py b/keras_nlp/models/roberta/roberta_backbone.py
@@ -33,7 +33,7 @@ def roberta_kernel_initializer(stddev=0.02):
     return keras.initializers.TruncatedNormal(stddev=stddev)
 
 
-@keras_nlp_export("keras_nlp.models.RobertBackbone")
+@keras_nlp_export("keras_nlp.models.RobertaBackbone")
 class RobertaBackbone(Backbone):
     """A RoBERTa encoder network.
 

diff --git a/keras_nlp/tokenizers/sentence_piece_tokenizer_trainer.py b/keras_nlp/tokenizers/sentence_piece_tokenizer_trainer.py
@@ -21,7 +21,10 @@
 except ImportError:
     spm = None
 
+from keras_nlp.api_export import keras_nlp_export
 
+
+@keras_nlp_export("keras_nlp.tokenizers.compute_sentence_piece_proto")
 def compute_sentence_piece_proto(
     data,
     vocabulary_size,

diff --git a/keras_nlp/tokenizers/word_piece_tokenizer_trainer.py b/keras_nlp/tokenizers/word_piece_tokenizer_trainer.py
@@ -15,6 +15,7 @@
 
 import tensorflow as tf
 
+from keras_nlp.api_export import keras_nlp_export
 from keras_nlp.tokenizers.word_piece_tokenizer import pretokenize
 from keras_nlp.utils.tf_utils import assert_tf_text_installed
 
@@ -26,6 +27,7 @@
     learner = None
 
 
+@keras_nlp_export("keras_nlp.tokenizers.compute_word_piece_vocabulary")
 def compute_word_piece_vocabulary(
     data,
     vocabulary_size,