Fix tf-addons for upcoming keras 3 default. (#2858) #6179

Job	Run time
Black Python code format	19s
Flake8	14s
Clang C++ code format	9s
Check that the source code is compliant	1m 4s
Valid build files	1m 53s
Bazel code format	9s
Check that we can build the docs	1m 36s
Check that the pre-commit works	25s
Run the cpu tests in a small python docker image	10m 32s
Fast build to run python-only tests	5m 14s
Notebook format	15s
Notebook lint	22s
	22m 12s

Provide feedback