qwak-core 0.1.37__py3-none-any.whl → 0.5.19__py3-none-any.whl

_qwak_proto/qwak/automation/v1/notification_pb2.py

@@ -2,12 +2,10 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: qwak/automation/v1/notification.proto
 """Generated protocol buffer code."""
-from google.protobuf.internal import enum_type_wrapper
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import message as _message
-from google.protobuf import reflection as _reflection
 from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -17,59 +15,9 @@ _sym_db = _symbol_database.Default()
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n%qwak/automation/v1/notification.proto\x12\x12qwak.automation.v1\"\xa9\x01\n\x0cNotification\x12L\n\x17post_slack_notification\x18\x01 \x01(\x0b\x32).qwak.automation.v1.PostSlackNotificationH\x00\x12;\n\x0e\x63ustom_webhook\x18\x02 \x01(\x0b\x32!.qwak.automation.v1.CustomWebhookH\x00\x42\x0e\n\x0cnotification\"(\n\x15PostSlackNotification\x12\x0f\n\x07webhook\x18\x01 \x01(\t\"\xae\x02\n\rCustomWebhook\x12\x0b\n\x03url\x18\x01 \x01(\t\x12\x37\n\x0bhttp_method\x18\x02 \x01(\x0e\x32\".qwak.automation.v1.HttpMethodType\x12?\n\x07headers\x18\x03 \x03(\x0b\x32..qwak.automation.v1.CustomWebhook.HeadersEntry\x12\x39\n\x04\x64\x61ta\x18\x04 \x03(\x0b\x32+.qwak.automation.v1.CustomWebhook.DataEntry\x1a.\n\x0cHeadersEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01\x1a+\n\tDataEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\t:\x02\x38\x01*\xd1\x01\n\x0eHttpMethodType\x12\x1c\n\x18HTTP_METHOD_TYPE_INVALID\x10\x00\x12\x18\n\x14HTTP_METHOD_TYPE_GET\x10\x01\x12\x19\n\x15HTTP_METHOD_TYPE_POST\x10\x02\x12\x18\n\x14HTTP_METHOD_TYPE_PUT\x10\x03\x12\x1b\n\x17HTTP_METHOD_TYPE_DELETE\x10\x04\x12\x1a\n\x16HTTP_METHOD_TYPE_PATCH\x10\x05\x12\x19\n\x15HTTP_METHOD_TYPE_HEAD\x10\x06\x42!\n\x1d\x63om.qwak.ai.automation.api.v1P\x01\x62\x06proto3')
 
-_HTTPMETHODTYPE = DESCRIPTOR.enum_types_by_name['HttpMethodType']
-HttpMethodType = enum_type_wrapper.EnumTypeWrapper(_HTTPMETHODTYPE)
-HTTP_METHOD_TYPE_INVALID = 0
-HTTP_METHOD_TYPE_GET = 1
-HTTP_METHOD_TYPE_POST = 2
-HTTP_METHOD_TYPE_PUT = 3
-HTTP_METHOD_TYPE_DELETE = 4
-HTTP_METHOD_TYPE_PATCH = 5
-HTTP_METHOD_TYPE_HEAD = 6
-
-
-_NOTIFICATION = DESCRIPTOR.message_types_by_name['Notification']
-_POSTSLACKNOTIFICATION = DESCRIPTOR.message_types_by_name['PostSlackNotification']
-_CUSTOMWEBHOOK = DESCRIPTOR.message_types_by_name['CustomWebhook']
-_CUSTOMWEBHOOK_HEADERSENTRY = _CUSTOMWEBHOOK.nested_types_by_name['HeadersEntry']
-_CUSTOMWEBHOOK_DATAENTRY = _CUSTOMWEBHOOK.nested_types_by_name['DataEntry']
-Notification = _reflection.GeneratedProtocolMessageType('Notification', (_message.Message,), {
-  'DESCRIPTOR' : _NOTIFICATION,
-  '__module__' : 'qwak.automation.v1.notification_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.Notification)
-  })
-_sym_db.RegisterMessage(Notification)
-
-PostSlackNotification = _reflection.GeneratedProtocolMessageType('PostSlackNotification', (_message.Message,), {
-  'DESCRIPTOR' : _POSTSLACKNOTIFICATION,
-  '__module__' : 'qwak.automation.v1.notification_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.PostSlackNotification)
-  })
-_sym_db.RegisterMessage(PostSlackNotification)
-
-CustomWebhook = _reflection.GeneratedProtocolMessageType('CustomWebhook', (_message.Message,), {
-
-  'HeadersEntry' : _reflection.GeneratedProtocolMessageType('HeadersEntry', (_message.Message,), {
-    'DESCRIPTOR' : _CUSTOMWEBHOOK_HEADERSENTRY,
-    '__module__' : 'qwak.automation.v1.notification_pb2'
-    # @@protoc_insertion_point(class_scope:qwak.automation.v1.CustomWebhook.HeadersEntry)
-    })
-  ,
-
-  'DataEntry' : _reflection.GeneratedProtocolMessageType('DataEntry', (_message.Message,), {
-    'DESCRIPTOR' : _CUSTOMWEBHOOK_DATAENTRY,
-    '__module__' : 'qwak.automation.v1.notification_pb2'
-    # @@protoc_insertion_point(class_scope:qwak.automation.v1.CustomWebhook.DataEntry)
-    })
-  ,
-  'DESCRIPTOR' : _CUSTOMWEBHOOK,
-  '__module__' : 'qwak.automation.v1.notification_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.CustomWebhook)
-  })
-_sym_db.RegisterMessage(CustomWebhook)
-_sym_db.RegisterMessage(CustomWebhook.HeadersEntry)
-_sym_db.RegisterMessage(CustomWebhook.DataEntry)
-
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.automation.v1.notification_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
@@ -78,16 +26,16 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _CUSTOMWEBHOOK_HEADERSENTRY._serialized_options = b'8\001'
   _CUSTOMWEBHOOK_DATAENTRY._options = None
   _CUSTOMWEBHOOK_DATAENTRY._serialized_options = b'8\001'
-  _HTTPMETHODTYPE._serialized_start=581
-  _HTTPMETHODTYPE._serialized_end=790
-  _NOTIFICATION._serialized_start=62
-  _NOTIFICATION._serialized_end=231
-  _POSTSLACKNOTIFICATION._serialized_start=233
-  _POSTSLACKNOTIFICATION._serialized_end=273
-  _CUSTOMWEBHOOK._serialized_start=276
-  _CUSTOMWEBHOOK._serialized_end=578
-  _CUSTOMWEBHOOK_HEADERSENTRY._serialized_start=487
-  _CUSTOMWEBHOOK_HEADERSENTRY._serialized_end=533
-  _CUSTOMWEBHOOK_DATAENTRY._serialized_start=535
-  _CUSTOMWEBHOOK_DATAENTRY._serialized_end=578
+  _globals['_HTTPMETHODTYPE']._serialized_start=581
+  _globals['_HTTPMETHODTYPE']._serialized_end=790
+  _globals['_NOTIFICATION']._serialized_start=62
+  _globals['_NOTIFICATION']._serialized_end=231
+  _globals['_POSTSLACKNOTIFICATION']._serialized_start=233
+  _globals['_POSTSLACKNOTIFICATION']._serialized_end=273
+  _globals['_CUSTOMWEBHOOK']._serialized_start=276
+  _globals['_CUSTOMWEBHOOK']._serialized_end=578
+  _globals['_CUSTOMWEBHOOK_HEADERSENTRY']._serialized_start=487
+  _globals['_CUSTOMWEBHOOK_HEADERSENTRY']._serialized_end=533
+  _globals['_CUSTOMWEBHOOK_DATAENTRY']._serialized_start=535
+  _globals['_CUSTOMWEBHOOK_DATAENTRY']._serialized_end=578
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/automation/v1/trigger_pb2.py

@@ -4,9 +4,8 @@
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import message as _message
-from google.protobuf import reflection as _reflection
 from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -17,60 +16,21 @@ from _qwak_proto.qwak.automation.v1 import common_pb2 as qwak_dot_automation_dot
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n qwak/automation/v1/trigger.proto\x12\x12qwak.automation.v1\x1a\x1fqwak/automation/v1/common.proto\"\x9e\x02\n\x07Trigger\x12\x41\n\x11scheduled_trigger\x18\x01 \x01(\x0b\x32$.qwak.automation.v1.ScheduledTriggerH\x00\x12\x46\n\x14metric_based_trigger\x18\x02 \x01(\x0b\x32&.qwak.automation.v1.MetricBasedTriggerH\x00\x12\x37\n\x0cnone_trigger\x18\x03 \x01(\x0b\x32\x1f.qwak.automation.v1.NoneTriggerH\x00\x12\x44\n\x13on_boarding_trigger\x18\x04 \x01(\x0b\x32%.qwak.automation.v1.OnBoardingTriggerH\x00\x42\t\n\x07trigger\"B\n\x10ScheduledTrigger\x12\x0e\n\x04\x63ron\x18\x01 \x01(\tH\x00\x12\x12\n\x08interval\x18\x02 \x01(\tH\x00\x42\n\n\x08schedule\"\xc3\x01\n\x12MetricBasedTrigger\x12\x0c\n\x04name\x18\x01 \x01(\t\x12*\n\x06metric\x18\x02 \x01(\x0b\x32\x1a.qwak.automation.v1.Metric\x12I\n\x13threshold_direction\x18\x03 \x01(\x0e\x32,.qwak.automation.v1.MetricThresholdDirection\x12\x11\n\tthreshold\x18\x04 \x01(\t\x12\x15\n\roverride_cron\x18\x05 \x01(\t\"\r\n\x0bNoneTrigger\"\x13\n\x11OnBoardingTriggerB!\n\x1d\x63om.qwak.ai.automation.api.v1P\x01\x62\x06proto3')
 
-
-
-_TRIGGER = DESCRIPTOR.message_types_by_name['Trigger']
-_SCHEDULEDTRIGGER = DESCRIPTOR.message_types_by_name['ScheduledTrigger']
-_METRICBASEDTRIGGER = DESCRIPTOR.message_types_by_name['MetricBasedTrigger']
-_NONETRIGGER = DESCRIPTOR.message_types_by_name['NoneTrigger']
-_ONBOARDINGTRIGGER = DESCRIPTOR.message_types_by_name['OnBoardingTrigger']
-Trigger = _reflection.GeneratedProtocolMessageType('Trigger', (_message.Message,), {
-  'DESCRIPTOR' : _TRIGGER,
-  '__module__' : 'qwak.automation.v1.trigger_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.Trigger)
-  })
-_sym_db.RegisterMessage(Trigger)
-
-ScheduledTrigger = _reflection.GeneratedProtocolMessageType('ScheduledTrigger', (_message.Message,), {
-  'DESCRIPTOR' : _SCHEDULEDTRIGGER,
-  '__module__' : 'qwak.automation.v1.trigger_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.ScheduledTrigger)
-  })
-_sym_db.RegisterMessage(ScheduledTrigger)
-
-MetricBasedTrigger = _reflection.GeneratedProtocolMessageType('MetricBasedTrigger', (_message.Message,), {
-  'DESCRIPTOR' : _METRICBASEDTRIGGER,
-  '__module__' : 'qwak.automation.v1.trigger_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.MetricBasedTrigger)
-  })
-_sym_db.RegisterMessage(MetricBasedTrigger)
-
-NoneTrigger = _reflection.GeneratedProtocolMessageType('NoneTrigger', (_message.Message,), {
-  'DESCRIPTOR' : _NONETRIGGER,
-  '__module__' : 'qwak.automation.v1.trigger_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.NoneTrigger)
-  })
-_sym_db.RegisterMessage(NoneTrigger)
-
-OnBoardingTrigger = _reflection.GeneratedProtocolMessageType('OnBoardingTrigger', (_message.Message,), {
-  'DESCRIPTOR' : _ONBOARDINGTRIGGER,
-  '__module__' : 'qwak.automation.v1.trigger_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.automation.v1.OnBoardingTrigger)
-  })
-_sym_db.RegisterMessage(OnBoardingTrigger)
-
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.automation.v1.trigger_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n\035com.qwak.ai.automation.api.v1P\001'
-  _TRIGGER._serialized_start=90
-  _TRIGGER._serialized_end=376
-  _SCHEDULEDTRIGGER._serialized_start=378
-  _SCHEDULEDTRIGGER._serialized_end=444
-  _METRICBASEDTRIGGER._serialized_start=447
-  _METRICBASEDTRIGGER._serialized_end=642
-  _NONETRIGGER._serialized_start=644
-  _NONETRIGGER._serialized_end=657
-  _ONBOARDINGTRIGGER._serialized_start=659
-  _ONBOARDINGTRIGGER._serialized_end=678
+  _globals['_TRIGGER']._serialized_start=90
+  _globals['_TRIGGER']._serialized_end=376
+  _globals['_SCHEDULEDTRIGGER']._serialized_start=378
+  _globals['_SCHEDULEDTRIGGER']._serialized_end=444
+  _globals['_METRICBASEDTRIGGER']._serialized_start=447
+  _globals['_METRICBASEDTRIGGER']._serialized_end=642
+  _globals['_NONETRIGGER']._serialized_start=644
+  _globals['_NONETRIGGER']._serialized_end=657
+  _globals['_ONBOARDINGTRIGGER']._serialized_start=659
+  _globals['_ONBOARDINGTRIGGER']._serialized_end=678
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/batch_job/v1/batch_job_events_pb2.py

@@ -2,12 +2,10 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: qwak/batch_job/v1/batch_job_events.proto
 """Generated protocol buffer code."""
-from google.protobuf.internal import enum_type_wrapper
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import message as _message
-from google.protobuf import reflection as _reflection
 from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -16,98 +14,11 @@ _sym_db = _symbol_database.Default()
 from _qwak_proto.qwak.user_application.common.v0 import resources_pb2 as qwak_dot_user__application_dot_common_dot_v0_dot_resources__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n(qwak/batch_job/v1/batch_job_events.proto\x12\rqwak.batchjob\x1a/qwak/user_application/common/v0/resources.proto\"\xa5\x01\n\x14\x42\x61tchJobEventMessage\x12\x45\n\x14\x62\x61tch_job_event_type\x18\x01 \x01(\x0e\x32\'.qwak.batchjob.BatchJobEventTypeMessage\x12\x14\n\x0c\x62\x61tch_job_id\x18\x02 \x01(\t\x12\x0f\n\x07message\x18\x03 \x01(\t\x12\r\n\x05\x64\x65lay\x18\x04 \x01(\x05\x12\x10\n\x08\x66\x61ilures\x18\x05 \x01(\x05\"\x84\x02\n\x15\x42\x61tchTaskEventMessage\x12G\n\x15\x62\x61tch_task_event_type\x18\x01 \x01(\x0e\x32(.qwak.batchjob.BatchTaskEventTypeMessage\x12\x14\n\x0c\x62\x61tch_job_id\x18\x02 \x01(\t\x12\x15\n\rbatch_task_id\x18\x03 \x01(\t\x12\x0f\n\x07message\x18\x04 \x01(\t\x12\r\n\x05\x64\x65lay\x18\x05 \x01(\x05\x12\x10\n\x08\x66\x61ilures\x18\x06 \x01(\x05\x12\x43\n\x19\x62\x61tch_task_failure_reason\x18\x07 \x01(\x0b\x32 .qwak.batchjob.TaskFailureReason\"\xec\x01\n\x12WarmupEventMessage\x12\x44\n\x15warmup_job_event_type\x18\x01 \x01(\x0e\x32%.qwak.batchjob.WarmupEventTypeMessage\x12\r\n\x05\x64\x65lay\x18\x02 \x01(\x05\x12\x10\n\x08\x66\x61ilures\x18\x03 \x01(\x05\x12.\n\x0bwarmup_size\x18\x04 \x01(\x0b\x32\x19.qwak.batchjob.WarmupSize\x12?\n\x10model_identifier\x18\x05 \x01(\x0b\x32%.qwak.batchjob.ModelIdentifierMessage\"\xb9\x02\n\nWarmupSize\x12\x0f\n\x03\x63pu\x18\x01 \x01(\x02\x42\x02\x18\x01\x12\x19\n\rmemory_amount\x18\x02 \x01(\x05\x42\x02\x18\x01\x12\x32\n\x0bmemory_unit\x18\x03 \x01(\x0e\x32\x19.qwak.batchjob.MemoryUnitB\x02\x18\x01\x12\x13\n\x0bnum_of_pods\x18\x04 \x01(\x05\x12\x17\n\x0ftimeout_seconds\x18\x05 \x01(\x05\x12H\n\rgpu_resources\x18\x06 \x01(\x0b\x32-.qwak.user_application.common.v0.GpuResourcesB\x02\x18\x01\x12S\n\x15pod_compute_resources\x18\x07 \x01(\x0b\x32\x34.qwak.user_application.common.v0.PodComputeResources\"|\n\x16ModelIdentifierMessage\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x15\n\tbranch_id\x18\x02 \x01(\tB\x02\x18\x01\x12\x10\n\x08\x62uild_id\x18\x03 \x01(\t\x12\x0f\n\x07user_id\x18\x04 \x01(\t\x12\x16\n\x0e\x65nvironment_id\x18\x05 \x01(\t\"\x80\x01\n\x11TaskFailureReason\x12\x46\n\x18task_failure_reason_code\x18\x01 \x01(\x0e\x32$.qwak.batchjob.TaskFailureReasonCode\x12#\n\x1btask_failure_reason_details\x18\x02 \x01(\t*\xf2\x01\n\x18\x42\x61tchJobEventTypeMessage\x12\x1d\n\x19UNDEFINED_BATCH_JOB_EVENT\x10\x00\x12\x17\n\x13\x42\x41TCH_JOB_COMMITTED\x10\x01\x12\x16\n\x12\x42\x41TCH_JOB_PREPARED\x10\x07\x12\x1b\n\x17\x42\x41TCH_JOB_TASKS_CREATED\x10\x02\x12\x17\n\x13\x42\x41TCH_JOB_CANCELLED\x10\x03\x12\x15\n\x11\x42\x41TCH_JOB_CLEANUP\x10\x04\x12\x1c\n\x18\x42\x41TCH_JOB_CHECK_FINISHED\x10\x05\x12\x1b\n\x17\x42\x41TCH_JOB_CHECK_TIMEOUT\x10\x06*\xce\x01\n\x19\x42\x61tchTaskEventTypeMessage\x12\x1e\n\x1aUNDEFINED_BATCH_TASK_EVENT\x10\x00\x12\x16\n\x12\x42\x41TCH_TASK_RUNNING\x10\x01\x12\x17\n\x13\x42\x41TCH_TASK_FINISHED\x10\x02\x12\x15\n\x11\x42\x41TCH_TASK_FAILED\x10\x03\x12\x1b\n\x17\x42\x41TCH_TASK_CHECK_STATUS\x10\x04\x12\x14\n\x10\x42\x41TCH_TASK_START\x10\x05\x12\x16\n\x12\x42\x41TCH_TASK_CLEANUP\x10\x06*e\n\x16WarmupEventTypeMessage\x12\x1e\n\x1aUNDEFINED_WARMUP_JOB_EVENT\x10\x00\x12\x18\n\x14WARMUP_JOB_COMMITTED\x10\x01\x12\x11\n\rWARMUP_CANCEL\x10\x02*+\n\nMemoryUnit\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02*\xc0\x04\n\x15TaskFailureReasonCode\x12\x35\n1TASK_FAILURE_REASON_CODE_FAILURE_CODE_UNSPECIFIED\x10\x00\x12\x44\n@TASK_FAILURE_REASON_CODE_NO_REPLICA_SETS_PODS_FOUND_FAILURE_CODE\x10\x01\x12\x43\n?TASK_FAILURE_REASON_CODE_MODEL_CONTAINER_NOT_FOUND_FAILURE_CODE\x10\x02\x12\x37\n3TASK_FAILURE_REASON_CODE_UNSCHEDULABLE_FAILURE_CODE\x10\x03\x12@\n<TASK_FAILURE_REASON_CODE_DOCKER_IMAGE_NOT_FOUND_FAILURE_CODE\x10\x04\x12?\n;TASK_FAILURE_REASON_CODE_MEMORY_LIMIT_EXCEEDED_FAILURE_CODE\x10\x05\x12\x34\n0TASK_FAILURE_REASON_CODE_CRASH_LOOP_FAILURE_CODE\x10\x06\x12\x46\nBTASK_FAILURE_REASON_CODE_CONTAINER_FAIL_TO_LOAD_FOR_UNKNOWN_REASON\x10\x07\x12+\n\'TASK_FAILURE_REASON_CODE_UNKNOWN_REASON\x10\x08\x42.\n\x19\x63om.qwak.ai.batch.job.apiP\x01Z\x0f./;inferencejobb\x06proto3')
-
-_BATCHJOBEVENTTYPEMESSAGE = DESCRIPTOR.enum_types_by_name['BatchJobEventTypeMessage']
-BatchJobEventTypeMessage = enum_type_wrapper.EnumTypeWrapper(_BATCHJOBEVENTTYPEMESSAGE)
-_BATCHTASKEVENTTYPEMESSAGE = DESCRIPTOR.enum_types_by_name['BatchTaskEventTypeMessage']
-BatchTaskEventTypeMessage = enum_type_wrapper.EnumTypeWrapper(_BATCHTASKEVENTTYPEMESSAGE)
-_WARMUPEVENTTYPEMESSAGE = DESCRIPTOR.enum_types_by_name['WarmupEventTypeMessage']
-WarmupEventTypeMessage = enum_type_wrapper.EnumTypeWrapper(_WARMUPEVENTTYPEMESSAGE)
-_MEMORYUNIT = DESCRIPTOR.enum_types_by_name['MemoryUnit']
-MemoryUnit = enum_type_wrapper.EnumTypeWrapper(_MEMORYUNIT)
-_TASKFAILUREREASONCODE = DESCRIPTOR.enum_types_by_name['TaskFailureReasonCode']
-TaskFailureReasonCode = enum_type_wrapper.EnumTypeWrapper(_TASKFAILUREREASONCODE)
-UNDEFINED_BATCH_JOB_EVENT = 0
-BATCH_JOB_COMMITTED = 1
-BATCH_JOB_PREPARED = 7
-BATCH_JOB_TASKS_CREATED = 2
-BATCH_JOB_CANCELLED = 3
-BATCH_JOB_CLEANUP = 4
-BATCH_JOB_CHECK_FINISHED = 5
-BATCH_JOB_CHECK_TIMEOUT = 6
-UNDEFINED_BATCH_TASK_EVENT = 0
-BATCH_TASK_RUNNING = 1
-BATCH_TASK_FINISHED = 2
-BATCH_TASK_FAILED = 3
-BATCH_TASK_CHECK_STATUS = 4
-BATCH_TASK_START = 5
-BATCH_TASK_CLEANUP = 6
-UNDEFINED_WARMUP_JOB_EVENT = 0
-WARMUP_JOB_COMMITTED = 1
-WARMUP_CANCEL = 2
-UNKNOWN = 0
-MIB = 1
-GIB = 2
-TASK_FAILURE_REASON_CODE_FAILURE_CODE_UNSPECIFIED = 0
-TASK_FAILURE_REASON_CODE_NO_REPLICA_SETS_PODS_FOUND_FAILURE_CODE = 1
-TASK_FAILURE_REASON_CODE_MODEL_CONTAINER_NOT_FOUND_FAILURE_CODE = 2
-TASK_FAILURE_REASON_CODE_UNSCHEDULABLE_FAILURE_CODE = 3
-TASK_FAILURE_REASON_CODE_DOCKER_IMAGE_NOT_FOUND_FAILURE_CODE = 4
-TASK_FAILURE_REASON_CODE_MEMORY_LIMIT_EXCEEDED_FAILURE_CODE = 5
-TASK_FAILURE_REASON_CODE_CRASH_LOOP_FAILURE_CODE = 6
-TASK_FAILURE_REASON_CODE_CONTAINER_FAIL_TO_LOAD_FOR_UNKNOWN_REASON = 7
-TASK_FAILURE_REASON_CODE_UNKNOWN_REASON = 8
-
-
-_BATCHJOBEVENTMESSAGE = DESCRIPTOR.message_types_by_name['BatchJobEventMessage']
-_BATCHTASKEVENTMESSAGE = DESCRIPTOR.message_types_by_name['BatchTaskEventMessage']
-_WARMUPEVENTMESSAGE = DESCRIPTOR.message_types_by_name['WarmupEventMessage']
-_WARMUPSIZE = DESCRIPTOR.message_types_by_name['WarmupSize']
-_MODELIDENTIFIERMESSAGE = DESCRIPTOR.message_types_by_name['ModelIdentifierMessage']
-_TASKFAILUREREASON = DESCRIPTOR.message_types_by_name['TaskFailureReason']
-BatchJobEventMessage = _reflection.GeneratedProtocolMessageType('BatchJobEventMessage', (_message.Message,), {
-  'DESCRIPTOR' : _BATCHJOBEVENTMESSAGE,
-  '__module__' : 'qwak.batch_job.v1.batch_job_events_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.batchjob.BatchJobEventMessage)
-  })
-_sym_db.RegisterMessage(BatchJobEventMessage)
-
-BatchTaskEventMessage = _reflection.GeneratedProtocolMessageType('BatchTaskEventMessage', (_message.Message,), {
-  'DESCRIPTOR' : _BATCHTASKEVENTMESSAGE,
-  '__module__' : 'qwak.batch_job.v1.batch_job_events_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.batchjob.BatchTaskEventMessage)
-  })
-_sym_db.RegisterMessage(BatchTaskEventMessage)
-
-WarmupEventMessage = _reflection.GeneratedProtocolMessageType('WarmupEventMessage', (_message.Message,), {
-  'DESCRIPTOR' : _WARMUPEVENTMESSAGE,
-  '__module__' : 'qwak.batch_job.v1.batch_job_events_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.batchjob.WarmupEventMessage)
-  })
-_sym_db.RegisterMessage(WarmupEventMessage)
-
-WarmupSize = _reflection.GeneratedProtocolMessageType('WarmupSize', (_message.Message,), {
-  'DESCRIPTOR' : _WARMUPSIZE,
-  '__module__' : 'qwak.batch_job.v1.batch_job_events_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.batchjob.WarmupSize)
-  })
-_sym_db.RegisterMessage(WarmupSize)
-
-ModelIdentifierMessage = _reflection.GeneratedProtocolMessageType('ModelIdentifierMessage', (_message.Message,), {
-  'DESCRIPTOR' : _MODELIDENTIFIERMESSAGE,
-  '__module__' : 'qwak.batch_job.v1.batch_job_events_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.batchjob.ModelIdentifierMessage)
-  })
-_sym_db.RegisterMessage(ModelIdentifierMessage)
-
-TaskFailureReason = _reflection.GeneratedProtocolMessageType('TaskFailureReason', (_message.Message,), {
-  'DESCRIPTOR' : _TASKFAILUREREASON,
-  '__module__' : 'qwak.batch_job.v1.batch_job_events_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.batchjob.TaskFailureReason)
-  })
-_sym_db.RegisterMessage(TaskFailureReason)
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n(qwak/batch_job/v1/batch_job_events.proto\x12\rqwak.batchjob\x1a/qwak/user_application/common/v0/resources.proto\"\xa5\x01\n\x14\x42\x61tchJobEventMessage\x12\x45\n\x14\x62\x61tch_job_event_type\x18\x01 \x01(\x0e\x32\'.qwak.batchjob.BatchJobEventTypeMessage\x12\x14\n\x0c\x62\x61tch_job_id\x18\x02 \x01(\t\x12\x0f\n\x07message\x18\x03 \x01(\t\x12\r\n\x05\x64\x65lay\x18\x04 \x01(\x05\x12\x10\n\x08\x66\x61ilures\x18\x05 \x01(\x05\"\x84\x02\n\x15\x42\x61tchTaskEventMessage\x12G\n\x15\x62\x61tch_task_event_type\x18\x01 \x01(\x0e\x32(.qwak.batchjob.BatchTaskEventTypeMessage\x12\x14\n\x0c\x62\x61tch_job_id\x18\x02 \x01(\t\x12\x15\n\rbatch_task_id\x18\x03 \x01(\t\x12\x0f\n\x07message\x18\x04 \x01(\t\x12\r\n\x05\x64\x65lay\x18\x05 \x01(\x05\x12\x10\n\x08\x66\x61ilures\x18\x06 \x01(\x05\x12\x43\n\x19\x62\x61tch_task_failure_reason\x18\x07 \x01(\x0b\x32 .qwak.batchjob.TaskFailureReason\"\xec\x01\n\x12WarmupEventMessage\x12\x44\n\x15warmup_job_event_type\x18\x01 \x01(\x0e\x32%.qwak.batchjob.WarmupEventTypeMessage\x12\r\n\x05\x64\x65lay\x18\x02 \x01(\x05\x12\x10\n\x08\x66\x61ilures\x18\x03 \x01(\x05\x12.\n\x0bwarmup_size\x18\x04 \x01(\x0b\x32\x19.qwak.batchjob.WarmupSize\x12?\n\x10model_identifier\x18\x05 \x01(\x0b\x32%.qwak.batchjob.ModelIdentifierMessage\"\xb9\x02\n\nWarmupSize\x12\x0f\n\x03\x63pu\x18\x01 \x01(\x02\x42\x02\x18\x01\x12\x19\n\rmemory_amount\x18\x02 \x01(\x05\x42\x02\x18\x01\x12\x32\n\x0bmemory_unit\x18\x03 \x01(\x0e\x32\x19.qwak.batchjob.MemoryUnitB\x02\x18\x01\x12\x13\n\x0bnum_of_pods\x18\x04 \x01(\x05\x12\x17\n\x0ftimeout_seconds\x18\x05 \x01(\x05\x12H\n\rgpu_resources\x18\x06 \x01(\x0b\x32-.qwak.user_application.common.v0.GpuResourcesB\x02\x18\x01\x12S\n\x15pod_compute_resources\x18\x07 \x01(\x0b\x32\x34.qwak.user_application.common.v0.PodComputeResources\"\x90\x01\n\x16ModelIdentifierMessage\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x15\n\tbranch_id\x18\x02 \x01(\tB\x02\x18\x01\x12\x10\n\x08\x62uild_id\x18\x03 \x01(\t\x12\x0f\n\x07user_id\x18\x04 \x01(\t\x12\x16\n\x0e\x65nvironment_id\x18\x05 \x01(\t\x12\x12\n\naccount_id\x18\x06 \x01(\t\"\x80\x01\n\x11TaskFailureReason\x12\x46\n\x18task_failure_reason_code\x18\x01 \x01(\x0e\x32$.qwak.batchjob.TaskFailureReasonCode\x12#\n\x1btask_failure_reason_details\x18\x02 \x01(\t*\xf2\x01\n\x18\x42\x61tchJobEventTypeMessage\x12\x1d\n\x19UNDEFINED_BATCH_JOB_EVENT\x10\x00\x12\x17\n\x13\x42\x41TCH_JOB_COMMITTED\x10\x01\x12\x16\n\x12\x42\x41TCH_JOB_PREPARED\x10\x07\x12\x1b\n\x17\x42\x41TCH_JOB_TASKS_CREATED\x10\x02\x12\x17\n\x13\x42\x41TCH_JOB_CANCELLED\x10\x03\x12\x15\n\x11\x42\x41TCH_JOB_CLEANUP\x10\x04\x12\x1c\n\x18\x42\x41TCH_JOB_CHECK_FINISHED\x10\x05\x12\x1b\n\x17\x42\x41TCH_JOB_CHECK_TIMEOUT\x10\x06*\xce\x01\n\x19\x42\x61tchTaskEventTypeMessage\x12\x1e\n\x1aUNDEFINED_BATCH_TASK_EVENT\x10\x00\x12\x16\n\x12\x42\x41TCH_TASK_RUNNING\x10\x01\x12\x17\n\x13\x42\x41TCH_TASK_FINISHED\x10\x02\x12\x15\n\x11\x42\x41TCH_TASK_FAILED\x10\x03\x12\x1b\n\x17\x42\x41TCH_TASK_CHECK_STATUS\x10\x04\x12\x14\n\x10\x42\x41TCH_TASK_START\x10\x05\x12\x16\n\x12\x42\x41TCH_TASK_CLEANUP\x10\x06*e\n\x16WarmupEventTypeMessage\x12\x1e\n\x1aUNDEFINED_WARMUP_JOB_EVENT\x10\x00\x12\x18\n\x14WARMUP_JOB_COMMITTED\x10\x01\x12\x11\n\rWARMUP_CANCEL\x10\x02*+\n\nMemoryUnit\x12\x0b\n\x07UNKNOWN\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02*\x9a\x05\n\x15TaskFailureReasonCode\x12\x35\n1TASK_FAILURE_REASON_CODE_FAILURE_CODE_UNSPECIFIED\x10\x00\x12\x44\n@TASK_FAILURE_REASON_CODE_NO_REPLICA_SETS_PODS_FOUND_FAILURE_CODE\x10\x01\x12\x43\n?TASK_FAILURE_REASON_CODE_MODEL_CONTAINER_NOT_FOUND_FAILURE_CODE\x10\x02\x12\x37\n3TASK_FAILURE_REASON_CODE_UNSCHEDULABLE_FAILURE_CODE\x10\x03\x12@\n<TASK_FAILURE_REASON_CODE_DOCKER_IMAGE_NOT_FOUND_FAILURE_CODE\x10\x04\x12?\n;TASK_FAILURE_REASON_CODE_MEMORY_LIMIT_EXCEEDED_FAILURE_CODE\x10\x05\x12\x34\n0TASK_FAILURE_REASON_CODE_CRASH_LOOP_FAILURE_CODE\x10\x06\x12\x46\nBTASK_FAILURE_REASON_CODE_CONTAINER_FAIL_TO_LOAD_FOR_UNKNOWN_REASON\x10\x07\x12+\n\'TASK_FAILURE_REASON_CODE_UNKNOWN_REASON\x10\x08\x12X\nTTASK_FAILURE_REASON_CODE_DOCKER_IMAGE_GETS_THROTTLING_BY_CLOUD_PROVIDER_FAILURE_CODE\x10\tB.\n\x19\x63om.qwak.ai.batch.job.apiP\x01Z\x0f./;inferencejobb\x06proto3')
 
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.batch_job.v1.batch_job_events_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
@@ -122,26 +33,26 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _WARMUPSIZE.fields_by_name['gpu_resources']._serialized_options = b'\030\001'
   _MODELIDENTIFIERMESSAGE.fields_by_name['branch_id']._options = None
   _MODELIDENTIFIERMESSAGE.fields_by_name['branch_id']._serialized_options = b'\030\001'
-  _BATCHJOBEVENTTYPEMESSAGE._serialized_start=1352
-  _BATCHJOBEVENTTYPEMESSAGE._serialized_end=1594
-  _BATCHTASKEVENTTYPEMESSAGE._serialized_start=1597
-  _BATCHTASKEVENTTYPEMESSAGE._serialized_end=1803
-  _WARMUPEVENTTYPEMESSAGE._serialized_start=1805
-  _WARMUPEVENTTYPEMESSAGE._serialized_end=1906
-  _MEMORYUNIT._serialized_start=1908
-  _MEMORYUNIT._serialized_end=1951
-  _TASKFAILUREREASONCODE._serialized_start=1954
-  _TASKFAILUREREASONCODE._serialized_end=2530
-  _BATCHJOBEVENTMESSAGE._serialized_start=109
-  _BATCHJOBEVENTMESSAGE._serialized_end=274
-  _BATCHTASKEVENTMESSAGE._serialized_start=277
-  _BATCHTASKEVENTMESSAGE._serialized_end=537
-  _WARMUPEVENTMESSAGE._serialized_start=540
-  _WARMUPEVENTMESSAGE._serialized_end=776
-  _WARMUPSIZE._serialized_start=779
-  _WARMUPSIZE._serialized_end=1092
-  _MODELIDENTIFIERMESSAGE._serialized_start=1094
-  _MODELIDENTIFIERMESSAGE._serialized_end=1218
-  _TASKFAILUREREASON._serialized_start=1221
-  _TASKFAILUREREASON._serialized_end=1349
+  _globals['_BATCHJOBEVENTTYPEMESSAGE']._serialized_start=1373
+  _globals['_BATCHJOBEVENTTYPEMESSAGE']._serialized_end=1615
+  _globals['_BATCHTASKEVENTTYPEMESSAGE']._serialized_start=1618
+  _globals['_BATCHTASKEVENTTYPEMESSAGE']._serialized_end=1824
+  _globals['_WARMUPEVENTTYPEMESSAGE']._serialized_start=1826
+  _globals['_WARMUPEVENTTYPEMESSAGE']._serialized_end=1927
+  _globals['_MEMORYUNIT']._serialized_start=1929
+  _globals['_MEMORYUNIT']._serialized_end=1972
+  _globals['_TASKFAILUREREASONCODE']._serialized_start=1975
+  _globals['_TASKFAILUREREASONCODE']._serialized_end=2641
+  _globals['_BATCHJOBEVENTMESSAGE']._serialized_start=109
+  _globals['_BATCHJOBEVENTMESSAGE']._serialized_end=274
+  _globals['_BATCHTASKEVENTMESSAGE']._serialized_start=277
+  _globals['_BATCHTASKEVENTMESSAGE']._serialized_end=537
+  _globals['_WARMUPEVENTMESSAGE']._serialized_start=540
+  _globals['_WARMUPEVENTMESSAGE']._serialized_end=776
+  _globals['_WARMUPSIZE']._serialized_start=779
+  _globals['_WARMUPSIZE']._serialized_end=1092
+  _globals['_MODELIDENTIFIERMESSAGE']._serialized_start=1095
+  _globals['_MODELIDENTIFIERMESSAGE']._serialized_end=1239
+  _globals['_TASKFAILUREREASON']._serialized_start=1242
+  _globals['_TASKFAILUREREASON']._serialized_end=1370
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/batch_job/v1/batch_job_events_pb2.pyi

@@ -118,6 +118,7 @@ class _TaskFailureReasonCodeEnumTypeWrapper(google.protobuf.internal.enum_type_w
     TASK_FAILURE_REASON_CODE_CRASH_LOOP_FAILURE_CODE: _TaskFailureReasonCode.ValueType  # 6
     TASK_FAILURE_REASON_CODE_CONTAINER_FAIL_TO_LOAD_FOR_UNKNOWN_REASON: _TaskFailureReasonCode.ValueType  # 7
     TASK_FAILURE_REASON_CODE_UNKNOWN_REASON: _TaskFailureReasonCode.ValueType  # 8
+    TASK_FAILURE_REASON_CODE_DOCKER_IMAGE_GETS_THROTTLING_BY_CLOUD_PROVIDER_FAILURE_CODE: _TaskFailureReasonCode.ValueType  # 9
 
 class TaskFailureReasonCode(_TaskFailureReasonCode, metaclass=_TaskFailureReasonCodeEnumTypeWrapper): ...
 
@@ -130,6 +131,7 @@ TASK_FAILURE_REASON_CODE_MEMORY_LIMIT_EXCEEDED_FAILURE_CODE: TaskFailureReasonCo
 TASK_FAILURE_REASON_CODE_CRASH_LOOP_FAILURE_CODE: TaskFailureReasonCode.ValueType  # 6
 TASK_FAILURE_REASON_CODE_CONTAINER_FAIL_TO_LOAD_FOR_UNKNOWN_REASON: TaskFailureReasonCode.ValueType  # 7
 TASK_FAILURE_REASON_CODE_UNKNOWN_REASON: TaskFailureReasonCode.ValueType  # 8
+TASK_FAILURE_REASON_CODE_DOCKER_IMAGE_GETS_THROTTLING_BY_CLOUD_PROVIDER_FAILURE_CODE: TaskFailureReasonCode.ValueType  # 9
 global___TaskFailureReasonCode = TaskFailureReasonCode
 
 class BatchJobEventMessage(google.protobuf.message.Message):
@@ -266,11 +268,13 @@ class ModelIdentifierMessage(google.protobuf.message.Message):
     BUILD_ID_FIELD_NUMBER: builtins.int
     USER_ID_FIELD_NUMBER: builtins.int
     ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
+    ACCOUNT_ID_FIELD_NUMBER: builtins.int
     model_id: builtins.str
     branch_id: builtins.str
     build_id: builtins.str
     user_id: builtins.str
     environment_id: builtins.str
+    account_id: builtins.str
     def __init__(
         self,
         *,
@@ -279,8 +283,9 @@ class ModelIdentifierMessage(google.protobuf.message.Message):
         build_id: builtins.str = ...,
         user_id: builtins.str = ...,
         environment_id: builtins.str = ...,
+        account_id: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["branch_id", b"branch_id", "build_id", b"build_id", "environment_id", b"environment_id", "model_id", b"model_id", "user_id", b"user_id"]) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["account_id", b"account_id", "branch_id", b"branch_id", "build_id", b"build_id", "environment_id", b"environment_id", "model_id", b"model_id", "user_id", b"user_id"]) -> None: ...
 
 global___ModelIdentifierMessage = ModelIdentifierMessage
 

_qwak_proto/qwak/batch_job/v1/batch_job_resources_pb2.py

@@ -2,12 +2,10 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: qwak/batch_job/v1/batch_job_resources.proto
 """Generated protocol buffer code."""
-from google.protobuf.internal import enum_type_wrapper
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import message as _message
-from google.protobuf import reflection as _reflection
 from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -15,32 +13,17 @@ _sym_db = _symbol_database.Default()
 
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n+qwak/batch_job/v1/batch_job_resources.proto\x12\rqwak.batchjob\"L\n\x0cGpuResources\x12(\n\x08gpu_type\x18\x01 \x01(\x0e\x32\x16.qwak.batchjob.GpuType\x12\x12\n\ngpu_amount\x18\x02 \x01(\x05*l\n\x07GpuType\x12\x0f\n\x0bINVALID_GPU\x10\x00\x12\x0e\n\nNVIDIA_K80\x10\x01\x12\x0f\n\x0bNVIDIA_V100\x10\x02\x12\x0f\n\x0bNVIDIA_A100\x10\x03\x12\r\n\tNVIDIA_T4\x10\x04\x12\x0f\n\x0bNVIDIA_A10G\x10\x05\x42.\n\x19\x63om.qwak.ai.batch.job.apiP\x01Z\x0f./;inferencejobb\x06proto3')
-
-_GPUTYPE = DESCRIPTOR.enum_types_by_name['GpuType']
-GpuType = enum_type_wrapper.EnumTypeWrapper(_GPUTYPE)
-INVALID_GPU = 0
-NVIDIA_K80 = 1
-NVIDIA_V100 = 2
-NVIDIA_A100 = 3
-NVIDIA_T4 = 4
-NVIDIA_A10G = 5
-
-
-_GPURESOURCES = DESCRIPTOR.message_types_by_name['GpuResources']
-GpuResources = _reflection.GeneratedProtocolMessageType('GpuResources', (_message.Message,), {
-  'DESCRIPTOR' : _GPURESOURCES,
-  '__module__' : 'qwak.batch_job.v1.batch_job_resources_pb2'
-  # @@protoc_insertion_point(class_scope:qwak.batchjob.GpuResources)
-  })
-_sym_db.RegisterMessage(GpuResources)
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n+qwak/batch_job/v1/batch_job_resources.proto\x12\rqwak.batchjob\"L\n\x0cGpuResources\x12(\n\x08gpu_type\x18\x01 \x01(\x0e\x32\x16.qwak.batchjob.GpuType\x12\x12\n\ngpu_amount\x18\x02 \x01(\x05*{\n\x07GpuType\x12\x0f\n\x0bINVALID_GPU\x10\x00\x12\x0e\n\nNVIDIA_K80\x10\x01\x12\x0f\n\x0bNVIDIA_V100\x10\x02\x12\x0f\n\x0bNVIDIA_A100\x10\x03\x12\r\n\tNVIDIA_T4\x10\x04\x12\x0f\n\x0bNVIDIA_A10G\x10\x05\x12\r\n\tNVIDIA_L4\x10\x06\x42.\n\x19\x63om.qwak.ai.batch.job.apiP\x01Z\x0f./;inferencejobb\x06proto3')
 
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.batch_job.v1.batch_job_resources_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'\n\031com.qwak.ai.batch.job.apiP\001Z\017./;inferencejob'
-  _GPUTYPE._serialized_start=140
-  _GPUTYPE._serialized_end=248
-  _GPURESOURCES._serialized_start=62
-  _GPURESOURCES._serialized_end=138
+  _globals['_GPUTYPE']._serialized_start=140
+  _globals['_GPUTYPE']._serialized_end=263
+  _globals['_GPURESOURCES']._serialized_start=62
+  _globals['_GPURESOURCES']._serialized_end=138
 # @@protoc_insertion_point(module_scope)

_qwak_proto/qwak/batch_job/v1/batch_job_resources_pb2.pyi

@@ -28,6 +28,7 @@ class _GpuTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTy
     NVIDIA_A100: _GpuType.ValueType  # 3
     NVIDIA_T4: _GpuType.ValueType  # 4
     NVIDIA_A10G: _GpuType.ValueType  # 5
+    NVIDIA_L4: _GpuType.ValueType  # 6
 
 class GpuType(_GpuType, metaclass=_GpuTypeEnumTypeWrapper): ...
 
@@ -37,6 +38,7 @@ NVIDIA_V100: GpuType.ValueType  # 2
 NVIDIA_A100: GpuType.ValueType  # 3
 NVIDIA_T4: GpuType.ValueType  # 4
 NVIDIA_A10G: GpuType.ValueType  # 5
+NVIDIA_L4: GpuType.ValueType  # 6
 global___GpuType = GpuType
 
 class GpuResources(google.protobuf.message.Message):