Skip to content

Commit 4f1b559

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
fix: Include tensorboard profiler requirements when installing google-cloud-aiplatform[tensorboard]
PiperOrigin-RevId: 642033215
1 parent c222f4b commit 4f1b559

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

setup.py

+9-6
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,15 @@
4646
if package.startswith("google.cloud.aiplatform.vertex_ray")
4747
]
4848

49-
tensorboard_extra_require = ["tensorflow >=2.3.0, <3.0.0dev; python_version<='3.11'"]
49+
profiler_extra_require = [
50+
"tensorboard-plugin-profile >= 2.4.0, <3.0.0dev",
51+
"werkzeug >= 2.0.0, <2.1.0dev",
52+
"tensorflow >=2.4.0, <3.0.0dev",
53+
]
54+
tensorboard_extra_require = [
55+
"tensorflow >=2.3.0, <3.0.0dev; python_version<='3.11'"
56+
] + profiler_extra_require
57+
5058
metadata_extra_require = ["pandas >= 1.0.0", "numpy>=1.15.0"]
5159
xai_extra_require = ["tensorflow >=2.3.0, <3.0.0dev"]
5260
lit_extra_require = [
@@ -55,11 +63,6 @@
5563
"lit-nlp == 0.4.0",
5664
"explainable-ai-sdk >= 1.0.0",
5765
]
58-
profiler_extra_require = [
59-
"tensorboard-plugin-profile >= 2.4.0, <3.0.0dev",
60-
"werkzeug >= 2.0.0, <2.1.0dev",
61-
"tensorflow >=2.4.0, <3.0.0dev",
62-
]
6366
featurestore_extra_require = [
6467
"google-cloud-bigquery-storage",
6568
"pandas >= 1.0.0",

0 commit comments

Comments
 (0)