diff --git a/samples/model-builder/experiment_tracking/get_experiment_backing_tensorboard_sample.py b/samples/model-builder/experiment_tracking/get_experiment_backing_tensorboard_sample.py index f1ef99bafe..2551eb527c 100644 --- a/samples/model-builder/experiment_tracking/get_experiment_backing_tensorboard_sample.py +++ b/samples/model-builder/experiment_tracking/get_experiment_backing_tensorboard_sample.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2024 Google LLC # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -12,11 +12,11 @@ # See the License for the specific language governing permissions and # limitations under the License. -# [START aiplatform_sdk_create_experiment_default_tensorboard_sample] +# [START aiplatform_sdk_get_experiment_backing_tensorboard_sample] from google.cloud import aiplatform -def get_experiment_backing_tensorboard_resource_name_sample( +def get_experiment_backing_tensorboard_sample( experiment_name: str, project: str, location: str, @@ -30,4 +30,4 @@ def get_experiment_backing_tensorboard_resource_name_sample( return backing_tensorboard.name -# [END aiplatform_sdk_create_experiment_default_tensorboard_sample] +# [END aiplatform_sdk_get_experiment_backing_tensorboard_sample] diff --git a/samples/model-builder/experiment_tracking/get_experiment_backing_tensorboard_sample_test.py b/samples/model-builder/experiment_tracking/get_experiment_backing_tensorboard_sample_test.py index 205bf3b739..f66c0b3315 100644 --- a/samples/model-builder/experiment_tracking/get_experiment_backing_tensorboard_sample_test.py +++ b/samples/model-builder/experiment_tracking/get_experiment_backing_tensorboard_sample_test.py @@ -16,11 +16,11 @@ import test_constants as constants -def test_get_experiment_backing_tensorboard_resource_name_sample( +def test_get_experiment_backing_tensorboard_sample( mock_get_experiment, mock_get_backing_tensorboard_resource ): - get_experiment_backing_tensorboard_sample.get_experiment_backing_tensorboard_resource_name_sample( + get_experiment_backing_tensorboard_sample.get_experiment_backing_tensorboard_sample( experiment_name=constants.EXPERIMENT_NAME, project=constants.PROJECT, location=constants.LOCATION,