Remove work in progress API for 0.4.1 release (#765)

Author: mattdangerw

keras_nlp/__init__.py

@@ -23,7 +23,6 @@
 from keras_nlp import layers
 from keras_nlp import metrics
 from keras_nlp import models
-from keras_nlp import samplers
 from keras_nlp import tokenizers
 from keras_nlp import utils
 

keras_nlp/models/__init__.py

@@ -20,8 +20,6 @@
 )
 from keras_nlp.models.albert.albert_preprocessor import AlbertPreprocessor
 from keras_nlp.models.albert.albert_tokenizer import AlbertTokenizer
-from keras_nlp.models.bart.bart_backbone import BartBackbone
-from keras_nlp.models.bart.bart_tokenizer import BartTokenizer
 from keras_nlp.models.bert.bert_backbone import BertBackbone
 from keras_nlp.models.bert.bert_classifier import BertClassifier
 from keras_nlp.models.bert.bert_preprocessor import BertPreprocessor
@@ -48,18 +46,8 @@
 from keras_nlp.models.f_net.f_net_classifier import FNetClassifier
 from keras_nlp.models.f_net.f_net_preprocessor import FNetPreprocessor
 from keras_nlp.models.f_net.f_net_tokenizer import FNetTokenizer
-from keras_nlp.models.gpt2.gpt2_backbone import GPT2Backbone
-from keras_nlp.models.gpt2.gpt2_causal_lm import GPT2CausalLM
-from keras_nlp.models.gpt2.gpt2_causal_lm_preprocessor import (
-    GPT2CausalLMPreprocessor,
-)
-from keras_nlp.models.gpt2.gpt2_preprocessor import GPT2Preprocessor
-from keras_nlp.models.gpt2.gpt2_tokenizer import GPT2Tokenizer
-from keras_nlp.models.opt.opt_backbone import OPTBackbone
-from keras_nlp.models.opt.opt_tokenizer import OPTTokenizer
 from keras_nlp.models.roberta.roberta_backbone import RobertaBackbone
 from keras_nlp.models.roberta.roberta_classifier import RobertaClassifier
-from keras_nlp.models.roberta.roberta_masked_lm import RobertaMaskedLM
 from keras_nlp.models.roberta.roberta_masked_lm_preprocessor import (
     RobertaMaskedLMPreprocessor,
 )

keras_nlp/tests/doc_tests/docstring_test.py

@@ -52,9 +52,8 @@ def test_docstrings():
     runner = unittest.TextTestRunner()
     suite = unittest.TestSuite()
     for module in keras_nlp_modules:
-        # Temporarily stop testing gpt2 & deberta docstrings until we are
-        # exporting the symbols.
-        if "gpt2" in module.__name__ or "deberta_v3" in module.__name__:
+        # Stop testing some docstrings until we are exporting the symbols.
+        if [name in module.__name__ for name in ["bart", "gpt2", "opt"]]:
             continue
         suite.addTest(
             doctest.DocTestSuite(
@@ -98,9 +97,8 @@ def test_fenced_docstrings():
     runner = unittest.TextTestRunner()
     suite = unittest.TestSuite()
     for module in keras_nlp_modules:
-        # Temporarily stop testing gpt2 & deberta docstrings until we are
-        # exporting the symbols.
-        if "gpt2" in module.__name__ or "deberta_v3" in module.__name__:
+        # Stop testing some docstrings until we are exporting the symbols.
+        if [name in module.__name__ for name in ["bart", "gpt2", "opt"]]:
             continue
         # Do not test certain modules.
         if module.__name__ in [