fix(tpu): Updating TPU samples to avoid NameError

Author: m-strzelczyk

tpu/queued_resources_create.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 import os
 
-from google.cloud.tpu_v2alpha1 import CreateQueuedResourceRequest, Node
+from google.cloud.tpu_v2alpha1 import Node
 
 
 def create_queued_resource(
@@ -49,7 +49,7 @@ def create_queued_resource(
     resource = tpu_v2alpha1.QueuedResource()
     resource.tpu = tpu_v2alpha1.QueuedResource.Tpu(node_spec=[node_spec])
 
-    request = CreateQueuedResourceRequest(
+    request = tpu_v2alpha1.CreateQueuedResourceRequest(
         parent=f"projects/{project_id}/locations/{zone}",
         queued_resource_id=queued_resource_name,
         queued_resource=resource,

tpu/queued_resources_create_network.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 import os
 
-from google.cloud.tpu_v2alpha1 import CreateQueuedResourceRequest, Node
+from google.cloud.tpu_v2alpha1 import Node
 
 
 def create_queued_resource_network(
@@ -56,7 +56,7 @@ def create_queued_resource_network(
     resource = tpu_v2alpha1.QueuedResource()
     resource.tpu = tpu_v2alpha1.QueuedResource.Tpu(node_spec=[node_spec])
 
-    request = CreateQueuedResourceRequest(
+    request = tpu_v2alpha1.CreateQueuedResourceRequest(
         parent=f"projects/{project_id}/locations/{zone}",
         queued_resource_id=queued_resource_name,
         queued_resource=resource,

tpu/queued_resources_create_spot.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 import os
 
-from google.cloud.tpu_v2alpha1 import CreateQueuedResourceRequest, Node
+from google.cloud.tpu_v2alpha1 import Node
 
 
 def create_queued_resource_spot(
@@ -51,7 +51,7 @@ def create_queued_resource_spot(
     # Create a spot resource
     resource.spot = tpu_v2alpha1.QueuedResource.Spot()
 
-    request = CreateQueuedResourceRequest(
+    request = tpu_v2alpha1.CreateQueuedResourceRequest(
         parent=f"projects/{project_id}/locations/{zone}",
         queued_resource_id=queued_resource_name,
         queued_resource=resource,

tpu/queued_resources_create_startup_script.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 import os
 
-from google.cloud.tpu_v2alpha1 import CreateQueuedResourceRequest, Node
+from google.cloud.tpu_v2alpha1 import Node
 
 
 def create_queued_resource_startup_script(
@@ -61,7 +61,7 @@ def create_queued_resource_startup_script(
     resource = tpu_v2alpha1.QueuedResource()
     resource.tpu = tpu_v2alpha1.QueuedResource.Tpu(node_spec=[node_spec])
 
-    request = CreateQueuedResourceRequest(
+    request = tpu_v2alpha1.CreateQueuedResourceRequest(
         parent=f"projects/{project_id}/locations/{zone}",
         queued_resource_id=queued_resource_name,
         queued_resource=resource,

tpu/queued_resources_create_time_bound.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 import os
 
-from google.cloud.tpu_v2alpha1 import CreateQueuedResourceRequest, Node
+from google.cloud.tpu_v2alpha1 import Node
 
 
 def create_queued_resource_time_bound(
@@ -57,7 +57,7 @@ def create_queued_resource_time_bound(
         # valid_until_time="2024-10-29T16:00:00Z",  # Specify a time before which the resource should be allocated
     )
 
-    request = CreateQueuedResourceRequest(
+    request = tpu_v2alpha1.CreateQueuedResourceRequest(
         parent=f"projects/{project_id}/locations/{zone}",
         queued_resource_id=queued_resource_name,
         queued_resource=resource,