def test_delete_tensorboard_run(self, delete_tensorboard_run_mock):
aiplatform.init(project=_TEST_PROJECT)
my_tensorboard_run = tensorboard.TensorboardRun(
tensorboard_run_name=_TEST_TENSORBOARD_RUN_NAME)
my_tensorboard_run.delete()
delete_tensorboard_run_mock.assert_called_once_with(
name=my_tensorboard_run.resource_name)
def test_init_tensorboard_run_with_tensorboard_and_experiment(
self, get_tensorboard_run_mock):
aiplatform.init(project=_TEST_PROJECT)
tensorboard.TensorboardRun(
tensorboard_run_name=_TEST_TENSORBOARD_RUN_ID,
tensorboard_experiment_id=_TEST_TENSORBOARD_EXPERIMENT_ID,
tensorboard_id=_TEST_ID,
)
get_tensorboard_run_mock.assert_called_once_with(
name=_TEST_TENSORBOARD_RUN_NAME, retry=base._DEFAULT_RETRY)
def test_init_tensorboard_run(self, get_tensorboard_run_mock):
aiplatform.init(project=_TEST_PROJECT)
tensorboard.TensorboardRun(tensorboard_run_name=_TEST_TENSORBOARD_RUN_NAME)
get_tensorboard_run_mock.assert_called_once_with(
name=_TEST_TENSORBOARD_RUN_NAME, retry=base._DEFAULT_RETRY
)