@@ -98,17 +98,19 @@ public void setUp()
98
98
@ After
99
99
public void tearDown () throws InterruptedException , ExecutionException , IOException {
100
100
// 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 );
108
109
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
+ }
112
114
}
113
115
System .out .flush ();
114
116
System .setOut (originalPrintStream );
0 commit comments