#2066 doctest update layers

.bazelversion

@@ -0,0 +1 @@
+3.1.0

.github/CODEOWNERS

@@ -105,6 +105,10 @@
 /tensorflow_addons/layers/tests/esn_test.py @pedrolarben
 /tensorflow_addons/layers/snake.py @failure-to-thrive
 /tensorflow_addons/layers/tests/snake_test.py @failure-to-thrive
+/tensorflow_addons/layers/stochastic_depth.py @mhstadler @windqaq
+/tensorflow_addons/layers/tests/stochastic_depth_test.py @mhstadler @windqaq
+/tensorflow_addons/layers/noisy_dense.py @leonshams
+/tensorflow_addons/layers/tests/noisy_dense_test.py @leonshams
 
 /tensorflow_addons/losses/contrastive.py @windqaq
 /tensorflow_addons/losses/tests/contrastive_test.py @windqaq
@@ -146,6 +150,8 @@
 /tensorflow_addons/optimizers/tests/conditional_gradient_test.py @pkan2 @lokhande-vishnu
 /tensorflow_addons/optimizers/cyclical_learning_rate.py @raphaelmeudec
 /tensorflow_addons/optimizers/tests/cyclical_learning_rate_test.py @raphaelmeudec
+/tensorflow_addons/optimizers/discriminative_layer_training.py @hyang0129
+/tensorflow_addons/optimizers/tests/discriminative_layer_training_test.py @hyang0129
 /tensorflow_addons/optimizers/lamb.py @junjiek
 /tensorflow_addons/optimizers/tests/lamb_test.py @junjiek
 /tensorflow_addons/optimizers/lazy_adam.py @ssaishruthi

.github/workflows/ci_test.yml

@@ -10,9 +10,6 @@ on:
       - master
       - r*
 
-env:
-  BAZEL_VERSION: 3.1.0
-
 jobs:
   flake8-test:
     name: Flake8
@@ -98,3 +95,32 @@ jobs:
       - run: pip install pygithub click
       - name: Check that the CODEOWNERS is valid
         run: python .github/workflows/notify_codeowners.py .github/CODEOWNERS
+  nbfmt:
+    name: Notebook format
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/setup-python@v1
+    - uses: actions/checkout@v2
+    - name: Install tensorflow-docs
+      run: python3 -m pip install -U git+https://github.com/tensorflow/docs
+    - name: Check notebook formatting
+      run: |
+        # Run on all notebooks to prevent upstream change.
+        echo "Check formatting with nbfmt:"
+        python3 -m tensorflow_docs.tools.nbfmt --test \
+            $(find docs/tutorials/ -type f -name *.ipynb)
+  nblint:
+    name: Notebook lint
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/setup-python@v1
+    - uses: actions/checkout@v2
+    - name: Install tensorflow-docs
+      run: python3 -m pip install -U git+https://github.com/tensorflow/docs
+    - name: Lint notebooks
+      run: |
+        # Run on all notebooks to prevent upstream change.
+        echo "Lint check with nblint:"
+        python3 -m tensorflow_docs.tools.nblint \
+            --arg=repo:tensorflow/addons \
+            $(find docs/tutorials/ -type f -name *.ipynb ! -path "docs/tutorials/_template.ipynb")

.github/workflows/release.yml

@@ -15,7 +15,6 @@ on:
       - r*
 
 env:
-  BAZEL_VERSION: '3.1.0'
   MIN_PY_VERSION: '3.5'
   MAX_PY_VERSION: '3.8'
 
@@ -31,7 +30,7 @@ jobs:
       - name: Build wheels
         run: |
           pip install --default-timeout=1000 -r tools/install_deps/pytest.txt -r tools/install_deps/tensorflow-cpu.txt -r requirements.txt
-          bash tools/install_deps/bazel_linux.sh
+          bash tools/install_deps/install_bazelisk.sh ./
           python configure.py
           bazel test -c opt -k --test_timeout 300,450,1200,3600 --test_output=errors //tensorflow_addons/...
   release-wheel:
@@ -60,11 +59,7 @@ jobs:
       - name: Setup Bazel
         # Ubuntu bazel is run inside of the docker image
         if: matrix.os != 'ubuntu-18.04'
-        # MacOS VMs share IP addr on GH actions. This bazel installation authenticates token to prevent api rate limiting.
-        uses: jwlawson/actions-setup-bazel@784af34 # Version 1.1
-        with:
-          bazel-version: ${{ env.BAZEL_VERSION}}
-          github-api-token: ${{ secrets.GITHUB_TOKEN }}
+        run: bash tools/install_deps/install_bazelisk.sh ./
       - name: Build wheels
         env:
           OS: ${{ runner.os }}

CONTRIBUTING.md

@@ -325,6 +325,24 @@ bazel test -c opt -k \
 //tensorflow_addons/...
 ```
 
+#### Testing docstrings
+
+We use [DocTest](https://docs.python.org/3/library/doctest.html) to test code snippets
+in Python docstrings. The snippet must be executable Python code.
+To enable testing, prepend the line with `>>>` (three left-angle brackets).
+Available namespace include `np` for numpy, `tf` for TensorFlow, and `tfa` for TensorFlow Addons.
+See [docs_ref](https://www.tensorflow.org/community/contribute/docs_ref) for more details.
+
+To test docstrings locally, run either
+```bash
+bash tools/run_cpu_tests.sh
+```
+on all files, or
+```bash
+pytest -v -n auto --durations=25 --doctest-modules /path/to/pyfile
+```
+on specific files.
+
 ## About type hints
 
 Ideally, we would like all the functions and classes constructors exposed in 

README.md

@@ -52,6 +52,15 @@ run the following:
 ```
 pip install tensorflow-addons
 ```
+
+To ensure you have a version of TensorFlow that is compatible with TensorFlow Addons, 
+you can specify the `tensorflow` extra requirement during install:
+
+```
+pip install tensorflow-addons[tensorflow]
+```
+
+Similar extras exist for the `tensorflow-gpu` and `tensorflow-cpu` packages.
 
 
 To use TensorFlow Addons:
@@ -72,7 +81,7 @@ what it was tested against.
 | TensorFlow Addons | TensorFlow | Python  |
 |:----------------------- |:---|:---------- |
 | tfa-nightly | 2.2, 2.3 | 3.5, 3.6, 3.7, 3.8 | 
-| tensorflow-addons-0.11.1 | 2.2, 2.3 |3.5, 3.6, 3.7, 3.8 |
+| tensorflow-addons-0.11.2 | 2.2, 2.3 |3.5, 3.6, 3.7, 3.8 |
 | tensorflow-addons-0.10.0 | 2.2 |3.5, 3.6, 3.7, 3.8 |
 | tensorflow-addons-0.9.1 | 2.1, 2.2 |3.5, 3.6, 3.7 |
 | tensorflow-addons-0.8.3 | 2.1 |3.5, 3.6, 3.7 |
@@ -98,7 +107,7 @@ is compiled differently. A typical example of this would be `conda`-installed Te
 | TensorFlow Addons | TensorFlow | Compiler  | cuDNN | CUDA | 
 |:----------------------- |:---- |:---------|:---------|:---------|
 | tfa-nightly | 2.3 | GCC 7.3.1 | 7.6 | 10.1 |
-| tensorflow-addons-0.11.1 | 2.3  | GCC 7.3.1 | 7.6 | 10.1 |
+| tensorflow-addons-0.11.2 | 2.3  | GCC 7.3.1 | 7.6 | 10.1 |
 | tensorflow-addons-0.10.0 | 2.2  | GCC 7.3.1 | 7.6 | 10.1 |
 | tensorflow-addons-0.9.1 | 2.1  | GCC 7.3.1 | 7.6 | 10.1 |
 | tensorflow-addons-0.8.3 | 2.1  | GCC 7.3.1 | 7.6 | 10.1 |

build_deps/toolchains/gpu/find_cuda_config.py

@@ -277,12 +277,16 @@ def get_nvcc_version(path):
 
     nvcc_name = "nvcc.exe" if _is_windows() else "nvcc"
     nvcc_path, nvcc_version = _find_versioned_file(
-        base_paths, ["", "bin",], nvcc_name, cuda_version, get_nvcc_version
+        base_paths,
+        ["", "bin"],
+        nvcc_name,
+        cuda_version,
+        get_nvcc_version,
     )
 
     nvvm_path = _find_file(
         base_paths,
-        ["nvvm/libdevice", "share/cuda", "lib/nvidia-cuda-toolkit/libdevice",],
+        ["nvvm/libdevice", "share/cuda", "lib/nvidia-cuda-toolkit/libdevice"],
         "libdevice*.10.bc",
     )
 

docs/README.md

@@ -11,23 +11,21 @@ Any markdown or notebook files in this directory will be published to tensorflow
 `tutorials/_toc.yaml` controls the left-nav on the tutorials tab. Make sure to keep that file up to date.
 Notify the tensorflow/docs team if you need to major changes. 
 
+The preferred formatting for TensorFlow notebooks is to use the [tensorflow/docs](https://github.com/tensorflow/docs) [`nbfmt` tool](https://github.com/tensorflow/docs/tree/master/tools/tensorflow_docs/tools). If modifying a tutorial gives you
+an unreadable diff use the following commands to re-apply the standard formatting:  
+
+```
+pip install git+https://github.com/tensorflow/docs
+python -m tensorflow_docs.tools.nbfmt {path to notebook file or directory}
+```
+
+
+
+
+
 
-## 2. Generated API docs
 
-[tensorflow.org/addons/api_docs/python/tfa](https://tensorflow.org/addons/api_docs/python/tfa)
 
-`build_docs.py` controls executed this docs generation. To test-run it:
 
-```bash
-# Install dependencies:
-pip install -r tools/install_deps/doc_requirements.txt
 
-# Build tool:
-bazel build docs:build_docs
 
-# Generate API doc:
-# Use current branch
-bazel-bin/docs/build_docs --git_branch=$(git rev-parse --abbrev-ref HEAD)
-# or specified explicitly
-bazel-bin/docs/build_docs --git_branch=master --output_dir=/tmp/tfa_api
-```