Class: Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RaySpec

Inherits:

Object

• Object
• Google::Apis::AiplatformV1beta1::GoogleCloudAiplatformV1beta1RaySpec

Includes:

Core::Hashable, Core::JsonObjectSupport

Defined in:

lib/google/apis/aiplatform_v1beta1/classes.rb,
lib/google/apis/aiplatform_v1beta1/representations.rb,
lib/google/apis/aiplatform_v1beta1/representations.rb

Overview

Configuration information for the Ray cluster. For experimental launch, Ray cluster creation and Persistent cluster creation are 1:1 mapping: We will provision all the nodes within the Persistent cluster as Ray nodes.

Instance Attribute Summary

• #head_node_resource_pool_id ⇒ String

  Optional.

• #image_uri ⇒ String

  Optional.

• #resource_pool_images ⇒ Hash<String,String>

  Optional.

Instance Method Summary

• #initialize(**args) ⇒ GoogleCloudAiplatformV1beta1RaySpec constructor

  A new instance of GoogleCloudAiplatformV1beta1RaySpec.

• #update!(**args) ⇒ Object

  Update properties of this object.

Constructor Details

#initialize(**args) ⇒ GoogleCloudAiplatformV1beta1RaySpec

Returns a new instance of GoogleCloudAiplatformV1beta1RaySpec.

# File 'lib/google/apis/aiplatform_v1beta1/classes.rb', line 14629

def initialize(**args)
   update!(**args)
end

Instance Attribute Details

#head_node_resource_pool_id ⇒ String

Optional. This will be used to indicate which resource pool will serve as the Ray head node(the first node within that pool). Will use the machine from the first workerpool as the head node by default if this field is not set. Corresponds to the JSON property headNodeResourcePoolId

Returns:

• (String)

# File 'lib/google/apis/aiplatform_v1beta1/classes.rb', line 14608

def head_node_resource_pool_id
  @head_node_resource_pool_id
end

#image_uri ⇒ String

Optional. Default image for user to choose a preferred ML framework(e.g. tensorflow or Pytorch) by choosing from Vertex prebuild images(https://cloud. google.com/vertex-ai/docs/training/pre-built-containers). Either this or the resource_pool_images is required. Use this field if you need all the resource pools to have the same Ray image, Otherwise, use the @code resource_pool_images field. Corresponds to the JSON property imageUri

Returns:

• (String)

# File 'lib/google/apis/aiplatform_v1beta1/classes.rb', line 14618

def image_uri
  @image_uri
end

#resource_pool_images ⇒ Hash<String,String>

Optional. Required if image_uri is not set. A map of resource_pool_id to prebuild Ray image if user need to use different images for different head/ worker pools. This map needs to cover all the resource pool ids. Example: " ray_head_node_pool": "head image" "ray_worker_node_pool1": "worker image" " ray_worker_node_pool2": "another worker image" Corresponds to the JSON property resourcePoolImages

Returns:

• (Hash<String,String>)

# File 'lib/google/apis/aiplatform_v1beta1/classes.rb', line 14627

def resource_pool_images
  @resource_pool_images
end

Instance Method Details

#update!(**args) ⇒ Object

Update properties of this object

# File 'lib/google/apis/aiplatform_v1beta1/classes.rb', line 14634

def update!(**args)
  @head_node_resource_pool_id = args[:head_node_resource_pool_id] if args.key?(:head_node_resource_pool_id)
  @image_uri = args[:image_uri] if args.key?(:image_uri)
  @resource_pool_images = args[:resource_pool_images] if args.key?(:resource_pool_images)
end