Merge remote-tracking branch 'upstream/master' into serializability_tests

Author: ashutosh1919

docs/tutorials/_toc.yaml

@@ -18,3 +18,7 @@ toc:
   path: /addons/tutorials/tqdm_progress_bar
 - title: Seq2Seq for Translation
   path: /addons/tutorials/networks_seq2seq_nmt
+- title: Moving Average Optimizer Checkpoint
+  path: /addons/tutorials/average_optimizers_callback
+- title: Time Stopping Callback
+  path: /addons/tutorials/time_stopping

tensorflow_addons/losses/npairs.py

@@ -14,22 +14,12 @@
 # ==============================================================================
 """Implements npairs loss."""
 
-import platform
-
 import tensorflow as tf
 
 from tensorflow_addons.utils.types import TensorLike
 from typeguard import typechecked
 
 
-def check_if_windows(name):
-    if platform.system() != "Windows":
-        return
-    raise NotImplementedError(
-        "The function {} is not yet available on Windows.".format(name)
-    )
-
-
 @tf.keras.utils.register_keras_serializable(package="Addons")
 @tf.function
 def npairs_loss(y_true: TensorLike, y_pred: TensorLike) -> tf.Tensor:
@@ -59,7 +49,6 @@ def npairs_loss(y_true: TensorLike, y_pred: TensorLike) -> tf.Tensor:
     Returns:
       npairs_loss: float scalar.
     """
-    check_if_windows("npairs_loss")
     y_pred = tf.convert_to_tensor(y_pred)
     y_true = tf.cast(y_true, y_pred.dtype)
 
@@ -119,7 +108,6 @@ def npairs_multilabel_loss(y_true: TensorLike, y_pred: TensorLike) -> tf.Tensor:
     Returns:
       npairs_multilabel_loss: float scalar.
     """
-    check_if_windows("npairs_multilabel_loss")
     y_pred = tf.convert_to_tensor(y_pred)
     y_true = tf.cast(y_true, y_pred.dtype)
 

tensorflow_addons/losses/tests/npairs_test.py

@@ -14,20 +14,10 @@
 # ==============================================================================
 """Tests for npairs loss."""
 
-import platform
-
 import numpy as np
-import pytest
 import tensorflow as tf
 from tensorflow_addons.losses import npairs
 
-IS_WINDOWS = platform.system() == "Windows"
-
-pytestmark = pytest.mark.skipif(
-    IS_WINDOWS,
-    reason="Doesn't work on Windows, see https://github.com/tensorflow/addons/issues/838",
-)
-
 
 def test_config():
     nl_obj = npairs.NpairsLoss(name="nl")