Update release notes and version for TFMOT release 0.8.1.

 - Make requirements.txt align with setup.py.
 - Mark broken tests (with tf-nightly) as TODOs.

PiperOrigin-RevId: 842903676

Author: Xhark

RELEASE.md

@@ -32,6 +32,14 @@ Keras pruning API:
 
 # TensorFlow Model Optimization next release TBD
 
+# TensorFlow Model Optimization 0.8.1
+
+TFMOT 0.8.1 improves compatibility with numpy v2 and dependencies.
+ * Fix keras importing failures at the colab.
+ * Add required packages to setup.py to match requirements.txt.
+ * Relax absl-py requirements for tensorflow_model_optimization.
+ * Mark compatibility with numpy v2.
+
 # TensorFlow Model Optimization 0.8.0
 
 TFMOT 0.8.0 forces users to use the keras v2 version.

pip_pkg.sh

@@ -34,7 +34,7 @@ mkdir -p "$1"
 DEST=$(dirname "${1}/does_not_exist")
 DEST=$(cd "$DEST" && pwd)
 
-cd bazel-bin/pip_pkg.runfiles/_main
+cd bazel-bin/pip_pkg.runfiles/tensorflow_model_optimization
 
 # Pass through remaining arguments (following the first argument, which
 # specifies the output dir) to setup.py, e.g.,

requirements.txt

@@ -1,8 +1,8 @@
-absl-py>=1.2.0
-numpy~=1.23.0
+absl-py~=1.2
+numpy>=1.23
 six~=1.14
 scipy
-enum34~=1.1
-mock
+enum34~=1.1;python_version<"3.4"
 dm-tree~=0.1.1
+mock
 tf-keras>=2.14.1

tensorflow_model_optimization/BUILD

@@ -1,3 +1,6 @@
+# Placeholder: load py_library
+load("@rules_license//rules:license.bzl", "license")
+
 # Description:
 #
 #  TensorFlow Optimization is a repository for the training-time
@@ -7,15 +10,15 @@
 #  https://github.com/tensorflow/model-optimization
 load("//tensorflow_model_optimization:tensorflow_model_optimization.bzl", "py_strict_binary")
 
-# Placeholder: load py_library
-load("//tools/build_defs/license:license.bzl", "license")
-
 package(
     default_applicable_licenses = [":license"],
     default_visibility = ["//visibility:public"],
 )
 
-license(name = "license")
+license(
+    name = "license",
+    license_kinds = ["@rules_license//licenses/spdx:Apache-2.0"],
+)
 
 licenses(["notice"])
 
@@ -24,7 +27,7 @@ exports_files(["LICENSE"])
 py_library(
     name = "tensorflow_model_optimization",
     srcs = ["__init__.py"],
-    strict_deps = True,
+    # strict_deps = True,
     deps = [
         # distutils dep1,
         "//tensorflow_model_optimization/python",  # buildcleaner: keep

tensorflow_model_optimization/python/BUILD

@@ -7,7 +7,7 @@ licenses(["notice"])
 py_library(
     name = "python",
     srcs = ["__init__.py"],
-    strict_deps = True,
+    # strict_deps = True,
     deps = [
         "//tensorflow_model_optimization/python/core",  # buildcleaner: keep
     ],

tensorflow_model_optimization/python/core/BUILD

@@ -7,7 +7,7 @@ licenses(["notice"])
 py_library(
     name = "core",
     srcs = ["__init__.py"],
-    strict_deps = True,
+    # strict_deps = True,
     deps = [
         "//tensorflow_model_optimization/python/core/api",  # buildcleaner: keep
         "//tensorflow_model_optimization/python/core/clustering",  # buildcleaner: keep
@@ -21,5 +21,5 @@ py_library(
 py_library(
     name = "version",
     srcs = ["version.py"],
-    strict_deps = True,
+    # strict_deps = True,
 )

tensorflow_model_optimization/python/core/api/BUILD

@@ -27,7 +27,7 @@ py_library(
         "sparsity/__init__.py",
         "sparsity/keras/__init__.py",
     ],
-    strict_deps = True,
+    # strict_deps = True,
     deps = [
         "//tensorflow_model_optimization/python/core/clustering/keras:cluster",
         "//tensorflow_model_optimization/python/core/clustering/keras:cluster_config",

tensorflow_model_optimization/python/core/clustering/BUILD

@@ -7,7 +7,7 @@ licenses(["notice"])
 py_library(
     name = "clustering",
     srcs = ["__init__.py"],
-    strict_deps = True,
+    # strict_deps = True,
     deps = [
         "//tensorflow_model_optimization/python/core/clustering/keras",  # buildcleaner: keep
     ],

tensorflow_model_optimization/python/core/clustering/keras/BUILD

@@ -13,7 +13,7 @@ py_library(
     srcs = [
         "__init__.py",
     ],
-    strict_deps = True,
+    # strict_deps = True,
     deps = [
         ":cluster",  # buildcleaner: keep
         ":clustering_callbacks",  # buildcleaner: keep
@@ -24,7 +24,7 @@ py_library(
 py_library(
     name = "cluster",
     srcs = ["cluster.py"],
-    strict_deps = True,
+    # strict_deps = True,
     visibility = ["//visibility:public"],
     deps = [
         ":cluster_config",
@@ -38,14 +38,14 @@ py_library(
 py_library(
     name = "cluster_config",
     srcs = ["cluster_config.py"],
-    strict_deps = True,
+    # strict_deps = True,
     visibility = ["//visibility:public"],
 )
 
 py_library(
     name = "clustering_registry",
     srcs = ["clustering_registry.py"],
-    strict_deps = True,
+    # strict_deps = True,
     visibility = ["//visibility:public"],
     deps = [
         ":clusterable_layer",
@@ -58,7 +58,7 @@ py_library(
 py_library(
     name = "clusterable_layer",
     srcs = ["clusterable_layer.py"],
-    strict_deps = True,
+    # strict_deps = True,
     visibility = ["//visibility:public"],
     deps = [
         # six dep1,
@@ -68,7 +68,7 @@ py_library(
 py_library(
     name = "clustering_centroids",
     srcs = ["clustering_centroids.py"],
-    strict_deps = True,
+    # strict_deps = True,
     visibility = ["//visibility:public"],
     deps = [
         ":cluster_config",
@@ -82,7 +82,7 @@ py_library(
 py_library(
     name = "cluster_wrapper",
     srcs = ["cluster_wrapper.py"],
-    strict_deps = True,
+    # strict_deps = True,
     visibility = ["//visibility:public"],
     deps = [
         ":cluster_config",
@@ -97,7 +97,7 @@ py_library(
 py_library(
     name = "clustering_algorithm",
     srcs = ["clustering_algorithm.py"],
-    strict_deps = True,
+    # strict_deps = True,
     visibility = ["//visibility:public"],
     deps = [
         # six dep1,
@@ -109,7 +109,7 @@ py_library(
 py_library(
     name = "clustering_callbacks",
     srcs = ["clustering_callbacks.py"],
-    strict_deps = True,
+    # strict_deps = True,
     visibility = ["//visibility:public"],
     deps = [
         # tensorflow dep1,
@@ -121,7 +121,7 @@ py_test(
     name = "cluster_test",
     size = "medium",
     srcs = ["cluster_test.py"],
-    strict_deps = False,
+    # strict_deps = False,
     visibility = ["//visibility:public"],
     deps = [
         ":cluster",
@@ -190,7 +190,7 @@ py_test(
     name = "cluster_integration_test",
     size = "medium",
     srcs = ["cluster_integration_test.py"],
-    strict_deps = False,
+    # strict_deps = False,
     visibility = ["//visibility:public"],
     deps = [
         ":cluster",

tensorflow_model_optimization/python/core/clustering/keras/experimental/BUILD

@@ -11,7 +11,7 @@ py_library(
     srcs = [
         "__init__.py",
     ],
-    strict_deps = True,
+    # strict_deps = True,
     deps = [
         ":cluster",  # buildcleaner: keep
     ],
@@ -20,7 +20,7 @@ py_library(
 py_library(
     name = "cluster",
     srcs = ["cluster.py"],
-    strict_deps = True,
+    # strict_deps = True,
     visibility = ["//visibility:public"],
     deps = [
         "//tensorflow_model_optimization/python/core/clustering/keras:cluster",