Function _node_pool_resource

axlearn/cloud/gcp/node_pool.py:234–249

Builds gcloud v1 node pool API resource. Args: credentials: Gcloud credentials used by googleapiclient.discovery. Returns: discovery.Resource object for the v1 node pool API.

(credentials: Credentials)

def _node_pool_resource(credentials: Credentials) -> discovery.Resource:
    """Builds gcloud v1 node pool API resource.

    Args:
        credentials: Gcloud credentials used by googleapiclient.discovery.

    Returns:
        discovery.Resource object for the v1 node pool API.
    """
    return (
        discovery.build("container", "v1", credentials=credentials, cache_discovery=False)
        .projects()
        .locations()
        .clusters()
        .nodePools()
    )


def _create_node_pool(resource: discovery.Resource, *, parent: str, body: Any):