frogml 1.1.62__py3-none-any.whl → 1.2.19__py3-none-any.whl

frogml/_proto/qwak/automation/v1/common_pb2.pyi

@@ -2,6 +2,7 @@
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
 """
+
 import builtins
 import google.protobuf.descriptor
 import google.protobuf.internal.enum_type_wrapper
@@ -20,7 +21,7 @@ class _MetricThresholdDirection:
     ValueType = typing.NewType("ValueType", builtins.int)
     V: typing_extensions.TypeAlias = ValueType
 
-class _MetricThresholdDirectionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_MetricThresholdDirection.ValueType], builtins.type):  # noqa: F821
+class _MetricThresholdDirectionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_MetricThresholdDirection.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     INVALID_METRIC_DIRECTION: _MetricThresholdDirection.ValueType  # 0
     BELOW: _MetricThresholdDirection.ValueType  # 1
@@ -39,6 +40,7 @@ BELOW_OR_EQUALS: MetricThresholdDirection.ValueType  # 4
 ABOVE_OR_EQUALS: MetricThresholdDirection.ValueType  # 5
 global___MetricThresholdDirection = MetricThresholdDirection
 
+@typing.final
 class Metric(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -50,12 +52,13 @@ class Metric(google.protobuf.message.Message):
         *,
         sql_metric: global___SqlMetric | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["metric", b"metric", "sql_metric", b"sql_metric"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["metric", b"metric", "sql_metric", b"sql_metric"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["metric", b"metric"]) -> typing_extensions.Literal["sql_metric"] | None: ...
+    def HasField(self, field_name: typing.Literal["metric", b"metric", "sql_metric", b"sql_metric"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["metric", b"metric", "sql_metric", b"sql_metric"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["metric", b"metric"]) -> typing.Literal["sql_metric"] | None: ...
 
 global___Metric = Metric
 
+@typing.final
 class SqlMetric(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -66,6 +69,6 @@ class SqlMetric(google.protobuf.message.Message):
         *,
         sql_query: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["sql_query", b"sql_query"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["sql_query", b"sql_query"]) -> None: ...
 
 global___SqlMetric = SqlMetric

frogml/_proto/qwak/automation/v1/notification_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.qwak.automation/v1/notification.proto
+# Protobuf Python Version: 4.25.1
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
@@ -19,13 +20,12 @@ _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['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\035com.qwak.ai.automation.api.v1P\001'
+  _globals['_CUSTOMWEBHOOK_HEADERSENTRY']._options = None
+  _globals['_CUSTOMWEBHOOK_HEADERSENTRY']._serialized_options = b'8\001'
+  _globals['_CUSTOMWEBHOOK_DATAENTRY']._options = None
+  _globals['_CUSTOMWEBHOOK_DATAENTRY']._serialized_options = b'8\001'
   _globals['_HTTPMETHODTYPE']._serialized_start=581
   _globals['_HTTPMETHODTYPE']._serialized_end=790
   _globals['_NOTIFICATION']._serialized_start=62

frogml/_proto/qwak/automation/v1/notification_pb2.pyi

@@ -2,6 +2,7 @@
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
 """
+
 import builtins
 import collections.abc
 import google.protobuf.descriptor
@@ -22,7 +23,7 @@ class _HttpMethodType:
     ValueType = typing.NewType("ValueType", builtins.int)
     V: typing_extensions.TypeAlias = ValueType
 
-class _HttpMethodTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_HttpMethodType.ValueType], builtins.type):  # noqa: F821
+class _HttpMethodTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_HttpMethodType.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     HTTP_METHOD_TYPE_INVALID: _HttpMethodType.ValueType  # 0
     HTTP_METHOD_TYPE_GET: _HttpMethodType.ValueType  # 1
@@ -43,6 +44,7 @@ HTTP_METHOD_TYPE_PATCH: HttpMethodType.ValueType  # 5
 HTTP_METHOD_TYPE_HEAD: HttpMethodType.ValueType  # 6
 global___HttpMethodType = HttpMethodType
 
+@typing.final
 class Notification(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -58,12 +60,13 @@ class Notification(google.protobuf.message.Message):
         post_slack_notification: global___PostSlackNotification | None = ...,
         custom_webhook: global___CustomWebhook | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["custom_webhook", b"custom_webhook", "notification", b"notification", "post_slack_notification", b"post_slack_notification"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["custom_webhook", b"custom_webhook", "notification", b"notification", "post_slack_notification", b"post_slack_notification"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["notification", b"notification"]) -> typing_extensions.Literal["post_slack_notification", "custom_webhook"] | None: ...
+    def HasField(self, field_name: typing.Literal["custom_webhook", b"custom_webhook", "notification", b"notification", "post_slack_notification", b"post_slack_notification"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["custom_webhook", b"custom_webhook", "notification", b"notification", "post_slack_notification", b"post_slack_notification"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["notification", b"notification"]) -> typing.Literal["post_slack_notification", "custom_webhook"] | None: ...
 
 global___Notification = Notification
 
+@typing.final
 class PostSlackNotification(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -74,13 +77,15 @@ class PostSlackNotification(google.protobuf.message.Message):
         *,
         webhook: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["webhook", b"webhook"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["webhook", b"webhook"]) -> None: ...
 
 global___PostSlackNotification = PostSlackNotification
 
+@typing.final
 class CustomWebhook(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
+    @typing.final
     class HeadersEntry(google.protobuf.message.Message):
         DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -94,8 +99,9 @@ class CustomWebhook(google.protobuf.message.Message):
             key: builtins.str = ...,
             value: builtins.str = ...,
         ) -> None: ...
-        def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
+        def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
 
+    @typing.final
     class DataEntry(google.protobuf.message.Message):
         DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -109,7 +115,7 @@ class CustomWebhook(google.protobuf.message.Message):
             key: builtins.str = ...,
             value: builtins.str = ...,
         ) -> None: ...
-        def ClearField(self, field_name: typing_extensions.Literal["key", b"key", "value", b"value"]) -> None: ...
+        def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
 
     URL_FIELD_NUMBER: builtins.int
     HTTP_METHOD_FIELD_NUMBER: builtins.int
@@ -129,6 +135,6 @@ class CustomWebhook(google.protobuf.message.Message):
         headers: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
         data: collections.abc.Mapping[builtins.str, builtins.str] | None = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["data", b"data", "headers", b"headers", "http_method", b"http_method", "url", b"url"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["data", b"data", "headers", b"headers", "http_method", b"http_method", "url", b"url"]) -> None: ...
 
 global___CustomWebhook = CustomWebhook

frogml/_proto/qwak/automation/v1/trigger_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.qwak.automation/v1/trigger.proto
+# Protobuf Python Version: 4.25.1
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
@@ -20,9 +21,8 @@ _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['DESCRIPTOR']._options = None
+  _globals['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

frogml/_proto/qwak/automation/v1/trigger_pb2.pyi

@@ -2,19 +2,16 @@
 @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
+import typing
 
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 
+@typing.final
 class Trigger(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -38,12 +35,13 @@ class Trigger(google.protobuf.message.Message):
         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: ...
+    def HasField(self, field_name: typing.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.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.Literal["trigger", b"trigger"]) -> typing.Literal["scheduled_trigger", "metric_based_trigger", "none_trigger", "on_boarding_trigger"] | None: ...
 
 global___Trigger = Trigger
 
+@typing.final
 class ScheduledTrigger(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -59,12 +57,13 @@ class ScheduledTrigger(google.protobuf.message.Message):
         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: ...
+    def HasField(self, field_name: typing.Literal["cron", b"cron", "interval", b"interval", "schedule", b"schedule"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["cron", b"cron", "interval", b"interval", "schedule", b"schedule"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["schedule", b"schedule"]) -> typing.Literal["cron", "interval"] | None: ...
 
 global___ScheduledTrigger = ScheduledTrigger
 
+@typing.final
 class MetricBasedTrigger(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -74,11 +73,11 @@ class MetricBasedTrigger(google.protobuf.message.Message):
     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
+    @property
+    def metric(self) -> frogml._proto.qwak.automation.v1.common_pb2.Metric: ...
     def __init__(
         self,
         *,
@@ -88,11 +87,12 @@ class MetricBasedTrigger(google.protobuf.message.Message):
         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: ...
+    def HasField(self, field_name: typing.Literal["metric", b"metric"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["metric", b"metric", "name", b"name", "override_cron", b"override_cron", "threshold", b"threshold", "threshold_direction", b"threshold_direction"]) -> None: ...
 
 global___MetricBasedTrigger = MetricBasedTrigger
 
+@typing.final
 class NoneTrigger(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -102,6 +102,7 @@ class NoneTrigger(google.protobuf.message.Message):
 
 global___NoneTrigger = NoneTrigger
 
+@typing.final
 class OnBoardingTrigger(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 

frogml/_proto/qwak/batch_job/v1/batch_job_events_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.qwak.batch_job/v1/batch_job_events.proto
+# Protobuf Python Version: 4.25.1
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
@@ -20,19 +21,18 @@ _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['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\031com.qwak.ai.batch.job.apiP\001Z\017./;inferencejob'
+  _globals['_WARMUPSIZE'].fields_by_name['cpu']._options = None
+  _globals['_WARMUPSIZE'].fields_by_name['cpu']._serialized_options = b'\030\001'
+  _globals['_WARMUPSIZE'].fields_by_name['memory_amount']._options = None
+  _globals['_WARMUPSIZE'].fields_by_name['memory_amount']._serialized_options = b'\030\001'
+  _globals['_WARMUPSIZE'].fields_by_name['memory_unit']._options = None
+  _globals['_WARMUPSIZE'].fields_by_name['memory_unit']._serialized_options = b'\030\001'
+  _globals['_WARMUPSIZE'].fields_by_name['gpu_resources']._options = None
+  _globals['_WARMUPSIZE'].fields_by_name['gpu_resources']._serialized_options = b'\030\001'
+  _globals['_MODELIDENTIFIERMESSAGE'].fields_by_name['branch_id']._options = None
+  _globals['_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

frogml/_proto/qwak/batch_job/v1/batch_job_events_pb2.pyi

@@ -2,6 +2,7 @@
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
 """
+
 import builtins
 import google.protobuf.descriptor
 import google.protobuf.internal.enum_type_wrapper
@@ -21,7 +22,7 @@ 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
+class _BatchJobEventTypeMessageEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_BatchJobEventTypeMessage.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     UNDEFINED_BATCH_JOB_EVENT: _BatchJobEventTypeMessage.ValueType  # 0
     BATCH_JOB_COMMITTED: _BatchJobEventTypeMessage.ValueType  # 1
@@ -48,7 +49,7 @@ 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
+class _BatchTaskEventTypeMessageEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_BatchTaskEventTypeMessage.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     UNDEFINED_BATCH_TASK_EVENT: _BatchTaskEventTypeMessage.ValueType  # 0
     BATCH_TASK_RUNNING: _BatchTaskEventTypeMessage.ValueType  # 1
@@ -73,7 +74,7 @@ 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
+class _WarmupEventTypeMessageEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_WarmupEventTypeMessage.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     UNDEFINED_WARMUP_JOB_EVENT: _WarmupEventTypeMessage.ValueType  # 0
     WARMUP_JOB_COMMITTED: _WarmupEventTypeMessage.ValueType  # 1
@@ -90,7 +91,7 @@ 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
+class _MemoryUnitEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_MemoryUnit.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     UNKNOWN: _MemoryUnit.ValueType  # 0
     MIB: _MemoryUnit.ValueType  # 1
@@ -107,7 +108,7 @@ 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
+class _TaskFailureReasonCodeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_TaskFailureReasonCode.ValueType], builtins.type):
     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
@@ -134,6 +135,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
 
+@typing.final
 class BatchJobEventMessage(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -156,10 +158,11 @@ class BatchJobEventMessage(google.protobuf.message.Message):
         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: ...
+    def ClearField(self, field_name: typing.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
 
+@typing.final
 class BatchTaskEventMessage(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -189,11 +192,12 @@ class BatchTaskEventMessage(google.protobuf.message.Message):
         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: ...
+    def HasField(self, field_name: typing.Literal["batch_task_failure_reason", b"batch_task_failure_reason"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.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
 
+@typing.final
 class WarmupEventMessage(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -218,11 +222,12 @@ class WarmupEventMessage(google.protobuf.message.Message):
         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: ...
+    def HasField(self, field_name: typing.Literal["model_identifier", b"model_identifier", "warmup_size", b"warmup_size"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.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
 
+@typing.final
 class WarmupSize(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -242,6 +247,7 @@ class WarmupSize(google.protobuf.message.Message):
     @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__(
@@ -255,11 +261,12 @@ class WarmupSize(google.protobuf.message.Message):
         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: ...
+    def HasField(self, field_name: typing.Literal["gpu_resources", b"gpu_resources", "pod_compute_resources", b"pod_compute_resources"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.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
 
+@typing.final
 class ModelIdentifierMessage(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -285,10 +292,11 @@ class ModelIdentifierMessage(google.protobuf.message.Message):
         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: ...
+    def ClearField(self, field_name: typing.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
 
+@typing.final
 class TaskFailureReason(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -302,6 +310,6 @@ class TaskFailureReason(google.protobuf.message.Message):
         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: ...
+    def ClearField(self, field_name: typing.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,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.qwak.batch_job/v1/batch_job_resources.proto
+# Protobuf Python Version: 4.25.1
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
@@ -19,9 +20,8 @@ _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['DESCRIPTOR']._options = None
+  _globals['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

frogml/_proto/qwak/batch_job/v1/batch_job_resources_pb2.pyi

@@ -2,6 +2,7 @@
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
 """
+
 import builtins
 import google.protobuf.descriptor
 import google.protobuf.internal.enum_type_wrapper
@@ -20,7 +21,7 @@ class _GpuType:
     ValueType = typing.NewType("ValueType", builtins.int)
     V: typing_extensions.TypeAlias = ValueType
 
-class _GpuTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_GpuType.ValueType], builtins.type):  # noqa: F821
+class _GpuTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_GpuType.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     INVALID_GPU: _GpuType.ValueType  # 0
     NVIDIA_K80: _GpuType.ValueType  # 1
@@ -41,6 +42,7 @@ NVIDIA_A10G: GpuType.ValueType  # 5
 NVIDIA_L4: GpuType.ValueType  # 6
 global___GpuType = GpuType
 
+@typing.final
 class GpuResources(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -56,6 +58,6 @@ class GpuResources(google.protobuf.message.Message):
         gpu_type: global___GpuType.ValueType = ...,
         gpu_amount: builtins.int = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["gpu_amount", b"gpu_amount", "gpu_type", b"gpu_type"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["gpu_amount", b"gpu_amount", "gpu_type", b"gpu_type"]) -> None: ...
 
 global___GpuResources = GpuResources