Skip to content

Commit 0cfd235

Browse files
committed
dynamodb actions on Table, not on dynamodb resource
1 parent 227e97f commit 0cfd235

File tree

2 files changed

+4
-6
lines changed

2 files changed

+4
-6
lines changed

hpc_provisioner/src/hpc_provisioner/dynamodb_actions.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -106,13 +106,15 @@ def register_cluster(dynamodb_resource, cluster: Cluster) -> None:
106106
if get_cluster_by_name(dynamodb_resource, cluster.name):
107107
raise ClusterAlreadyRegistered(f"Cluster {cluster} already registered")
108108

109-
dynamodb_resource.update_item(
109+
table = dynamodb_resource.Table(CLUSTER_TABLE_NAME)
110+
table.update_item(
110111
TableName=CLUSTER_TABLE_NAME, Key={"name": {"S": cluster.name}}, AttributeUpdates={}
111112
)
112113

113114

114115
def delete_cluster(dynamodb_resource, cluster: Cluster) -> None:
115-
dynamodb_resource.delete_item(Key={"name": cluster.name})
116+
table = dynamodb_resource.Table(CLUSTER_TABLE_NAME)
117+
table.delete_item(Key={"name": cluster.name})
116118

117119

118120
def _update_cluster_claim(dynamodb_resource, cluster: Cluster, new_value: int) -> None:

hpc_provisioner/src/hpc_provisioner/pcluster_manager.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -286,11 +286,7 @@ def pcluster_describe(cluster: Cluster):
286286

287287
def pcluster_delete(cluster: Cluster):
288288
"""Destroy a cluster, given the vlab_id and project_id"""
289-
logger.debug("Releasing subnets")
290289
release_subnets(cluster.name)
291-
logger.debug("Removing admin key")
292290
remove_key(get_keypair_name(cluster))
293-
logger.debug("Removin sim key")
294291
remove_key(get_keypair_name(cluster, "sim"))
295-
logger.debug(f"Removing cluster {cluster.name}")
296292
return pc.delete_cluster(cluster_name=cluster.name, region=REGION)

0 commit comments

Comments
 (0)