This CL temporarily hardcodes the version of keras to test the TB nightly

vkarampudi · Responsible ML Infra Team · commit c03ee55d9cca · 2025-01-20T13:37:56.000-08:00
PiperOrigin-RevId: 717617353
diff --git a/tensorboard_plugin/setup.py b/tensorboard_plugin/setup.py
@@ -45,6 +45,7 @@ def select_constraint(default, nightly=None, git_master=None):
     'protobuf>=3.20.3,<5',
     'tensorboard>=2.16.2,<2.17.0',
     'tensorflow>=2.16,<2.17',
+    'tf-keras>=2.16,<2.17',
     'tensorflow-model-analysis>=0.47,<0.48',
     'werkzeug<2',
 ]

Original file line number	Diff line number	Diff line change
`@@ -45,6 +45,7 @@ def select_constraint(default, nightly=None, git_master=None):`
`45`	`45`	`'protobuf>=3.20.3,<5',`
`46`	`46`	`'tensorboard>=2.16.2,<2.17.0',`
`47`	`47`	`'tensorflow>=2.16,<2.17',`
	`48`	`+ 'tf-keras>=2.16,<2.17',`
`48`	`49`	`'tensorflow-model-analysis>=0.47,<0.48',`
`49`	`50`	`'werkzeug<2',`
`50`	`51`	`]`