Expose tfa.types doc

tensorflow_addons/__init__.py

@@ -29,5 +29,6 @@
 from tensorflow_addons import seq2seq
 from tensorflow_addons import text
 from tensorflow_addons.register import register_all
+from tensorflow_addons.utils import types
 
 from tensorflow_addons.version import __version__

tensorflow_addons/image/__init__.py

@@ -34,6 +34,9 @@
 from tensorflow_addons.image.transform_ops import shear_x
 from tensorflow_addons.image.transform_ops import shear_y
 from tensorflow_addons.image.sparse_image_warp import sparse_image_warp
+from tensorflow_addons.image.transform_ops import compose_transforms
+from tensorflow_addons.image.transform_ops import angles_to_projective_transforms
 from tensorflow_addons.image.transform_ops import transform
 from tensorflow_addons.image.translate_ops import translate
 from tensorflow_addons.image.translate_ops import translate_xy
+from tensorflow_addons.image.translate_ops import translations_to_projective_transforms

tensorflow_addons/metrics/__init__.py

@@ -16,7 +16,11 @@
 
 from tensorflow_addons.metrics.cohens_kappa import CohenKappa
 from tensorflow_addons.metrics.f_scores import F1Score, FBetaScore
-from tensorflow_addons.metrics.hamming import HammingLoss, hamming_distance
+from tensorflow_addons.metrics.hamming import (
+    HammingLoss,
+    hamming_distance,
+    hamming_loss_fn,
+)
 from tensorflow_addons.metrics.utils import MeanMetricWrapper
 from tensorflow_addons.metrics.matthews_correlation_coefficient import (
     MatthewsCorrelationCoefficient,

tensorflow_addons/optimizers/__init__.py

@@ -42,4 +42,7 @@
 from tensorflow_addons.optimizers.weight_decay_optimizers import (
     extend_with_decoupled_weight_decay,
 )
+from tensorflow_addons.optimizers.weight_decay_optimizers import (
+    DecoupledWeightDecayExtension,
+)
 from tensorflow_addons.optimizers.yogi import Yogi

tensorflow_addons/text/__init__.py

@@ -15,6 +15,8 @@
 """Additional text-processing ops."""
 
 # Conditional Random Field
+from tensorflow_addons.text import crf
+from tensorflow_addons.text.crf import CrfDecodeForwardRnnCell
 from tensorflow_addons.text.crf import crf_binary_score
 from tensorflow_addons.text.crf import crf_constrained_decode
 from tensorflow_addons.text.crf import crf_decode

tools/docs/build_docs.py

@@ -74,8 +74,11 @@ def main(argv):
         py_modules=[(PROJECT_SHORT_NAME, tfa)],
         code_url_prefix=code_url_prefix,
         private_map={"tfa": ["__version__", "utils", "version"]},
-        # This callback usually cleans up a lot of aliases caused by internal imports.
-        callbacks=[public_api.local_definitions_filter],
+        # These callbacks usually clean up a lot of aliases caused by internal imports.
+        callbacks=[
+            public_api.local_definitions_filter,
+            public_api.explicit_package_contents_filter,
+        ],
         search_hints=FLAGS.search_hints,
         site_path=FLAGS.site_path,
     )

tools/install_deps/doc_requirements.txt

@@ -1,2 +1,2 @@
-git+https://github.com/tensorflow/docs@8192df1
+git+https://github.com/tensorflow/docs
 pyyaml