Pin tf_keras to tf or tf_nightly

PiperOrigin-RevId: 607045302

Author: sampathweb

oss_setup.py

@@ -37,6 +37,12 @@
             "`pip_build.py` instead of `setup.py`."
         )
 
+# pin version to that of tensorflow or tf_nightly.
+if "nightly" in "{{PACKAGE}}":
+    install_requires = ["tf-nightly~={{VERSION}}.dev"]
+else:
+    install_requires = ["tensorflow~={{VERSION}}"]
+
 setuptools.setup(
     name="{{PACKAGE}}",
     # Version strings with `-` characters are semver compatible,
@@ -49,7 +55,7 @@
     author="Keras team",
     author_email="keras-users@googlegroups.com",
     packages=setuptools.find_packages(),
-    install_requires=[],
+    install_requires=install_requires,
     # Supported Python versions
     python_requires=">=3.9",
     # PyPI package information.
@@ -63,6 +69,7 @@
         "Programming Language :: Python :: 3.9",
         "Programming Language :: Python :: 3.10",
         "Programming Language :: Python :: 3.11",
+        "Programming Language :: Python :: 3.12",
         "Programming Language :: Python :: 3 :: Only",
         "Topic :: Scientific/Engineering",
         "Topic :: Scientific/Engineering :: Mathematics",

pip_build.py

@@ -449,13 +449,14 @@ def test_wheel(wheel_path, expected_version, requirements_path):
     checks = ";".join(symbols_to_check)
     # Uninstall `keras-nightly` after installing requirements
     # otherwise both will register `experimentalOptimizer` and test will fail.
+    # Skip install deps for `tf_keras` as TensorFlow installed from requirements
     script = (
         "#!/bin/bash\n"
         "virtualenv kenv\n"
         f"source {os.path.join('kenv', 'bin', 'activate')}\n"
         f"pip3 install -r {requirements_path}\n"
         "pip3 uninstall -y keras-nightly\n"
-        f"pip3 install {wheel_path} --force-reinstall\n"
+        f"pip3 install {wheel_path} --force-reinstall --no-deps\n"
         f"python3 -c 'import tf_keras;{checks};print(tf_keras.__version__)'\n"
     )
     try: