diff --git a/tensorflow_addons/losses/npairs.py b/tensorflow_addons/losses/npairs.py index fb7872c29d..1693123505 100644 --- a/tensorflow_addons/losses/npairs.py +++ b/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) diff --git a/tensorflow_addons/losses/tests/npairs_test.py b/tensorflow_addons/losses/tests/npairs_test.py index e1abb72234..22cf5a4f67 100644 --- a/tensorflow_addons/losses/tests/npairs_test.py +++ b/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")