Skip to content

Commit 138dc1a

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
docs: Add sample to retrieve experiment backing tensorboard resource name
PiperOrigin-RevId: 682406056
1 parent 85a8c41 commit 138dc1a

File tree

2 files changed

+68
-0
lines changed

2 files changed

+68
-0
lines changed
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,33 @@
1+
# Copyright 2022 Google LLC
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# https://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
15+
# [START aiplatform_sdk_create_experiment_default_tensorboard_sample]
16+
from google.cloud import aiplatform
17+
18+
19+
def get_experiment_backing_tensorboard_resource_name_sample(
20+
experiment_name: str,
21+
project: str,
22+
location: str,
23+
):
24+
backing_tensorboard = aiplatform.Experiment(
25+
project=project,
26+
location=location,
27+
experiment_name=experiment_name
28+
).get_backing_tensorboard_resource()
29+
30+
return backing_tensorboard.name
31+
32+
33+
# [END aiplatform_sdk_create_experiment_default_tensorboard_sample]
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,35 @@
1+
# Copyright 2022 Google LLC
2+
#
3+
# Licensed under the Apache License, Version 2.0 (the "License");
4+
# you may not use this file except in compliance with the License.
5+
# You may obtain a copy of the License at
6+
#
7+
# https://www.apache.org/licenses/LICENSE-2.0
8+
#
9+
# Unless required by applicable law or agreed to in writing, software
10+
# distributed under the License is distributed on an "AS IS" BASIS,
11+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12+
# See the License for the specific language governing permissions and
13+
# limitations under the License.
14+
15+
from experiment_tracking import get_experiment_backing_tensorboard_sample
16+
import test_constants as constants
17+
18+
19+
def test_get_experiment_backing_tensorboard_resource_name_sample(
20+
mock_get_experiment, mock_get_backing_tensorboard_resource
21+
):
22+
23+
get_experiment_backing_tensorboard_sample.get_experiment_backing_tensorboard_resource_name_sample(
24+
experiment_name=constants.EXPERIMENT_NAME,
25+
project=constants.PROJECT,
26+
location=constants.LOCATION,
27+
)
28+
29+
mock_get_experiment.assert_called_with(
30+
project=constants.PROJECT,
31+
location=constants.LOCATION,
32+
experiment_name=constants.EXPERIMENT_NAME
33+
)
34+
35+
mock_get_backing_tensorboard_resource.assert_called_once()

0 commit comments

Comments
 (0)