frogml-core 0.0.1__py3-none-any.whl → 0.0.2__py3-none-any.whl

frogml_proto/qwak/automation/v1/notification_pb2.py

@@ -0,0 +1,93 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml_proto.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
+# @@protoc_insertion_point(imports)
+
+_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)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\035com.qwak.ai.automation.api.v1P\001'
+  _CUSTOMWEBHOOK_HEADERSENTRY._options = None
+  _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
+# @@protoc_insertion_point(module_scope)

frogml_proto/qwak/automation/v1/trigger_pb2.py

@@ -0,0 +1,76 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml_proto.qwak.automation/v1/trigger.proto
+"""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
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from frogml_proto.qwak.automation.v1 import common_pb2 as qwak_dot_automation_dot_v1_dot_common__pb2
+
+
+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)
+
+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
+# @@protoc_insertion_point(module_scope)

frogml_proto/qwak/automation/v1/trigger_pb2.pyi

@@ -0,0 +1,112 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.message
+import frogml_proto.qwak.automation.v1.common_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class Trigger(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    SCHEDULED_TRIGGER_FIELD_NUMBER: builtins.int
+    METRIC_BASED_TRIGGER_FIELD_NUMBER: builtins.int
+    NONE_TRIGGER_FIELD_NUMBER: builtins.int
+    ON_BOARDING_TRIGGER_FIELD_NUMBER: builtins.int
+    @property
+    def scheduled_trigger(self) -> global___ScheduledTrigger: ...
+    @property
+    def metric_based_trigger(self) -> global___MetricBasedTrigger: ...
+    @property
+    def none_trigger(self) -> global___NoneTrigger: ...
+    @property
+    def on_boarding_trigger(self) -> global___OnBoardingTrigger: ...
+    def __init__(
+        self,
+        *,
+        scheduled_trigger: global___ScheduledTrigger | None = ...,
+        metric_based_trigger: global___MetricBasedTrigger | None = ...,
+        none_trigger: global___NoneTrigger | None = ...,
+        on_boarding_trigger: global___OnBoardingTrigger | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["metric_based_trigger", b"metric_based_trigger", "none_trigger", b"none_trigger", "on_boarding_trigger", b"on_boarding_trigger", "scheduled_trigger", b"scheduled_trigger", "trigger", b"trigger"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["metric_based_trigger", b"metric_based_trigger", "none_trigger", b"none_trigger", "on_boarding_trigger", b"on_boarding_trigger", "scheduled_trigger", b"scheduled_trigger", "trigger", b"trigger"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["trigger", b"trigger"]) -> typing_extensions.Literal["scheduled_trigger", "metric_based_trigger", "none_trigger", "on_boarding_trigger"] | None: ...
+
+global___Trigger = Trigger
+
+class ScheduledTrigger(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    CRON_FIELD_NUMBER: builtins.int
+    INTERVAL_FIELD_NUMBER: builtins.int
+    cron: builtins.str
+    """Cron defined schedule"""
+    interval: builtins.str
+    """Cron interval schedule"""
+    def __init__(
+        self,
+        *,
+        cron: builtins.str = ...,
+        interval: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["cron", b"cron", "interval", b"interval", "schedule", b"schedule"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["cron", b"cron", "interval", b"interval", "schedule", b"schedule"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["schedule", b"schedule"]) -> typing_extensions.Literal["cron", "interval"] | None: ...
+
+global___ScheduledTrigger = ScheduledTrigger
+
+class MetricBasedTrigger(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    NAME_FIELD_NUMBER: builtins.int
+    METRIC_FIELD_NUMBER: builtins.int
+    THRESHOLD_DIRECTION_FIELD_NUMBER: builtins.int
+    THRESHOLD_FIELD_NUMBER: builtins.int
+    OVERRIDE_CRON_FIELD_NUMBER: builtins.int
+    name: builtins.str
+    @property
+    def metric(self) -> frogml_proto.qwak.automation.v1.common_pb2.Metric: ...
+    threshold_direction: frogml_proto.qwak.automation.v1.common_pb2.MetricThresholdDirection.ValueType
+    threshold: builtins.str
+    override_cron: builtins.str
+    def __init__(
+        self,
+        *,
+        name: builtins.str = ...,
+        metric: frogml_proto.qwak.automation.v1.common_pb2.Metric | None = ...,
+        threshold_direction: frogml_proto.qwak.automation.v1.common_pb2.MetricThresholdDirection.ValueType = ...,
+        threshold: builtins.str = ...,
+        override_cron: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["metric", b"metric"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["metric", b"metric", "name", b"name", "override_cron", b"override_cron", "threshold", b"threshold", "threshold_direction", b"threshold_direction"]) -> None: ...
+
+global___MetricBasedTrigger = MetricBasedTrigger
+
+class NoneTrigger(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___NoneTrigger = NoneTrigger
+
+class OnBoardingTrigger(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___OnBoardingTrigger = OnBoardingTrigger

frogml_proto/qwak/batch_job/v1/batch_job_events_pb2.py

@@ -0,0 +1,148 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml_proto.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
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from frogml_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\"\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')
+
+_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
+TASK_FAILURE_REASON_CODE_DOCKER_IMAGE_GETS_THROTTLING_BY_CLOUD_PROVIDER_FAILURE_CODE = 9
+
+
+_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)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\031com.qwak.ai.batch.job.apiP\001Z\017./;inferencejob'
+  _WARMUPSIZE.fields_by_name['cpu']._options = None
+  _WARMUPSIZE.fields_by_name['cpu']._serialized_options = b'\030\001'
+  _WARMUPSIZE.fields_by_name['memory_amount']._options = None
+  _WARMUPSIZE.fields_by_name['memory_amount']._serialized_options = b'\030\001'
+  _WARMUPSIZE.fields_by_name['memory_unit']._options = None
+  _WARMUPSIZE.fields_by_name['memory_unit']._serialized_options = b'\030\001'
+  _WARMUPSIZE.fields_by_name['gpu_resources']._options = None
+  _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=1373
+  _BATCHJOBEVENTTYPEMESSAGE._serialized_end=1615
+  _BATCHTASKEVENTTYPEMESSAGE._serialized_start=1618
+  _BATCHTASKEVENTTYPEMESSAGE._serialized_end=1824
+  _WARMUPEVENTTYPEMESSAGE._serialized_start=1826
+  _WARMUPEVENTTYPEMESSAGE._serialized_end=1927
+  _MEMORYUNIT._serialized_start=1929
+  _MEMORYUNIT._serialized_end=1972
+  _TASKFAILUREREASONCODE._serialized_start=1975
+  _TASKFAILUREREASONCODE._serialized_end=2641
+  _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=1095
+  _MODELIDENTIFIERMESSAGE._serialized_end=1239
+  _TASKFAILUREREASON._serialized_start=1242
+  _TASKFAILUREREASON._serialized_end=1370
+# @@protoc_insertion_point(module_scope)