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

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

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.qwak.automation/v1/action.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
@@ -24,23 +25,22 @@ _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.automation.v1.action_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n\035com.qwak.ai.automation.api.v1P\001'
-  _DEPLOYMENTSIZE.fields_by_name['cpu']._options = None
-  _DEPLOYMENTSIZE.fields_by_name['cpu']._serialized_options = b'\030\001'
-  _DEPLOYMENTSIZE.fields_by_name['memory_amount']._options = None
-  _DEPLOYMENTSIZE.fields_by_name['memory_amount']._serialized_options = b'\030\001'
-  _DEPLOYMENTSIZE.fields_by_name['memory_units']._options = None
-  _DEPLOYMENTSIZE.fields_by_name['memory_units']._serialized_options = b'\030\001'
-  _DEPLOYMENTSIZE.fields_by_name['gpu_resources']._options = None
-  _DEPLOYMENTSIZE.fields_by_name['gpu_resources']._serialized_options = b'\030\001'
-  _BUILDSPEC_PARAMETERSENTRY._options = None
-  _BUILDSPEC_PARAMETERSENTRY._serialized_options = b'8\001'
-  _RESOURCES.fields_by_name['cpu_resources']._options = None
-  _RESOURCES.fields_by_name['cpu_resources']._serialized_options = b'\030\001'
-  _RESOURCES.fields_by_name['gpu_resources']._options = None
-  _RESOURCES.fields_by_name['gpu_resources']._serialized_options = b'\030\001'
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\035com.qwak.ai.automation.api.v1P\001'
+  _globals['_DEPLOYMENTSIZE'].fields_by_name['cpu']._options = None
+  _globals['_DEPLOYMENTSIZE'].fields_by_name['cpu']._serialized_options = b'\030\001'
+  _globals['_DEPLOYMENTSIZE'].fields_by_name['memory_amount']._options = None
+  _globals['_DEPLOYMENTSIZE'].fields_by_name['memory_amount']._serialized_options = b'\030\001'
+  _globals['_DEPLOYMENTSIZE'].fields_by_name['memory_units']._options = None
+  _globals['_DEPLOYMENTSIZE'].fields_by_name['memory_units']._serialized_options = b'\030\001'
+  _globals['_DEPLOYMENTSIZE'].fields_by_name['gpu_resources']._options = None
+  _globals['_DEPLOYMENTSIZE'].fields_by_name['gpu_resources']._serialized_options = b'\030\001'
+  _globals['_BUILDSPEC_PARAMETERSENTRY']._options = None
+  _globals['_BUILDSPEC_PARAMETERSENTRY']._serialized_options = b'8\001'
+  _globals['_RESOURCES'].fields_by_name['cpu_resources']._options = None
+  _globals['_RESOURCES'].fields_by_name['cpu_resources']._serialized_options = b'\030\001'
+  _globals['_RESOURCES'].fields_by_name['gpu_resources']._options = None
+  _globals['_RESOURCES'].fields_by_name['gpu_resources']._serialized_options = b'\030\001'
   _globals['_GPUTYPE']._serialized_start=3170
   _globals['_GPUTYPE']._serialized_end=3289
   _globals['_MEMORYUNIT']._serialized_start=3291

frogml/_proto/qwak/automation/v1/action_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
@@ -27,7 +28,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: _GpuType.ValueType  # 0
     NVIDIA_K80: _GpuType.ValueType  # 1
@@ -52,7 +53,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
@@ -70,7 +71,7 @@ class _PurchaseOption:
     ValueType = typing.NewType("ValueType", builtins.int)
     V: typing_extensions.TypeAlias = ValueType
 
-class _PurchaseOptionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_PurchaseOption.ValueType], builtins.type):  # noqa: F821
+class _PurchaseOptionEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_PurchaseOption.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     ON_DEMAND: _PurchaseOption.ValueType  # 0
     SPOT: _PurchaseOption.ValueType  # 1
@@ -81,6 +82,7 @@ ON_DEMAND: PurchaseOption.ValueType  # 0
 SPOT: PurchaseOption.ValueType  # 1
 global___PurchaseOption = PurchaseOption
 
+@typing.final
 class Action(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -96,12 +98,13 @@ class Action(google.protobuf.message.Message):
         build_deploy: global___BuildAndDeployAction | None = ...,
         batch_execution: global___BatchExecutionAction | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["action", b"action", "batch_execution", b"batch_execution", "build_deploy", b"build_deploy"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["action", b"action", "batch_execution", b"batch_execution", "build_deploy", b"build_deploy"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["action", b"action"]) -> typing_extensions.Literal["build_deploy", "batch_execution"] | None: ...
+    def HasField(self, field_name: typing.Literal["action", b"action", "batch_execution", b"batch_execution", "build_deploy", b"build_deploy"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["action", b"action", "batch_execution", b"batch_execution", "build_deploy", b"build_deploy"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["action", b"action"]) -> typing.Literal["build_deploy", "batch_execution"] | None: ...
 
 global___Action = Action
 
+@typing.final
 class BatchExecutionAction(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -113,11 +116,12 @@ class BatchExecutionAction(google.protobuf.message.Message):
         *,
         execute_batch_job: frogml._proto.qwak.batch_job.v1.batch_job_service_pb2.BatchJobRequest | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["execute_batch_job", b"execute_batch_job"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["execute_batch_job", b"execute_batch_job"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["execute_batch_job", b"execute_batch_job"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["execute_batch_job", b"execute_batch_job"]) -> None: ...
 
 global___BatchExecutionAction = BatchExecutionAction
 
+@typing.final
 class BuildAndDeployAction(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -137,11 +141,12 @@ class BuildAndDeployAction(google.protobuf.message.Message):
         deployment_condition: global___DeploymentCondition | None = ...,
         deployment_spec: global___DeploymentSpec | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["build_spec", b"build_spec", "deployment_condition", b"deployment_condition", "deployment_spec", b"deployment_spec"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_spec", b"build_spec", "deployment_condition", b"deployment_condition", "deployment_spec", b"deployment_spec"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["build_spec", b"build_spec", "deployment_condition", b"deployment_condition", "deployment_spec", b"deployment_spec"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["build_spec", b"build_spec", "deployment_condition", b"deployment_condition", "deployment_spec", b"deployment_spec"]) -> None: ...
 
 global___BuildAndDeployAction = BuildAndDeployAction
 
+@typing.final
 class DeploymentCondition(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -157,12 +162,13 @@ class DeploymentCondition(google.protobuf.message.Message):
         build_metric: global___BuildMetricCondition | None = ...,
         deployed_build_metric: global___DeployedBuildMetricCondition | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["build_metric", b"build_metric", "condition", b"condition", "deployed_build_metric", b"deployed_build_metric"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_metric", b"build_metric", "condition", b"condition", "deployed_build_metric", b"deployed_build_metric"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["condition", b"condition"]) -> typing_extensions.Literal["build_metric", "deployed_build_metric"] | None: ...
+    def HasField(self, field_name: typing.Literal["build_metric", b"build_metric", "condition", b"condition", "deployed_build_metric", b"deployed_build_metric"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["build_metric", b"build_metric", "condition", b"condition", "deployed_build_metric", b"deployed_build_metric"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["condition", b"condition"]) -> typing.Literal["build_metric", "deployed_build_metric"] | None: ...
 
 global___DeploymentCondition = DeploymentCondition
 
+@typing.final
 class BuildMetricCondition(google.protobuf.message.Message):
     """checks the metric of the current build the automation is executing"""
 
@@ -184,10 +190,11 @@ class BuildMetricCondition(google.protobuf.message.Message):
         threshold_direction: frogml._proto.qwak.automation.v1.common_pb2.MetricThresholdDirection.ValueType = ...,
         threshold: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["metric_name", b"metric_name", "threshold", b"threshold", "threshold_direction", b"threshold_direction"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["metric_name", b"metric_name", "threshold", b"threshold", "threshold_direction", b"threshold_direction"]) -> None: ...
 
 global___BuildMetricCondition = BuildMetricCondition
 
+@typing.final
 class DeployedBuildMetricCondition(google.protobuf.message.Message):
     """checks the metric of the deployed build (if there is one)"""
 
@@ -209,10 +216,11 @@ class DeployedBuildMetricCondition(google.protobuf.message.Message):
         threshold_direction: frogml._proto.qwak.automation.v1.common_pb2.MetricThresholdDirection.ValueType = ...,
         variation: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["metric_name", b"metric_name", "threshold_direction", b"threshold_direction", "variation", b"variation"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["metric_name", b"metric_name", "threshold_direction", b"threshold_direction", "variation", b"variation"]) -> None: ...
 
 global___DeployedBuildMetricCondition = DeployedBuildMetricCondition
 
+@typing.final
 class DeploymentSpec(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -221,19 +229,22 @@ class DeploymentSpec(google.protobuf.message.Message):
     SELECTED_VARIATION_NAME_FIELD_NUMBER: builtins.int
     ENVIRONMENTS_FIELD_NUMBER: builtins.int
     ENV_VARS_FIELD_NUMBER: builtins.int
+    selected_variation_name: builtins.str
+    """The selected variation name"""
     @property
     def deployment_size(self) -> global___DeploymentSize:
         """The deployment"""
+
     @property
     def advanced_options(self) -> global___AdvancedDeploymentOptions: ...
-    selected_variation_name: builtins.str
-    """The selected variation name"""
     @property
     def environments(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
         """The selected environment names"""
+
     @property
     def env_vars(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
         """Environment variables for the model code"""
+
     def __init__(
         self,
         *,
@@ -243,11 +254,12 @@ class DeploymentSpec(google.protobuf.message.Message):
         environments: collections.abc.Iterable[builtins.str] | None = ...,
         env_vars: collections.abc.Iterable[builtins.str] | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["advanced_options", b"advanced_options", "deployment_size", b"deployment_size"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["advanced_options", b"advanced_options", "deployment_size", b"deployment_size", "env_vars", b"env_vars", "environments", b"environments", "selected_variation_name", b"selected_variation_name"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["advanced_options", b"advanced_options", "deployment_size", b"deployment_size"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["advanced_options", b"advanced_options", "deployment_size", b"deployment_size", "env_vars", b"env_vars", "environments", b"environments", "selected_variation_name", b"selected_variation_name"]) -> None: ...
 
 global___DeploymentSpec = DeploymentSpec
 
+@typing.final
 class AdvancedDeploymentOptions(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -267,15 +279,16 @@ class AdvancedDeploymentOptions(google.protobuf.message.Message):
     """Serve model container in daemon mode"""
     custom_iam_role_arn: builtins.str
     """Custom IAM Role ARN"""
-    @property
-    def auto_scaling_config(self) -> frogml._proto.qwak.automation.v1.auto_scaling_pb2.AutoScalingConfig:
-        """Auto scaling config"""
     max_batch_size: builtins.int
     """Max batch size of records"""
     deployment_process_timeout_limit: builtins.int
     """The timeout for the deployment (in seconds)"""
     service_account_key_secret_name: builtins.str
     """Custom service account for GCP"""
+    @property
+    def auto_scaling_config(self) -> frogml._proto.qwak.automation.v1.auto_scaling_pb2.AutoScalingConfig:
+        """Auto scaling config"""
+
     def __init__(
         self,
         *,
@@ -288,11 +301,12 @@ class AdvancedDeploymentOptions(google.protobuf.message.Message):
         deployment_process_timeout_limit: builtins.int = ...,
         service_account_key_secret_name: builtins.str = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["auto_scaling_config", b"auto_scaling_config"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["auto_scaling_config", b"auto_scaling_config", "custom_iam_role_arn", b"custom_iam_role_arn", "daemon_mode", b"daemon_mode", "deployment_process_timeout_limit", b"deployment_process_timeout_limit", "http_request_timeout_ms", b"http_request_timeout_ms", "max_batch_size", b"max_batch_size", "number_of_http_server_workers", b"number_of_http_server_workers", "service_account_key_secret_name", b"service_account_key_secret_name"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["auto_scaling_config", b"auto_scaling_config"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["auto_scaling_config", b"auto_scaling_config", "custom_iam_role_arn", b"custom_iam_role_arn", "daemon_mode", b"daemon_mode", "deployment_process_timeout_limit", b"deployment_process_timeout_limit", "http_request_timeout_ms", b"http_request_timeout_ms", "max_batch_size", b"max_batch_size", "number_of_http_server_workers", b"number_of_http_server_workers", "service_account_key_secret_name", b"service_account_key_secret_name"]) -> None: ...
 
 global___AdvancedDeploymentOptions = AdvancedDeploymentOptions
 
+@typing.final
 class DeploymentSize(google.protobuf.message.Message):
     """Should be common - from deployment manager"""
 
@@ -315,9 +329,11 @@ class DeploymentSize(google.protobuf.message.Message):
     @property
     def gpu_resources(self) -> global___GpuResources:
         """GPU Resource for deployed model"""
+
     @property
     def client_pod_compute_resources(self) -> frogml._proto.qwak.user_application.common.v0.resources_pb2.ClientPodComputeResources:
         """Client pod compute resources"""
+
     def __init__(
         self,
         *,
@@ -328,14 +344,16 @@ class DeploymentSize(google.protobuf.message.Message):
         gpu_resources: global___GpuResources | None = ...,
         client_pod_compute_resources: frogml._proto.qwak.user_application.common.v0.resources_pb2.ClientPodComputeResources | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["client_pod_compute_resources", b"client_pod_compute_resources", "gpu_resources", b"gpu_resources"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["client_pod_compute_resources", b"client_pod_compute_resources", "cpu", b"cpu", "gpu_resources", b"gpu_resources", "memory_amount", b"memory_amount", "memory_units", b"memory_units", "number_of_pods", b"number_of_pods"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["client_pod_compute_resources", b"client_pod_compute_resources", "gpu_resources", b"gpu_resources"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["client_pod_compute_resources", b"client_pod_compute_resources", "cpu", b"cpu", "gpu_resources", b"gpu_resources", "memory_amount", b"memory_amount", "memory_units", b"memory_units", "number_of_pods", b"number_of_pods"]) -> None: ...
 
 global___DeploymentSize = DeploymentSize
 
+@typing.final
 class BuildSpec(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
+    @typing.final
     class ParametersEntry(google.protobuf.message.Message):
         DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -349,7 +367,7 @@ class BuildSpec(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: ...
 
     GIT_MODEL_SOURCE_FIELD_NUMBER: builtins.int
     PARAMETERS_FIELD_NUMBER: builtins.int
@@ -365,34 +383,38 @@ class BuildSpec(google.protobuf.message.Message):
     PURCHASE_OPTION_FIELD_NUMBER: builtins.int
     SERVICE_ACCOUNT_KEY_SECRET_NAME_FIELD_NUMBER: builtins.int
     PROVISION_INSTANCE_TIMEOUT_FIELD_NUMBER: builtins.int
+    base_image: builtins.str
+    assumed_iam_role: builtins.str
+    main_dir: builtins.str
+    """The main dir of the model"""
+    dependency_file_path: builtins.str
+    """The custom dependency file path."""
+    gpu_compatible: builtins.bool
+    """If the model is GPU compatible"""
+    purchase_option: global___PurchaseOption.ValueType
+    service_account_key_secret_name: builtins.str
+    """Custom service account for GCP"""
+    provision_instance_timeout: builtins.int
     @property
     def git_model_source(self) -> global___GitModelSource:
         """Git Details to get model from."""
+
     @property
     def parameters(self) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
         """Parameters for the build"""
+
     @property
     def tags(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
         """Tags to save on the build"""
-    base_image: builtins.str
-    assumed_iam_role: builtins.str
+
     @property
     def resource(self) -> global___Resources: ...
-    main_dir: builtins.str
-    """The main dir of the model"""
-    dependency_file_path: builtins.str
-    """The custom dependency file path."""
     @property
     def env_vars(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
         """Environment variables for the model code"""
-    gpu_compatible: builtins.bool
-    """If the model is GPU compatible"""
+
     @property
     def push_image(self) -> google.protobuf.wrappers_pb2.BoolValue: ...
-    purchase_option: global___PurchaseOption.ValueType
-    service_account_key_secret_name: builtins.str
-    """Custom service account for GCP"""
-    provision_instance_timeout: builtins.int
     def __init__(
         self,
         *,
@@ -411,11 +433,12 @@ class BuildSpec(google.protobuf.message.Message):
         service_account_key_secret_name: builtins.str = ...,
         provision_instance_timeout: builtins.int = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["git_model_source", b"git_model_source", "push_image", b"push_image", "resource", b"resource"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["assumed_iam_role", b"assumed_iam_role", "base_image", b"base_image", "dependency_file_path", b"dependency_file_path", "env_vars", b"env_vars", "git_model_source", b"git_model_source", "gpu_compatible", b"gpu_compatible", "main_dir", b"main_dir", "parameters", b"parameters", "provision_instance_timeout", b"provision_instance_timeout", "purchase_option", b"purchase_option", "push_image", b"push_image", "resource", b"resource", "service_account_key_secret_name", b"service_account_key_secret_name", "tags", b"tags"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["git_model_source", b"git_model_source", "push_image", b"push_image", "resource", b"resource"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["assumed_iam_role", b"assumed_iam_role", "base_image", b"base_image", "dependency_file_path", b"dependency_file_path", "env_vars", b"env_vars", "git_model_source", b"git_model_source", "gpu_compatible", b"gpu_compatible", "main_dir", b"main_dir", "parameters", b"parameters", "provision_instance_timeout", b"provision_instance_timeout", "purchase_option", b"purchase_option", "push_image", b"push_image", "resource", b"resource", "service_account_key_secret_name", b"service_account_key_secret_name", "tags", b"tags"]) -> None: ...
 
 global___BuildSpec = BuildSpec
 
+@typing.final
 class Resources(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -425,12 +448,15 @@ class Resources(google.protobuf.message.Message):
     @property
     def cpu_resources(self) -> global___CpuResources:
         """CPU Properties"""
+
     @property
     def gpu_resources(self) -> global___GpuResources:
         """GPU Properties"""
+
     @property
     def client_pod_compute_resources(self) -> frogml._proto.qwak.user_application.common.v0.resources_pb2.ClientPodComputeResources:
         """Client pod compute resources"""
+
     def __init__(
         self,
         *,
@@ -438,12 +464,13 @@ class Resources(google.protobuf.message.Message):
         gpu_resources: global___GpuResources | None = ...,
         client_pod_compute_resources: frogml._proto.qwak.user_application.common.v0.resources_pb2.ClientPodComputeResources | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["client_pod_compute_resources", b"client_pod_compute_resources", "cpu_resources", b"cpu_resources", "gpu_resources", b"gpu_resources", "resource", b"resource"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["client_pod_compute_resources", b"client_pod_compute_resources", "cpu_resources", b"cpu_resources", "gpu_resources", b"gpu_resources", "resource", b"resource"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["resource", b"resource"]) -> typing_extensions.Literal["cpu_resources", "gpu_resources", "client_pod_compute_resources"] | None: ...
+    def HasField(self, field_name: typing.Literal["client_pod_compute_resources", b"client_pod_compute_resources", "cpu_resources", b"cpu_resources", "gpu_resources", b"gpu_resources", "resource", b"resource"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["client_pod_compute_resources", b"client_pod_compute_resources", "cpu_resources", b"cpu_resources", "gpu_resources", b"gpu_resources", "resource", b"resource"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["resource", b"resource"]) -> typing.Literal["cpu_resources", "gpu_resources", "client_pod_compute_resources"] | None: ...
 
 global___Resources = Resources
 
+@typing.final
 class CpuResources(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -463,10 +490,11 @@ class CpuResources(google.protobuf.message.Message):
         memory_amount: builtins.int = ...,
         memory_units: global___MemoryUnit.ValueType = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["cpu", b"cpu", "memory_amount", b"memory_amount", "memory_units", b"memory_units"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["cpu", b"cpu", "memory_amount", b"memory_amount", "memory_units", b"memory_units"]) -> None: ...
 
 global___CpuResources = CpuResources
 
+@typing.final
 class GpuResources(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -482,10 +510,11 @@ 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
 
+@typing.final
 class GitModelSource(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -508,6 +537,6 @@ class GitModelSource(google.protobuf.message.Message):
         git_branch: builtins.str = ...,
         git_ssh_secret_name: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["git_branch", b"git_branch", "git_credentials_secret_name", b"git_credentials_secret_name", "git_ssh_secret_name", b"git_ssh_secret_name", "git_uri", b"git_uri"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["git_branch", b"git_branch", "git_credentials_secret_name", b"git_credentials_secret_name", "git_ssh_secret_name", b"git_ssh_secret_name", "git_uri", b"git_uri"]) -> None: ...
 
 global___GitModelSource = GitModelSource

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

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.qwak.automation/v1/auto_scaling.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.automation.v1.auto_scaling_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['_AUTOSCALINGMETRICTYPE']._serialized_start=689
   _globals['_AUTOSCALINGMETRICTYPE']._serialized_end=886
   _globals['_AUTOSCALINGAGGREGATIONTYPE']._serialized_start=889

frogml/_proto/qwak/automation/v1/auto_scaling_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 _AutoScalingMetricType:
     ValueType = typing.NewType("ValueType", builtins.int)
     V: typing_extensions.TypeAlias = ValueType
 
-class _AutoScalingMetricTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_AutoScalingMetricType.ValueType], builtins.type):  # noqa: F821
+class _AutoScalingMetricTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_AutoScalingMetricType.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     METRIC_TYPE_NOT_VALID: _AutoScalingMetricType.ValueType  # 0
     METRIC_TYPE_CPU: _AutoScalingMetricType.ValueType  # 1
@@ -47,7 +48,7 @@ class _AutoScalingAggregationType:
     ValueType = typing.NewType("ValueType", builtins.int)
     V: typing_extensions.TypeAlias = ValueType
 
-class _AutoScalingAggregationTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_AutoScalingAggregationType.ValueType], builtins.type):  # noqa: F821
+class _AutoScalingAggregationTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_AutoScalingAggregationType.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     AGGREGATION_TYPE_NOT_VALID: _AutoScalingAggregationType.ValueType  # 0
     AGGREGATION_TYPE_MIN: _AutoScalingAggregationType.ValueType  # 1
@@ -72,6 +73,7 @@ AGGREGATION_TYPE_P95: AutoScalingAggregationType.ValueType  # 7
 AGGREGATION_TYPE_P99: AutoScalingAggregationType.ValueType  # 8
 global___AutoScalingAggregationType = AutoScalingAggregationType
 
+@typing.final
 class AutoScalingConfig(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -99,11 +101,12 @@ class AutoScalingConfig(google.protobuf.message.Message):
         cool_down_period: builtins.int = ...,
         triggers: global___Triggers | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["triggers", b"triggers"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["cool_down_period", b"cool_down_period", "max_replica_count", b"max_replica_count", "min_replica_count", b"min_replica_count", "polling_interval", b"polling_interval", "triggers", b"triggers"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["triggers", b"triggers"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["cool_down_period", b"cool_down_period", "max_replica_count", b"max_replica_count", "min_replica_count", b"min_replica_count", "polling_interval", b"polling_interval", "triggers", b"triggers"]) -> None: ...
 
 global___AutoScalingConfig = AutoScalingConfig
 
+@typing.final
 class Triggers(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -111,15 +114,17 @@ class Triggers(google.protobuf.message.Message):
     @property
     def triggers(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ScaleTrigger]:
         """List of triggers to activate scaling(up/down) of the target resource based on"""
+
     def __init__(
         self,
         *,
         triggers: collections.abc.Iterable[global___ScaleTrigger] | None = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["triggers", b"triggers"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["triggers", b"triggers"]) -> None: ...
 
 global___Triggers = Triggers
 
+@typing.final
 class ScaleTrigger(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -127,38 +132,42 @@ class ScaleTrigger(google.protobuf.message.Message):
     @property
     def prometheus_trigger(self) -> global___AutoScalingPrometheusTrigger:
         """Prometheus trigger type"""
+
     def __init__(
         self,
         *,
         prometheus_trigger: global___AutoScalingPrometheusTrigger | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["prometheus_trigger", b"prometheus_trigger", "trigger_type", b"trigger_type"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["prometheus_trigger", b"prometheus_trigger", "trigger_type", b"trigger_type"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["trigger_type", b"trigger_type"]) -> typing_extensions.Literal["prometheus_trigger"] | None: ...
+    def HasField(self, field_name: typing.Literal["prometheus_trigger", b"prometheus_trigger", "trigger_type", b"trigger_type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["prometheus_trigger", b"prometheus_trigger", "trigger_type", b"trigger_type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["trigger_type", b"trigger_type"]) -> typing.Literal["prometheus_trigger"] | None: ...
 
 global___ScaleTrigger = ScaleTrigger
 
+@typing.final
 class AutoScalingPrometheusTrigger(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     QUERY_SPEC_FIELD_NUMBER: builtins.int
     THRESHOLD_FIELD_NUMBER: builtins.int
+    threshold: builtins.int
+    """Value to start scaling for"""
     @property
     def query_spec(self) -> global___QuerySpec:
         """Query specification"""
-    threshold: builtins.int
-    """Value to start scaling for"""
+
     def __init__(
         self,
         *,
         query_spec: global___QuerySpec | None = ...,
         threshold: builtins.int = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["query_spec", b"query_spec"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["query_spec", b"query_spec", "threshold", b"threshold"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["query_spec", b"query_spec"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["query_spec", b"query_spec", "threshold", b"threshold"]) -> None: ...
 
 global___AutoScalingPrometheusTrigger = AutoScalingPrometheusTrigger
 
+@typing.final
 class QuerySpec(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -178,6 +187,6 @@ class QuerySpec(google.protobuf.message.Message):
         aggregation_type: global___AutoScalingAggregationType.ValueType = ...,
         time_period: builtins.int = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["aggregation_type", b"aggregation_type", "metric_type", b"metric_type", "time_period", b"time_period"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["aggregation_type", b"aggregation_type", "metric_type", b"metric_type", "time_period", b"time_period"]) -> None: ...
 
 global___QuerySpec = QuerySpec

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

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.qwak.automation/v1/automation_execution.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.automation_execution_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['_EXECUTIONSTATUSMESSAGE']._serialized_start=628
   _globals['_EXECUTIONSTATUSMESSAGE']._serialized_end=708
   _globals['_AUTOMATIONEXECUTIONMESSAGE']._serialized_start=102