migrate api and fix build issue (#2682)

* migrate api and fix build issue

* fix some bugs

Author: fsx950223

configure.py

@@ -145,6 +145,7 @@ def create_build_configuration():
             write("build --copt=-mavx")
         write("build --cxxopt=-std=c++14")
         write("build --host_cxxopt=-std=c++14")
+        write('build --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0"')
 
     if os.getenv("TF_NEED_CUDA", "0") == "1":
         print("> Building GPU & CPU ops")

requirements.txt

@@ -1 +1,2 @@
 typeguard>=2.7
+packaging

tensorflow_addons/text/tests/crf_test.py

@@ -15,7 +15,7 @@
 """Tests for CRF."""
 
 import itertools
-from distutils.version import LooseVersion
+from packaging.version import Version
 
 import pytest
 import numpy as np
@@ -536,7 +536,7 @@ def call(self, y_true, y_pred):
 
     tensor_name = (
         "tf.math.multiply"
-        if LooseVersion(tf.__version__) >= "2.5.0"
+        if Version(tf.__version__) >= Version("2.5.0")
         else "tf_op_layer_Mul"
     )
     y_data = {tensor_name: np.random.randint(0, 3, (5, 10))}

tensorflow_addons/utils/ensure_tf_install.py

@@ -18,7 +18,7 @@
 # needs to happen before anything else, since the imports below will try to
 # import TensorFlow, too.
 
-from distutils.version import LooseVersion
+from packaging.version import Version
 import warnings
 
 import tensorflow as tf
@@ -44,10 +44,10 @@ def _check_tf_version():
         )
         return
 
-    min_version = LooseVersion(INCLUSIVE_MIN_TF_VERSION)
-    max_version = LooseVersion(EXCLUSIVE_MAX_TF_VERSION)
+    min_version = Version(INCLUSIVE_MIN_TF_VERSION)
+    max_version = Version(EXCLUSIVE_MAX_TF_VERSION)
 
-    if min_version <= LooseVersion(tf.__version__) < max_version:
+    if min_version <= Version(tf.__version__) < max_version:
         return
 
     warnings.warn(

tensorflow_addons/utils/resource_loader.py

@@ -14,7 +14,7 @@
 # ==============================================================================
 """Utilities similar to tf.python.platform.resource_loader."""
 
-from distutils.version import LooseVersion
+from packaging.version import Version
 import os
 import warnings
 
@@ -117,6 +117,6 @@ def abi_is_compatible():
         # tf-nightly
         return False
 
-    min_version = LooseVersion(INCLUSIVE_MIN_TF_VERSION_FOR_ABI_COMPATIBILITY)
-    max_version = LooseVersion(EXCLUSIVE_MAX_TF_VERSION_FOR_ABI_COMPATIBILITY)
-    return min_version <= LooseVersion(tf.__version__) < max_version
+    min_version = Version(INCLUSIVE_MIN_TF_VERSION_FOR_ABI_COMPATIBILITY)
+    max_version = Version(EXCLUSIVE_MAX_TF_VERSION_FOR_ABI_COMPATIBILITY)
+    return min_version <= Version(tf.__version__) < max_version