Remove references to internal symbols in public docstrings.

PiperOrigin-RevId: 379414888
dgrahn · Jun 15, 2021 · 0eb1954 · 0eb1954
1 parent c0fb505
commit 0eb1954
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 7 deletions.
diff --git a/keras/layers/preprocessing/index_lookup.py b/keras/layers/preprocessing/index_lookup.py
@@ -133,7 +133,7 @@ def __init__(self,
                oov_token,
                vocabulary=None,
                invert=False,
-               output_mode=INT,
+               output_mode="int",
                sparse=False,
                pad_to_max_tokens=False,
                **kwargs):

diff --git a/keras/layers/preprocessing/integer_lookup.py b/keras/layers/preprocessing/integer_lookup.py
@@ -265,7 +265,7 @@ def __init__(self,
                oov_token=-1,
                vocabulary=None,
                invert=False,
-               output_mode=index_lookup.INT,
+               output_mode="int",
                sparse=False,
                pad_to_max_tokens=False,
                **kwargs):

diff --git a/keras/layers/preprocessing/string_lookup.py b/keras/layers/preprocessing/string_lookup.py
@@ -263,7 +263,7 @@ def __init__(self,
                vocabulary=None,
                encoding=None,
                invert=False,
-               output_mode=index_lookup.INT,
+               output_mode="int",
                sparse=False,
                pad_to_max_tokens=False,
                **kwargs):

diff --git a/keras/layers/preprocessing/text_vectorization.py b/keras/layers/preprocessing/text_vectorization.py
@@ -217,15 +217,14 @@ class TextVectorization(base_preprocessing_layer.CombinerPreprocessingLayer):
 
   def __init__(self,
                max_tokens=None,
-               standardize=LOWER_AND_STRIP_PUNCTUATION,
-               split=SPLIT_ON_WHITESPACE,
+               standardize="lower_and_strip_punctuation",
+               split="whitespace",
                ngrams=None,
-               output_mode=INT,
+               output_mode="int",
                output_sequence_length=None,
                pad_to_max_tokens=False,
                vocabulary=None,
                **kwargs):
-
     # This layer only applies to string processing, and so should only have
     # a dtype of 'string'.
     if "dtype" in kwargs and kwargs["dtype"] != tf.string: