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

frogml_proto/qwak/automation/v1/notification_pb2.py

@@ -1,41 +0,0 @@
-# -*- 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)

frogml_proto/qwak/automation/v1/trigger_pb2.py

@@ -1,36 +0,0 @@
-# -*- 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)

frogml_proto/qwak/automation/v1/trigger_pb2.pyi

@@ -1,112 +0,0 @@
-"""
-@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

@@ -1,58 +0,0 @@
-# -*- 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)

frogml_proto/qwak/batch_job/v1/batch_job_events_pb2.pyi

@@ -1,307 +0,0 @@
-"""
-@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

frogml_proto/qwak/batch_job/v1/batch_job_resources_pb2.py

@@ -1,29 +0,0 @@
-# -*- 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)