# Generated by the protocol buffer compiler.  DO NOT EDIT!
# source: google/cloud/aiplatform/v1/custom_job.proto

require 'google/api/field_behavior_pb'
require 'google/api/resource_pb'
require 'google/cloud/aiplatform/v1/encryption_spec_pb'
require 'google/cloud/aiplatform/v1/env_var_pb'
require 'google/cloud/aiplatform/v1/io_pb'
require 'google/cloud/aiplatform/v1/job_state_pb'
require 'google/cloud/aiplatform/v1/machine_resources_pb'
require 'google/protobuf/duration_pb'
require 'google/protobuf/timestamp_pb'
require 'google/rpc/status_pb'
require 'google/protobuf'

Google::Protobuf::DescriptorPool.generated_pool.build do
  add_file("google/cloud/aiplatform/v1/custom_job.proto", :syntax => :proto3) do
    add_message "google.cloud.aiplatform.v1.CustomJob" do
      optional :name, :string, 1
      optional :display_name, :string, 2
      optional :job_spec, :message, 4, "google.cloud.aiplatform.v1.CustomJobSpec"
      optional :state, :enum, 5, "google.cloud.aiplatform.v1.JobState"
      optional :create_time, :message, 6, "google.protobuf.Timestamp"
      optional :start_time, :message, 7, "google.protobuf.Timestamp"
      optional :end_time, :message, 8, "google.protobuf.Timestamp"
      optional :update_time, :message, 9, "google.protobuf.Timestamp"
      optional :error, :message, 10, "google.rpc.Status"
      map :labels, :string, :string, 11
      optional :encryption_spec, :message, 12, "google.cloud.aiplatform.v1.EncryptionSpec"
      map :web_access_uris, :string, :string, 16
    end
    add_message "google.cloud.aiplatform.v1.CustomJobSpec" do
      repeated :worker_pool_specs, :message, 1, "google.cloud.aiplatform.v1.WorkerPoolSpec"
      optional :scheduling, :message, 3, "google.cloud.aiplatform.v1.Scheduling"
      optional :service_account, :string, 4
      optional :network, :string, 5
      optional :base_output_directory, :message, 6, "google.cloud.aiplatform.v1.GcsDestination"
      optional :tensorboard, :string, 7
      optional :enable_web_access, :bool, 10
    end
    add_message "google.cloud.aiplatform.v1.WorkerPoolSpec" do
      optional :machine_spec, :message, 1, "google.cloud.aiplatform.v1.MachineSpec"
      optional :replica_count, :int64, 2
      optional :disk_spec, :message, 5, "google.cloud.aiplatform.v1.DiskSpec"
      oneof :task do
        optional :container_spec, :message, 6, "google.cloud.aiplatform.v1.ContainerSpec"
        optional :python_package_spec, :message, 7, "google.cloud.aiplatform.v1.PythonPackageSpec"
      end
    end
    add_message "google.cloud.aiplatform.v1.ContainerSpec" do
      optional :image_uri, :string, 1
      repeated :command, :string, 2
      repeated :args, :string, 3
      repeated :env, :message, 4, "google.cloud.aiplatform.v1.EnvVar"
    end
    add_message "google.cloud.aiplatform.v1.PythonPackageSpec" do
      optional :executor_image_uri, :string, 1
      repeated :package_uris, :string, 2
      optional :python_module, :string, 3
      repeated :args, :string, 4
      repeated :env, :message, 5, "google.cloud.aiplatform.v1.EnvVar"
    end
    add_message "google.cloud.aiplatform.v1.Scheduling" do
      optional :timeout, :message, 1, "google.protobuf.Duration"
      optional :restart_job_on_worker_restart, :bool, 3
    end
  end
end

module Google
  module Cloud
    module AIPlatform
      module V1
        CustomJob = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CustomJob").msgclass
        CustomJobSpec = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.CustomJobSpec").msgclass
        WorkerPoolSpec = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.WorkerPoolSpec").msgclass
        ContainerSpec = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.ContainerSpec").msgclass
        PythonPackageSpec = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.PythonPackageSpec").msgclass
        Scheduling = ::Google::Protobuf::DescriptorPool.generated_pool.lookup("google.cloud.aiplatform.v1.Scheduling").msgclass
      end
    end
  end
end