Fixed code as requested in comments

Author: TetyanaYahodska

tpu/src/main/java/tpu/CreateQueuedResource.java

@@ -17,17 +17,14 @@
 package tpu;
 
 //[START tpu_queued_resources_create]
-import com.google.api.gax.retrying.RetrySettings;
 import com.google.cloud.tpu.v2alpha1.CreateQueuedResourceRequest;
 import com.google.cloud.tpu.v2alpha1.Node;
 import com.google.cloud.tpu.v2alpha1.QueuedResource;
 import com.google.cloud.tpu.v2alpha1.TpuClient;
-import com.google.cloud.tpu.v2alpha1.TpuSettings;
 import java.io.IOException;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.TimeoutException;
-import org.threeten.bp.Duration;
 
 public class CreateQueuedResource {
   public static void main(String[] args)
@@ -59,26 +56,11 @@ public static void main(String[] args)
   public static QueuedResource createQueuedResource(String projectId, String zone,
       String queuedResourceId, String nodeName, String tpuType, String tpuSoftwareVersion)
           throws IOException, ExecutionException, InterruptedException, TimeoutException {
-    // With these settings the client library handles the Operation's polling mechanism
-    // and prevent CancellationException error
-    TpuSettings.Builder clientSettings =
-        TpuSettings.newBuilder();
-    clientSettings
-        .createQueuedResourceSettings()
-        .setRetrySettings(
-            RetrySettings.newBuilder()
-                .setInitialRetryDelay(Duration.ofMillis(5000L))
-                .setRetryDelayMultiplier(2.0)
-                .setInitialRpcTimeout(Duration.ZERO)
-                .setRpcTimeoutMultiplier(1.0)
-                .setMaxRetryDelay(Duration.ofMillis(45000L))
-                .setTotalTimeout(Duration.ofHours(24L))
-                .build());
     String resource = String.format("projects/%s/locations/%s/queuedResources/%s",
             projectId, zone, queuedResourceId);
     // Initialize client that will be used to send requests. This client only needs to be created
     // once, and can be reused for multiple requests.
-    try (TpuClient tpuClient = TpuClient.create(clientSettings.build())) {
+    try (TpuClient tpuClient = TpuClient.create()) {
       String parent = String.format("projects/%s/locations/%s", projectId, zone);
       Node node =
           Node.newBuilder()
@@ -100,9 +82,6 @@ public static QueuedResource createQueuedResource(String projectId, String zone,
                               .setNodeId(nodeName)
                               .build())
                       .build())
-              // You can request a queued resource using a reservation by specifying it in code
-              //.setReservationName(
-              // "projects/YOUR_PROJECT_ID/locations/YOUR_ZONE/reservations/YOUR_RESERVATION_NAME")
               .build();
 
       CreateQueuedResourceRequest request =

tpu/src/main/java/tpu/DeleteQueuedResource.java

@@ -17,13 +17,10 @@
 package tpu;
 
 //[START tpu_queued_resources_delete]
-import com.google.api.gax.retrying.RetrySettings;
 import com.google.cloud.tpu.v2alpha1.DeleteQueuedResourceRequest;
 import com.google.cloud.tpu.v2alpha1.TpuClient;
-import com.google.cloud.tpu.v2alpha1.TpuSettings;
 import java.io.IOException;
 import java.util.concurrent.ExecutionException;
-import org.threeten.bp.Duration;
 
 public class DeleteQueuedResource {
   public static void main(String[] args)
@@ -44,36 +41,17 @@ public static void deleteQueuedResource(String projectId, String zone, String qu
       throws ExecutionException, InterruptedException, IOException {
     String name = String.format("projects/%s/locations/%s/queuedResources/%s",
         projectId, zone, queuedResourceId);
-    // With these settings the client library handles the Operation's polling mechanism
-    // and prevent CancellationException error
-    TpuSettings.Builder clientSettings =
-        TpuSettings.newBuilder();
-    clientSettings
-        .deleteQueuedResourceSettings()
-        .setRetrySettings(
-            RetrySettings.newBuilder()
-                .setInitialRetryDelay(Duration.ofMillis(5000L))
-                .setRetryDelayMultiplier(2.0)
-                .setInitialRpcTimeout(Duration.ZERO)
-                .setRpcTimeoutMultiplier(1.0)
-                .setMaxRetryDelay(Duration.ofMillis(45000L))
-                .setTotalTimeout(Duration.ofHours(24L))
-                .build());
     // Initialize client that will be used to send requests. This client only needs to be created
     // once, and can be reused for multiple requests.
-    try (TpuClient tpuClient = TpuClient.create(clientSettings.build())) {
+    try (TpuClient tpuClient = TpuClient.create()) {
       // Before deleting the queued resource it is required to delete the TPU VM.
       // For more information about deleting TPU
       // see https://cloud.google.com/tpu/docs/managing-tpus-tpu-vm
 
       DeleteQueuedResourceRequest request =
               DeleteQueuedResourceRequest.newBuilder().setName(name).build();
 
-      // Waiting for updates in the library. Until then, the operation will complete successfully,
-      // but the user will receive an error message with UnknownException and IllegalStateException.
       tpuClient.deleteQueuedResourceAsync(request).get();
-
-      System.out.printf("Deleted Queued Resource: %s\n", name);
     }
   }
 }

tpu/src/test/java/tpu/QueuedResourceIT.java

@@ -59,8 +59,7 @@ public void testCreateQueuedResource() throws Exception {
       TpuClient mockTpuClient = mock(TpuClient.class);
       OperationFuture mockFuture = mock(OperationFuture.class);
 
-      mockedTpuClient.when(() -> TpuClient.create(any(TpuSettings.class)))
-              .thenReturn(mockTpuClient);
+      mockedTpuClient.when(TpuClient::create).thenReturn(mockTpuClient);
       when(mockTpuClient.createQueuedResourceAsync(any(CreateQueuedResourceRequest.class)))
               .thenReturn(mockFuture);
       when(mockFuture.get(anyLong(), any(TimeUnit.class))).thenReturn(mockQueuedResource);
@@ -147,8 +146,7 @@ public void testDeleteQueuedResource()
       TpuClient mockTpuClient = mock(TpuClient.class);
       OperationFuture mockFuture = mock(OperationFuture.class);
 
-      mockedTpuClient.when(() -> TpuClient.create(any(TpuSettings.class)))
-              .thenReturn(mockTpuClient);
+      mockedTpuClient.when(TpuClient::create).thenReturn(mockTpuClient);
       when(mockTpuClient.deleteQueuedResourceAsync(any(DeleteQueuedResourceRequest.class)))
               .thenReturn(mockFuture);
       when(mockFuture.get()).thenReturn(null);