Upgrade TFMD to align with TF 2.21.0 and Added Python 3.12/3.13 support (#63)

* Align TFMD with TF 2.21 and Python 3.12

* Update RELEASE.md for TF 2.21 and Python 3.12

* Added python 3.13 updated to RELEASE.md

Author: vkarampudi

.github/workflows/test.yml

@@ -9,15 +9,15 @@ on:
   workflow_dispatch:
 
 env:
-  USE_BAZEL_VERSION: "6.5.0"
+  USE_BAZEL_VERSION: "7.7.0"
 
 jobs:
   tests:
     runs-on: ${{ matrix.os }}
     strategy:
       matrix:
         os: [ubuntu-latest, macos-latest]
-        python-version: ['3.9', '3.10', '3.11', '3.12', '3.13']
+        python-version: ['3.10', '3.11', '3.12', '3.13']
 
     steps:
     - uses: actions/checkout@v4

.github/workflows/wheels.yml

@@ -7,7 +7,7 @@ on:
     types: [published]
 
 env:
-  USE_BAZEL_VERSION: "6.5.0"
+  USE_BAZEL_VERSION: "7.7.0"
 
 
 jobs:
@@ -19,7 +19,7 @@ jobs:
       # fail-fast: true
       matrix:
         os: [ubuntu, macos]
-        python-version: ['3.9', '3.10', '3.11', '3.12', '3.13']
+        python-version: ['3.10', '3.11', '3.12', '3.13']
 
 
     runs-on: ${{ format('{0}-latest', matrix.os) }}

RELEASE.md

@@ -6,8 +6,15 @@
 
 ## Bug Fixes and Other Changes
 
+*   Align with TensorFlow 2.21 dependencies.
+*   Upgrade Protobuf to `6.31.1`.
+*   Upgrade `bazel_skylib` to `1.9.0`.
+*   Add support for Python 3.12 and 3.13.
+
 ## Breaking Changes
 
+*   Drop support for Python 3.9.
+
 ## Deprecations
 
 # Version 1.17.3

WORKSPACE

@@ -4,21 +4,19 @@ load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
 
 http_archive(
     name = "bazel_skylib",
-    sha256 = "97e70364e9249702246c0e9444bccdc4b847bed1eb03c5a3ece4f83dfe6abc44",
     urls = [
-        "https://mirror.bazel.build/github.com/bazelbuild/bazel-skylib/releases/download/1.0.2/bazel-skylib-1.0.2.tar.gz",
-        "https://github.com/bazelbuild/bazel-skylib/releases/download/1.0.2/bazel-skylib-1.0.2.tar.gz",
+        "https://github.com/bazelbuild/bazel-skylib/releases/download/1.9.0/bazel-skylib-1.9.0.tar.gz",
     ],
 )
 
-_PROTOBUF_VERSION = "4.25.6"
+_PROTOBUF_VERSION = "6.31.1"
 
 http_archive(
     name = "com_google_protobuf",
-    sha256 = "ff6e9c3db65f985461d200c96c771328b6186ee0b10bc7cb2bbc87cf02ebd864",
+    sha256 = "6e09bbc950ba60c3a7b30280210cd285af8d7d8ed5e0a6ed101c72aff22e8d88",
     strip_prefix = "protobuf-%s" % _PROTOBUF_VERSION,
     urls = [
-        "https://github.com/protocolbuffers/protobuf/archive/v%s.zip" % _PROTOBUF_VERSION,
+        "https://github.com/protocolbuffers/protobuf/archive/refs/tags/v%s.zip" % _PROTOBUF_VERSION,
     ],
 )
 
@@ -45,8 +43,22 @@ bind(
     actual = "@six_archive//:six",
 )
 
+http_archive(
+    name = "rules_python",
+    urls = ["https://github.com/bazelbuild/rules_python/archive/refs/tags/0.31.0.tar.gz"],
+    strip_prefix = "rules_python-0.31.0",
+)
+
+load("@rules_python//python:repositories.bzl", "py_repositories")
+py_repositories()
+
+local_repository(
+    name = "compatibility_proxy",
+    path = "third_party/dummy_compatibility_proxy",
+)
+
 load("@com_google_protobuf//:protobuf_deps.bzl", "protobuf_deps")
 protobuf_deps()
 
 load("@bazel_skylib//lib:versions.bzl", "versions")
-versions.check("6.5.0")
+versions.check("7.4.1")

setup.py

@@ -121,7 +121,6 @@ def run(self):
         "Operating System :: OS Independent",
         "Programming Language :: Python",
         "Programming Language :: Python :: 3",
-        "Programming Language :: Python :: 3.9",
         "Programming Language :: Python :: 3.10",
         "Programming Language :: Python :: 3.11",
         "Programming Language :: Python :: 3.12",
@@ -140,7 +139,7 @@ def run(self):
         'protobuf>=4.25.2;python_version>="3.11"',
         'protobuf>=4.21.6,<=6.32;python_version<"3.11"',
     ],
-    python_requires=">=3.9,<4",
+    python_requires=">=3.10,<4",
     packages=find_packages(),
     extras_require={
         "dev": ["precommit"],

third_party/dummy_compatibility_proxy/BUILD

@@ -0,0 +1,4 @@
+JavaInfo = provider()
+java_common = struct()
+java_common_internal_compile = struct()
+java_info_internal_merge = struct()