frogml-core 0.0.116__py3-none-any.whl → 0.0.118__py3-none-any.whl

_proto/qwak/automation/v1/notification_pb2.py

@@ -0,0 +1,41 @@
+# -*- 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 import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+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()
+
+
+
+
+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')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.automation.v1.notification_pb2', _globals)
+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'
+  _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)

_proto/qwak/automation/v1/trigger_pb2.py

@@ -0,0 +1,36 @@
+# -*- 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 symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@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')
+
+_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'
+  _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)

_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

_proto/qwak/batch_job/v1/batch_job_events_pb2.py

@@ -0,0 +1,58 @@
+# -*- 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 import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+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()
+
+
+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')
+
+_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
+  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'
+  _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)

_proto/qwak/batch_job/v1/batch_job_events_pb2.pyi

@@ -0,0 +1,307 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+import frogml._proto.qwak.user_application.common.v0.resources_pb2
+import sys
+import typing
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class _BatchJobEventTypeMessage:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _BatchJobEventTypeMessageEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_BatchJobEventTypeMessage.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    UNDEFINED_BATCH_JOB_EVENT: _BatchJobEventTypeMessage.ValueType  # 0
+    BATCH_JOB_COMMITTED: _BatchJobEventTypeMessage.ValueType  # 1
+    BATCH_JOB_PREPARED: _BatchJobEventTypeMessage.ValueType  # 7
+    BATCH_JOB_TASKS_CREATED: _BatchJobEventTypeMessage.ValueType  # 2
+    BATCH_JOB_CANCELLED: _BatchJobEventTypeMessage.ValueType  # 3
+    BATCH_JOB_CLEANUP: _BatchJobEventTypeMessage.ValueType  # 4
+    BATCH_JOB_CHECK_FINISHED: _BatchJobEventTypeMessage.ValueType  # 5
+    BATCH_JOB_CHECK_TIMEOUT: _BatchJobEventTypeMessage.ValueType  # 6
+
+class BatchJobEventTypeMessage(_BatchJobEventTypeMessage, metaclass=_BatchJobEventTypeMessageEnumTypeWrapper): ...
+
+UNDEFINED_BATCH_JOB_EVENT: BatchJobEventTypeMessage.ValueType  # 0
+BATCH_JOB_COMMITTED: BatchJobEventTypeMessage.ValueType  # 1
+BATCH_JOB_PREPARED: BatchJobEventTypeMessage.ValueType  # 7
+BATCH_JOB_TASKS_CREATED: BatchJobEventTypeMessage.ValueType  # 2
+BATCH_JOB_CANCELLED: BatchJobEventTypeMessage.ValueType  # 3
+BATCH_JOB_CLEANUP: BatchJobEventTypeMessage.ValueType  # 4
+BATCH_JOB_CHECK_FINISHED: BatchJobEventTypeMessage.ValueType  # 5
+BATCH_JOB_CHECK_TIMEOUT: BatchJobEventTypeMessage.ValueType  # 6
+global___BatchJobEventTypeMessage = BatchJobEventTypeMessage
+
+class _BatchTaskEventTypeMessage:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _BatchTaskEventTypeMessageEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_BatchTaskEventTypeMessage.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    UNDEFINED_BATCH_TASK_EVENT: _BatchTaskEventTypeMessage.ValueType  # 0
+    BATCH_TASK_RUNNING: _BatchTaskEventTypeMessage.ValueType  # 1
+    BATCH_TASK_FINISHED: _BatchTaskEventTypeMessage.ValueType  # 2
+    BATCH_TASK_FAILED: _BatchTaskEventTypeMessage.ValueType  # 3
+    BATCH_TASK_CHECK_STATUS: _BatchTaskEventTypeMessage.ValueType  # 4
+    BATCH_TASK_START: _BatchTaskEventTypeMessage.ValueType  # 5
+    BATCH_TASK_CLEANUP: _BatchTaskEventTypeMessage.ValueType  # 6
+
+class BatchTaskEventTypeMessage(_BatchTaskEventTypeMessage, metaclass=_BatchTaskEventTypeMessageEnumTypeWrapper): ...
+
+UNDEFINED_BATCH_TASK_EVENT: BatchTaskEventTypeMessage.ValueType  # 0
+BATCH_TASK_RUNNING: BatchTaskEventTypeMessage.ValueType  # 1
+BATCH_TASK_FINISHED: BatchTaskEventTypeMessage.ValueType  # 2
+BATCH_TASK_FAILED: BatchTaskEventTypeMessage.ValueType  # 3
+BATCH_TASK_CHECK_STATUS: BatchTaskEventTypeMessage.ValueType  # 4
+BATCH_TASK_START: BatchTaskEventTypeMessage.ValueType  # 5
+BATCH_TASK_CLEANUP: BatchTaskEventTypeMessage.ValueType  # 6
+global___BatchTaskEventTypeMessage = BatchTaskEventTypeMessage
+
+class _WarmupEventTypeMessage:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _WarmupEventTypeMessageEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_WarmupEventTypeMessage.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    UNDEFINED_WARMUP_JOB_EVENT: _WarmupEventTypeMessage.ValueType  # 0
+    WARMUP_JOB_COMMITTED: _WarmupEventTypeMessage.ValueType  # 1
+    WARMUP_CANCEL: _WarmupEventTypeMessage.ValueType  # 2
+
+class WarmupEventTypeMessage(_WarmupEventTypeMessage, metaclass=_WarmupEventTypeMessageEnumTypeWrapper): ...
+
+UNDEFINED_WARMUP_JOB_EVENT: WarmupEventTypeMessage.ValueType  # 0
+WARMUP_JOB_COMMITTED: WarmupEventTypeMessage.ValueType  # 1
+WARMUP_CANCEL: WarmupEventTypeMessage.ValueType  # 2
+global___WarmupEventTypeMessage = WarmupEventTypeMessage
+
+class _MemoryUnit:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _MemoryUnitEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_MemoryUnit.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    UNKNOWN: _MemoryUnit.ValueType  # 0
+    MIB: _MemoryUnit.ValueType  # 1
+    GIB: _MemoryUnit.ValueType  # 2
+
+class MemoryUnit(_MemoryUnit, metaclass=_MemoryUnitEnumTypeWrapper): ...
+
+UNKNOWN: MemoryUnit.ValueType  # 0
+MIB: MemoryUnit.ValueType  # 1
+GIB: MemoryUnit.ValueType  # 2
+global___MemoryUnit = MemoryUnit
+
+class _TaskFailureReasonCode:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _TaskFailureReasonCodeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_TaskFailureReasonCode.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    TASK_FAILURE_REASON_CODE_FAILURE_CODE_UNSPECIFIED: _TaskFailureReasonCode.ValueType  # 0
+    TASK_FAILURE_REASON_CODE_NO_REPLICA_SETS_PODS_FOUND_FAILURE_CODE: _TaskFailureReasonCode.ValueType  # 1
+    TASK_FAILURE_REASON_CODE_MODEL_CONTAINER_NOT_FOUND_FAILURE_CODE: _TaskFailureReasonCode.ValueType  # 2
+    TASK_FAILURE_REASON_CODE_UNSCHEDULABLE_FAILURE_CODE: _TaskFailureReasonCode.ValueType  # 3
+    TASK_FAILURE_REASON_CODE_DOCKER_IMAGE_NOT_FOUND_FAILURE_CODE: _TaskFailureReasonCode.ValueType  # 4
+    TASK_FAILURE_REASON_CODE_MEMORY_LIMIT_EXCEEDED_FAILURE_CODE: _TaskFailureReasonCode.ValueType  # 5
+    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): ...
+
+TASK_FAILURE_REASON_CODE_FAILURE_CODE_UNSPECIFIED: TaskFailureReasonCode.ValueType  # 0
+TASK_FAILURE_REASON_CODE_NO_REPLICA_SETS_PODS_FOUND_FAILURE_CODE: TaskFailureReasonCode.ValueType  # 1
+TASK_FAILURE_REASON_CODE_MODEL_CONTAINER_NOT_FOUND_FAILURE_CODE: TaskFailureReasonCode.ValueType  # 2
+TASK_FAILURE_REASON_CODE_UNSCHEDULABLE_FAILURE_CODE: TaskFailureReasonCode.ValueType  # 3
+TASK_FAILURE_REASON_CODE_DOCKER_IMAGE_NOT_FOUND_FAILURE_CODE: TaskFailureReasonCode.ValueType  # 4
+TASK_FAILURE_REASON_CODE_MEMORY_LIMIT_EXCEEDED_FAILURE_CODE: TaskFailureReasonCode.ValueType  # 5
+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):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    BATCH_JOB_EVENT_TYPE_FIELD_NUMBER: builtins.int
+    BATCH_JOB_ID_FIELD_NUMBER: builtins.int
+    MESSAGE_FIELD_NUMBER: builtins.int
+    DELAY_FIELD_NUMBER: builtins.int
+    FAILURES_FIELD_NUMBER: builtins.int
+    batch_job_event_type: global___BatchJobEventTypeMessage.ValueType
+    batch_job_id: builtins.str
+    message: builtins.str
+    delay: builtins.int
+    failures: builtins.int
+    def __init__(
+        self,
+        *,
+        batch_job_event_type: global___BatchJobEventTypeMessage.ValueType = ...,
+        batch_job_id: builtins.str = ...,
+        message: builtins.str = ...,
+        delay: builtins.int = ...,
+        failures: builtins.int = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["batch_job_event_type", b"batch_job_event_type", "batch_job_id", b"batch_job_id", "delay", b"delay", "failures", b"failures", "message", b"message"]) -> None: ...
+
+global___BatchJobEventMessage = BatchJobEventMessage
+
+class BatchTaskEventMessage(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    BATCH_TASK_EVENT_TYPE_FIELD_NUMBER: builtins.int
+    BATCH_JOB_ID_FIELD_NUMBER: builtins.int
+    BATCH_TASK_ID_FIELD_NUMBER: builtins.int
+    MESSAGE_FIELD_NUMBER: builtins.int
+    DELAY_FIELD_NUMBER: builtins.int
+    FAILURES_FIELD_NUMBER: builtins.int
+    BATCH_TASK_FAILURE_REASON_FIELD_NUMBER: builtins.int
+    batch_task_event_type: global___BatchTaskEventTypeMessage.ValueType
+    batch_job_id: builtins.str
+    batch_task_id: builtins.str
+    message: builtins.str
+    delay: builtins.int
+    failures: builtins.int
+    @property
+    def batch_task_failure_reason(self) -> global___TaskFailureReason: ...
+    def __init__(
+        self,
+        *,
+        batch_task_event_type: global___BatchTaskEventTypeMessage.ValueType = ...,
+        batch_job_id: builtins.str = ...,
+        batch_task_id: builtins.str = ...,
+        message: builtins.str = ...,
+        delay: builtins.int = ...,
+        failures: builtins.int = ...,
+        batch_task_failure_reason: global___TaskFailureReason | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["batch_task_failure_reason", b"batch_task_failure_reason"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["batch_job_id", b"batch_job_id", "batch_task_event_type", b"batch_task_event_type", "batch_task_failure_reason", b"batch_task_failure_reason", "batch_task_id", b"batch_task_id", "delay", b"delay", "failures", b"failures", "message", b"message"]) -> None: ...
+
+global___BatchTaskEventMessage = BatchTaskEventMessage
+
+class WarmupEventMessage(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    WARMUP_JOB_EVENT_TYPE_FIELD_NUMBER: builtins.int
+    DELAY_FIELD_NUMBER: builtins.int
+    FAILURES_FIELD_NUMBER: builtins.int
+    WARMUP_SIZE_FIELD_NUMBER: builtins.int
+    MODEL_IDENTIFIER_FIELD_NUMBER: builtins.int
+    warmup_job_event_type: global___WarmupEventTypeMessage.ValueType
+    delay: builtins.int
+    failures: builtins.int
+    @property
+    def warmup_size(self) -> global___WarmupSize: ...
+    @property
+    def model_identifier(self) -> global___ModelIdentifierMessage: ...
+    def __init__(
+        self,
+        *,
+        warmup_job_event_type: global___WarmupEventTypeMessage.ValueType = ...,
+        delay: builtins.int = ...,
+        failures: builtins.int = ...,
+        warmup_size: global___WarmupSize | None = ...,
+        model_identifier: global___ModelIdentifierMessage | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["model_identifier", b"model_identifier", "warmup_size", b"warmup_size"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["delay", b"delay", "failures", b"failures", "model_identifier", b"model_identifier", "warmup_job_event_type", b"warmup_job_event_type", "warmup_size", b"warmup_size"]) -> None: ...
+
+global___WarmupEventMessage = WarmupEventMessage
+
+class WarmupSize(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    CPU_FIELD_NUMBER: builtins.int
+    MEMORY_AMOUNT_FIELD_NUMBER: builtins.int
+    MEMORY_UNIT_FIELD_NUMBER: builtins.int
+    NUM_OF_PODS_FIELD_NUMBER: builtins.int
+    TIMEOUT_SECONDS_FIELD_NUMBER: builtins.int
+    GPU_RESOURCES_FIELD_NUMBER: builtins.int
+    POD_COMPUTE_RESOURCES_FIELD_NUMBER: builtins.int
+    cpu: builtins.float
+    memory_amount: builtins.int
+    memory_unit: global___MemoryUnit.ValueType
+    num_of_pods: builtins.int
+    timeout_seconds: builtins.int
+    """The timeout for the warmup"""
+    @property
+    def gpu_resources(self) -> frogml._proto.qwak.user_application.common.v0.resources_pb2.GpuResources:
+        """Optional GPU resources for batch"""
+    @property
+    def pod_compute_resources(self) -> frogml._proto.qwak.user_application.common.v0.resources_pb2.PodComputeResources: ...
+    def __init__(
+        self,
+        *,
+        cpu: builtins.float = ...,
+        memory_amount: builtins.int = ...,
+        memory_unit: global___MemoryUnit.ValueType = ...,
+        num_of_pods: builtins.int = ...,
+        timeout_seconds: builtins.int = ...,
+        gpu_resources: frogml._proto.qwak.user_application.common.v0.resources_pb2.GpuResources | None = ...,
+        pod_compute_resources: frogml._proto.qwak.user_application.common.v0.resources_pb2.PodComputeResources | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["gpu_resources", b"gpu_resources", "pod_compute_resources", b"pod_compute_resources"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["cpu", b"cpu", "gpu_resources", b"gpu_resources", "memory_amount", b"memory_amount", "memory_unit", b"memory_unit", "num_of_pods", b"num_of_pods", "pod_compute_resources", b"pod_compute_resources", "timeout_seconds", b"timeout_seconds"]) -> None: ...
+
+global___WarmupSize = WarmupSize
+
+class ModelIdentifierMessage(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_ID_FIELD_NUMBER: builtins.int
+    BRANCH_ID_FIELD_NUMBER: builtins.int
+    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,
+        *,
+        model_id: builtins.str = ...,
+        branch_id: builtins.str = ...,
+        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["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
+
+class TaskFailureReason(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    TASK_FAILURE_REASON_CODE_FIELD_NUMBER: builtins.int
+    TASK_FAILURE_REASON_DETAILS_FIELD_NUMBER: builtins.int
+    task_failure_reason_code: global___TaskFailureReasonCode.ValueType
+    task_failure_reason_details: builtins.str
+    def __init__(
+        self,
+        *,
+        task_failure_reason_code: global___TaskFailureReasonCode.ValueType = ...,
+        task_failure_reason_details: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["task_failure_reason_code", b"task_failure_reason_code", "task_failure_reason_details", b"task_failure_reason_details"]) -> None: ...
+
+global___TaskFailureReason = TaskFailureReason

_proto/qwak/batch_job/v1/batch_job_resources_pb2.py

@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.batch_job/v1/batch_job_resources.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 symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_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*{\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'
+  _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)