Skip to content

Commit bb411cf

Browse files
committed
fix(tpu): Updating the state return value
1 parent 549adf5 commit bb411cf

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

tpu/queued_resources_create.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ def create_queued_resource(
6060

6161
response = operation.result()
6262
print(response.name)
63-
print(response.state.state)
63+
print(response.state)
6464
# Example response:
6565
# projects/[project_id]/locations/[zone]/queuedResources/resource-name
6666
# State.WAITING_FOR_RESOURCES

tpu/queued_resources_create_spot.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def create_queued_resource_spot(
6262
response = operation.result()
6363

6464
print(response.name)
65-
print(response.state.state)
65+
print(response.state)
6666
# Example response:
6767
# projects/[project_id]/locations/[zone]/queuedResources/resource-name
6868
# State.WAITING_FOR_RESOURCES

tpu/queued_resources_get.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -33,7 +33,7 @@ def get_queued_resource(
3333
)
3434
resource = client.get_queued_resource(name=name)
3535
print("Resource name:", resource.name)
36-
print(resource.state.state)
36+
print(resource.state)
3737
# Example response:
3838
# Resource name: projects/{project_id}/locations/{zone}/queuedResources/resource-name
3939
# State.ACTIVE

0 commit comments

Comments
 (0)