Skip to content

Commit a5960a4

Browse files
test: Add null check in teardown (#752)
* test: Add null check in teardown * test: Remove extra exception type * 🦉 Updates from OwlBot See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md * chore: Updated shared-configuration in pom.xml * chore: revert changes to auto-generated pom.xml files * chore: update changes to auto-generated pom.xml files Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com>
1 parent 56939de commit a5960a4

File tree

1 file changed

+12
-10
lines changed

1 file changed

+12
-10
lines changed

aiplatform/snippets/src/test/java/aiplatform/ImportDataVideoObjectTrackingSampleTest.java

+12-10
Original file line numberDiff line numberDiff line change
@@ -98,17 +98,19 @@ public void setUp()
9898
@After
9999
public void tearDown() throws InterruptedException, ExecutionException, IOException {
100100
// delete the temp dataset
101-
DatasetServiceSettings datasetServiceSettings =
102-
DatasetServiceSettings.newBuilder()
103-
.setEndpoint("us-central1-aiplatform.googleapis.com:443")
104-
.build();
105-
try (DatasetServiceClient datasetServiceClient =
106-
DatasetServiceClient.create(datasetServiceSettings)) {
107-
DatasetName datasetName = DatasetName.of(PROJECT, LOCATION, datasetId);
101+
if (datasetId != null) {
102+
DatasetServiceSettings datasetServiceSettings =
103+
DatasetServiceSettings.newBuilder()
104+
.setEndpoint("us-central1-aiplatform.googleapis.com:443")
105+
.build();
106+
try (DatasetServiceClient datasetServiceClient =
107+
DatasetServiceClient.create(datasetServiceSettings)) {
108+
DatasetName datasetName = DatasetName.of(PROJECT, LOCATION, datasetId);
108109

109-
OperationFuture<Empty, DeleteOperationMetadata> operationFuture =
110-
datasetServiceClient.deleteDatasetAsync(datasetName);
111-
operationFuture.get();
110+
OperationFuture<Empty, DeleteOperationMetadata> operationFuture =
111+
datasetServiceClient.deleteDatasetAsync(datasetName);
112+
operationFuture.get();
113+
}
112114
}
113115
System.out.flush();
114116
System.setOut(originalPrintStream);

0 commit comments

Comments
 (0)