Support TFX 1.10 (#187)

* Initial release candidate testing for tfx==1.10.0rc0 * remove suffux from test_utils * add missing extra line * use tfx 1.10 constraint * add firebase_publisher to readme
tensorflow · Oct 5, 2022 · c428694 · c428694
1 parent 95c413a
commit c428694
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -53,6 +53,7 @@ tfxa.feast_examplegen.FeastExampleGen(...)
 * [tfxa.sampling](https://github.com/tensorflow/tfx-addons/tree/main/tfx_addons/sampling)
 * [tfxa.message_exit_handler](https://github.com/tensorflow/tfx-addons/tree/main/tfx_addons/message_exit_handler) 
 * [tfxa.pandas_transform](https://github.com/tensorflow/tfx-addons/tree/main/tfx_addons/pandas_transform) 
+* [tfxa.firebase_publisher](https://github.com/tensorflow/tfx-addons/tree/main/tfx_addons/firebase_publisher) 
 
 Check out [proposals](https://github.com/tensorflow/tfx-addons/tree/main/proposals) for a list of existing or upcoming projects proposals for TFX Addons.
 

diff --git a/tfx_addons/utils/test_utils.py b/tfx_addons/utils/test_utils.py
@@ -20,5 +20,7 @@
 def get_tfx_version(version: str) -> List[int]:
   """
     Returns the TFX version as integers.
-    """
+  """
+  # NB(gcasassaez): Remove suffix to avoid parsing issues
+  version = version.split("-")[0]
   return tuple([int(x) for x in version.split(".")])  # pylint: disable=R1728
diff --git a/tfx_addons/version.py b/tfx_addons/version.py
@@ -33,10 +33,10 @@
 
 # Required TFX version [min, max)
 _INCLUSIVE_MIN_TFX_VERSION = "1.4.0"
-_EXCLUSIVE_MAX_TFX_VERSION = "1.10.0"
+_EXCLUSIVE_MAX_TFX_VERSION = "1.11.0"
 _TFXVERSION_CONSTRAINT = (
     f">={_INCLUSIVE_MIN_TFX_VERSION},<{_EXCLUSIVE_MAX_TFX_VERSION}")
-_CI_MAX_CONSTRAINTS = ["tfx~=1.9.0", "tensorflow~=2.9.0"]
+_CI_MAX_CONSTRAINTS = ["tfx~=1.10.0", "tensorflow~=2.9.0"]
 _CI_MIN_CONSTRAINTS = [
     f"tfx~={_INCLUSIVE_MIN_TFX_VERSION}", "tensorflow~=2.6.0",
     "apache-beam[gcp]<2.35", "firebase-admin<5.0.3"