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

frogml/_proto/qwak/fitness_service/constructs_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 _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_MEMORY_UNIT: _MemoryUnit.ValueType  # 0
     MIB: _MemoryUnit.ValueType  # 1
@@ -39,7 +40,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
@@ -62,6 +63,18 @@ class _GpuTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTy
     """a100.xl"""
     NVIDIA_V100_4_32_208: _GpuType.ValueType  # 12
     """v100.4xl"""
+    NVIDIA_H100_NVL: _GpuType.ValueType  # 13
+    """H100 NVL XLarge"""
+    NVIDIA_A10: _GpuType.ValueType  # 14
+    """A10 Small"""
+    NVIDIA_H100_SXM: _GpuType.ValueType  # 15
+    """H100 SXM 8XLarge"""
+    NVIDIA_H100_40GB: _GpuType.ValueType  # 16
+    """H100 40GB 8XLarge"""
+    NVIDIA_H100_80GB: _GpuType.ValueType  # 17
+    """H100 80GB 8XLarge"""
+    NVIDIA_M60: _GpuType.ValueType  # 18
+    """M60 XLarge"""
 
 class GpuType(_GpuType, metaclass=_GpuTypeEnumTypeWrapper): ...
 
@@ -86,8 +99,21 @@ NVIDIA_V100_1_8_52: GpuType.ValueType  # 11
 """a100.xl"""
 NVIDIA_V100_4_32_208: GpuType.ValueType  # 12
 """v100.4xl"""
+NVIDIA_H100_NVL: GpuType.ValueType  # 13
+"""H100 NVL XLarge"""
+NVIDIA_A10: GpuType.ValueType  # 14
+"""A10 Small"""
+NVIDIA_H100_SXM: GpuType.ValueType  # 15
+"""H100 SXM 8XLarge"""
+NVIDIA_H100_40GB: GpuType.ValueType  # 16
+"""H100 40GB 8XLarge"""
+NVIDIA_H100_80GB: GpuType.ValueType  # 17
+"""H100 80GB 8XLarge"""
+NVIDIA_M60: GpuType.ValueType  # 18
+"""M60 XLarge"""
 global___GpuType = GpuType
 
+@typing.final
 class BuildProperties(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -103,14 +129,16 @@ class BuildProperties(google.protobuf.message.Message):
     """Branch name. When none provided - 'main' will be the selected branch"""
     model_id: builtins.str
     """Model id"""
+    gpu_compatible: builtins.bool
+    """Is GPU Compatible"""
     @property
     def tags(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
         """Build tags"""
+
     @property
     def model_uri(self) -> global___ModelUriSpec:
         """Model URI spec"""
-    gpu_compatible: builtins.bool
-    """Is GPU Compatible"""
+
     def __init__(
         self,
         *,
@@ -121,11 +149,12 @@ class BuildProperties(google.protobuf.message.Message):
         model_uri: global___ModelUriSpec | None = ...,
         gpu_compatible: builtins.bool = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["model_uri", b"model_uri"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["branch", b"branch", "build_id", b"build_id", "gpu_compatible", b"gpu_compatible", "model_id", b"model_id", "model_uri", b"model_uri", "tags", b"tags"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["model_uri", b"model_uri"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["branch", b"branch", "build_id", b"build_id", "gpu_compatible", b"gpu_compatible", "model_id", b"model_id", "model_uri", b"model_uri", "tags", b"tags"]) -> None: ...
 
 global___BuildProperties = BuildProperties
 
+@typing.final
 class ModelUriSpec(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -157,10 +186,11 @@ class ModelUriSpec(google.protobuf.message.Message):
         commit_id: builtins.str = ...,
         main_dir: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["commit_id", b"commit_id", "git_branch", b"git_branch", "git_credentials", b"git_credentials", "git_credentials_secret", b"git_credentials_secret", "main_dir", b"main_dir", "uri", b"uri"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["commit_id", b"commit_id", "git_branch", b"git_branch", "git_credentials", b"git_credentials", "git_credentials_secret", b"git_credentials_secret", "main_dir", b"main_dir", "uri", b"uri"]) -> None: ...
 
 global___ModelUriSpec = ModelUriSpec
 
+@typing.final
 class BuildEnv(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -176,11 +206,12 @@ class BuildEnv(google.protobuf.message.Message):
         docker_env: global___DockerEnv | None = ...,
         python_env: global___PythonEnv | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["docker_env", b"docker_env", "python_env", b"python_env"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["docker_env", b"docker_env", "python_env", b"python_env"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["docker_env", b"docker_env", "python_env", b"python_env"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["docker_env", b"docker_env", "python_env", b"python_env"]) -> None: ...
 
 global___BuildEnv = BuildEnv
 
+@typing.final
 class DockerEnv(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -191,9 +222,6 @@ class DockerEnv(google.protobuf.message.Message):
     SERVICE_ACCOUNT_KEY_SECRET_NAME_FIELD_NUMBER: builtins.int
     base_image: builtins.str
     """Base image to use for serving"""
-    @property
-    def env_vars(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
-        """Environment variables for the model code"""
     no_cache: builtins.bool
     """Don't use remote cache to build the images, defaults to use cache,
     if the model dependencies changed significantly it could be faster and better to run with no cache
@@ -202,6 +230,10 @@ class DockerEnv(google.protobuf.message.Message):
     """Custom IAM role to be assumed by model"""
     service_account_key_secret_name: builtins.str
     """Service account key secret name for gcp"""
+    @property
+    def env_vars(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
+        """Environment variables for the model code"""
+
     def __init__(
         self,
         *,
@@ -211,10 +243,11 @@ class DockerEnv(google.protobuf.message.Message):
         assumed_iam_role_arn: builtins.str = ...,
         service_account_key_secret_name: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["assumed_iam_role_arn", b"assumed_iam_role_arn", "base_image", b"base_image", "env_vars", b"env_vars", "no_cache", b"no_cache", "service_account_key_secret_name", b"service_account_key_secret_name"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["assumed_iam_role_arn", b"assumed_iam_role_arn", "base_image", b"base_image", "env_vars", b"env_vars", "no_cache", b"no_cache", "service_account_key_secret_name", b"service_account_key_secret_name"]) -> None: ...
 
 global___DockerEnv = DockerEnv
 
+@typing.final
 class PythonEnv(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -238,10 +271,11 @@ class PythonEnv(google.protobuf.message.Message):
         qwak_sdk_version: builtins.str = ...,
         qwak_sdk_extra_index: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["git_credentials", b"git_credentials", "git_credentials_secret", b"git_credentials_secret", "qwak_sdk_extra_index", b"qwak_sdk_extra_index", "qwak_sdk_version", b"qwak_sdk_version"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["git_credentials", b"git_credentials", "git_credentials_secret", b"git_credentials_secret", "qwak_sdk_extra_index", b"qwak_sdk_extra_index", "qwak_sdk_version", b"qwak_sdk_version"]) -> None: ...
 
 global___PythonEnv = PythonEnv
 
+@typing.final
 class CpuResources(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -261,10 +295,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
 
@@ -292,6 +327,6 @@ class GpuResources(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", "gpu_amount", b"gpu_amount", "gpu_type", b"gpu_type", "memory_amount", b"memory_amount", "memory_units", b"memory_units"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["cpu", b"cpu", "gpu_amount", b"gpu_amount", "gpu_type", b"gpu_type", "memory_amount", b"memory_amount", "memory_units", b"memory_units"]) -> None: ...
 
 global___GpuResources = GpuResources

frogml/_proto/qwak/fitness_service/fitness_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.qwak.fitness_service/fitness.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.fitness_service.fitness_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n\037com.qwak.ai.fitness.service.apiP\001'
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\037com.qwak.ai.fitness.service.apiP\001'
   _globals['_PURCHASEOPTION']._serialized_start=793
   _globals['_PURCHASEOPTION']._serialized_end=894
   _globals['_BUILDSPEC']._serialized_start=100

frogml/_proto/qwak/fitness_service/fitness_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 _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
     INVALID_PURCHASE_OPTION: _PurchaseOption.ValueType  # 0
     SPOT_PURCHASE_OPTION: _PurchaseOption.ValueType  # 1
@@ -34,6 +35,7 @@ SPOT_PURCHASE_OPTION: PurchaseOption.ValueType  # 1
 ONDEMAND_PURCHASE_OPTION: PurchaseOption.ValueType  # 2
 global___PurchaseOption = PurchaseOption
 
+@typing.final
 class BuildSpec(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -50,25 +52,12 @@ class BuildSpec(google.protobuf.message.Message):
     PROVISION_INSTANCE_TIMEOUT_FIELD_NUMBER: builtins.int
     ARTIFACTORY_PROJECT_KEY_FIELD_NUMBER: builtins.int
     ARTIFACTORY_BASE_URL_FIELD_NUMBER: builtins.int
-    @property
-    def build_properties(self) -> frogml._proto.qwak.fitness_service.constructs_pb2.BuildProperties: ...
-    @property
-    def build_env(self) -> frogml._proto.qwak.fitness_service.constructs_pb2.BuildEnv: ...
-    @property
-    def cpu_resources(self) -> frogml._proto.qwak.fitness_service.constructs_pb2.CpuResources:
-        """CPU Properties"""
-    @property
-    def gpu_resources(self) -> frogml._proto.qwak.fitness_service.constructs_pb2.GpuResources:
-        """GPU Properties"""
     verbose: builtins.int
     """Logs verbose mode 0: warning, 1: info, 2: debug (by qwak.tools.logger.py VERBOSITY_LEVEL_MAPPING from frogml._proto.qwak.sdk)"""
     build_config: builtins.str
     """replace the config passed as proto json format"""
     build_v1_flag: builtins.bool
     """Build v1 flag"""
-    @property
-    def build_properties_v1(self) -> global___BuildPropertiesV1:
-        """Build Properties V1"""
     purchase_option: global___PurchaseOption.ValueType
     """Purchase Option"""
     build_destined_image: builtins.str
@@ -79,6 +68,22 @@ class BuildSpec(google.protobuf.message.Message):
     """artifactory project key"""
     artifactory_base_url: builtins.str
     """artifactory base url"""
+    @property
+    def build_properties(self) -> frogml._proto.qwak.fitness_service.constructs_pb2.BuildProperties: ...
+    @property
+    def build_env(self) -> frogml._proto.qwak.fitness_service.constructs_pb2.BuildEnv: ...
+    @property
+    def cpu_resources(self) -> frogml._proto.qwak.fitness_service.constructs_pb2.CpuResources:
+        """CPU Properties"""
+
+    @property
+    def gpu_resources(self) -> frogml._proto.qwak.fitness_service.constructs_pb2.GpuResources:
+        """GPU Properties"""
+
+    @property
+    def build_properties_v1(self) -> global___BuildPropertiesV1:
+        """Build Properties V1"""
+
     def __init__(
         self,
         *,
@@ -96,12 +101,13 @@ class BuildSpec(google.protobuf.message.Message):
         artifactory_project_key: builtins.str = ...,
         artifactory_base_url: builtins.str = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["Resources", b"Resources", "build_env", b"build_env", "build_properties", b"build_properties", "build_properties_v1", b"build_properties_v1", "cpu_resources", b"cpu_resources", "gpu_resources", b"gpu_resources"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["Resources", b"Resources", "artifactory_base_url", b"artifactory_base_url", "artifactory_project_key", b"artifactory_project_key", "build_config", b"build_config", "build_destined_image", b"build_destined_image", "build_env", b"build_env", "build_properties", b"build_properties", "build_properties_v1", b"build_properties_v1", "build_v1_flag", b"build_v1_flag", "cpu_resources", b"cpu_resources", "gpu_resources", b"gpu_resources", "provision_instance_timeout", b"provision_instance_timeout", "purchase_option", b"purchase_option", "verbose", b"verbose"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["Resources", b"Resources"]) -> typing_extensions.Literal["cpu_resources", "gpu_resources"] | None: ...
+    def HasField(self, field_name: typing.Literal["Resources", b"Resources", "build_env", b"build_env", "build_properties", b"build_properties", "build_properties_v1", b"build_properties_v1", "cpu_resources", b"cpu_resources", "gpu_resources", b"gpu_resources"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["Resources", b"Resources", "artifactory_base_url", b"artifactory_base_url", "artifactory_project_key", b"artifactory_project_key", "build_config", b"build_config", "build_destined_image", b"build_destined_image", "build_env", b"build_env", "build_properties", b"build_properties", "build_properties_v1", b"build_properties_v1", "build_v1_flag", b"build_v1_flag", "cpu_resources", b"cpu_resources", "gpu_resources", b"gpu_resources", "provision_instance_timeout", b"provision_instance_timeout", "purchase_option", b"purchase_option", "verbose", b"verbose"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["Resources", b"Resources"]) -> typing.Literal["cpu_resources", "gpu_resources"] | None: ...
 
 global___BuildSpec = BuildSpec
 
+@typing.final
 class BuildPropertiesV1(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -121,6 +127,6 @@ class BuildPropertiesV1(google.protobuf.message.Message):
         qwak_sdk_wheel_url: builtins.str = ...,
         qwak_sdk_version_url: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_config_url", b"build_config_url", "qwak_sdk_version_url", b"qwak_sdk_version_url", "qwak_sdk_wheel_url", b"qwak_sdk_wheel_url"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["build_config_url", b"build_config_url", "qwak_sdk_version_url", b"qwak_sdk_version_url", "qwak_sdk_wheel_url", b"qwak_sdk_wheel_url"]) -> None: ...
 
 global___BuildPropertiesV1 = BuildPropertiesV1

frogml/_proto/qwak/fitness_service/fitness_service_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.qwak.fitness_service/fitness_service.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
@@ -21,9 +22,8 @@ _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.fitness_service.fitness_service_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n\037com.qwak.ai.fitness.service.apiP\001'
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\037com.qwak.ai.fitness.service.apiP\001'
   _globals['_BUILDMODELREQUEST']._serialized_start=139
   _globals['_BUILDMODELREQUEST']._serialized_end=211
   _globals['_BUILDMODELRESPONSE']._serialized_start=213

frogml/_proto/qwak/fitness_service/fitness_service_pb2.pyi

@@ -2,20 +2,17 @@
 @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.fitness_service.fitness_pb2
 import frogml._proto.qwak.fitness_service.status_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 BuildModelRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -23,16 +20,18 @@ class BuildModelRequest(google.protobuf.message.Message):
     @property
     def build_spec(self) -> frogml._proto.qwak.fitness_service.fitness_pb2.BuildSpec:
         """Requested build model spec"""
+
     def __init__(
         self,
         *,
         build_spec: frogml._proto.qwak.fitness_service.fitness_pb2.BuildSpec | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["build_spec", b"build_spec"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_spec", b"build_spec"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["build_spec", b"build_spec"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["build_spec", b"build_spec"]) -> None: ...
 
 global___BuildModelRequest = BuildModelRequest
 
+@typing.final
 class BuildModelResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -42,6 +41,7 @@ class BuildModelResponse(google.protobuf.message.Message):
 
 global___BuildModelResponse = BuildModelResponse
 
+@typing.final
 class CancelBuildModelRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -53,10 +53,11 @@ class CancelBuildModelRequest(google.protobuf.message.Message):
         *,
         build_id: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_id", b"build_id"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["build_id", b"build_id"]) -> None: ...
 
 global___CancelBuildModelRequest = CancelBuildModelRequest
 
+@typing.final
 class CancelBuildModelResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -66,6 +67,7 @@ class CancelBuildModelResponse(google.protobuf.message.Message):
 
 global___CancelBuildModelResponse = CancelBuildModelResponse
 
+@typing.final
 class GetRemoteBuildingStatusRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -77,10 +79,11 @@ class GetRemoteBuildingStatusRequest(google.protobuf.message.Message):
         *,
         build_id: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_id", b"build_id"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["build_id", b"build_id"]) -> None: ...
 
 global___GetRemoteBuildingStatusRequest = GetRemoteBuildingStatusRequest
 
+@typing.final
 class GetRemoteBuildingStatusResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -92,11 +95,12 @@ class GetRemoteBuildingStatusResponse(google.protobuf.message.Message):
         *,
         status: frogml._proto.qwak.fitness_service.status_pb2.RemoteBuildingStatus | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["status", b"status"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["status", b"status"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["status", b"status"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["status", b"status"]) -> None: ...
 
 global___GetRemoteBuildingStatusResponse = GetRemoteBuildingStatusResponse
 
+@typing.final
 class CleanRemoteBuildResourcesRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -107,10 +111,11 @@ class CleanRemoteBuildResourcesRequest(google.protobuf.message.Message):
         *,
         build_id: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_id", b"build_id"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["build_id", b"build_id"]) -> None: ...
 
 global___CleanRemoteBuildResourcesRequest = CleanRemoteBuildResourcesRequest
 
+@typing.final
 class CleanRemoteBuildResourcesResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 

frogml/_proto/qwak/fitness_service/status_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.qwak.fitness_service/status.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.fitness_service.status_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n\037com.qwak.ai.fitness.service.apiP\001'
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\037com.qwak.ai.fitness.service.apiP\001'
   _globals['_REMOTEBUILDINGSTATUSCODE']._serialized_start=1343
   _globals['_REMOTEBUILDINGSTATUSCODE']._serialized_end=1495
   _globals['_REMOTEBUILDINGFAILUREREASONCODE']._serialized_start=1498

frogml/_proto/qwak/fitness_service/status_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 _RemoteBuildingStatusCode:
     ValueType = typing.NewType("ValueType", builtins.int)
     V: typing_extensions.TypeAlias = ValueType
 
-class _RemoteBuildingStatusCodeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_RemoteBuildingStatusCode.ValueType], builtins.type):  # noqa: F821
+class _RemoteBuildingStatusCodeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_RemoteBuildingStatusCode.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     INVALID_STATUS: _RemoteBuildingStatusCode.ValueType  # 0
     """Invalid remote build status"""
@@ -65,7 +66,7 @@ class _RemoteBuildingFailureReasonCode:
     ValueType = typing.NewType("ValueType", builtins.int)
     V: typing_extensions.TypeAlias = ValueType
 
-class _RemoteBuildingFailureReasonCodeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_RemoteBuildingFailureReasonCode.ValueType], builtins.type):  # noqa: F821
+class _RemoteBuildingFailureReasonCodeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_RemoteBuildingFailureReasonCode.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     INVALID_CODE: _RemoteBuildingFailureReasonCode.ValueType  # 0
     """Invalid reason"""
@@ -92,6 +93,7 @@ MEMORY_LIMIT_EXCEEDED: RemoteBuildingFailureReasonCode.ValueType  # 4
 """Out of memory"""
 global___RemoteBuildingFailureReasonCode = RemoteBuildingFailureReasonCode
 
+@typing.final
 class RemoteBuildingStatus(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -103,6 +105,7 @@ class RemoteBuildingStatus(google.protobuf.message.Message):
     @property
     def failure_details(self) -> global___RemoteBuildingFailureReasonDetails:
         """Failure details"""
+
     @property
     def pod_status(self) -> global___PodStatus: ...
     def __init__(
@@ -112,11 +115,12 @@ class RemoteBuildingStatus(google.protobuf.message.Message):
         failure_details: global___RemoteBuildingFailureReasonDetails | None = ...,
         pod_status: global___PodStatus | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["failure_details", b"failure_details", "pod_status", b"pod_status"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["failure_details", b"failure_details", "pod_status", b"pod_status", "remote_building_status_code", b"remote_building_status_code"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["failure_details", b"failure_details", "pod_status", b"pod_status"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["failure_details", b"failure_details", "pod_status", b"pod_status", "remote_building_status_code", b"remote_building_status_code"]) -> None: ...
 
 global___RemoteBuildingStatus = RemoteBuildingStatus
 
+@typing.final
 class RemoteBuildingFailureReasonDetails(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -132,10 +136,11 @@ class RemoteBuildingFailureReasonDetails(google.protobuf.message.Message):
         failure_code: global___RemoteBuildingFailureReasonCode.ValueType = ...,
         technical_details: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["failure_code", b"failure_code", "technical_details", b"technical_details"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["failure_code", b"failure_code", "technical_details", b"technical_details"]) -> None: ...
 
 global___RemoteBuildingFailureReasonDetails = RemoteBuildingFailureReasonDetails
 
+@typing.final
 class PodStatus(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -145,13 +150,13 @@ class PodStatus(google.protobuf.message.Message):
     REASON_FIELD_NUMBER: builtins.int
     START_TIME_FIELD_NUMBER: builtins.int
     POD_ID_FIELD_NUMBER: builtins.int
-    @property
-    def container_statuses(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ContainerStatus]: ...
     message: builtins.str
     phase: builtins.str
     reason: builtins.str
     start_time: builtins.str
     pod_id: builtins.str
+    @property
+    def container_statuses(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ContainerStatus]: ...
     def __init__(
         self,
         *,
@@ -162,10 +167,11 @@ class PodStatus(google.protobuf.message.Message):
         start_time: builtins.str = ...,
         pod_id: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["container_statuses", b"container_statuses", "message", b"message", "phase", b"phase", "pod_id", b"pod_id", "reason", b"reason", "start_time", b"start_time"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["container_statuses", b"container_statuses", "message", b"message", "phase", b"phase", "pod_id", b"pod_id", "reason", b"reason", "start_time", b"start_time"]) -> None: ...
 
 global___PodStatus = PodStatus
 
+@typing.final
 class ContainerStatus(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -189,11 +195,12 @@ class ContainerStatus(google.protobuf.message.Message):
         started: builtins.bool = ...,
         container_state: global___ContainerState | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["container_state", b"container_state"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["container_id", b"container_id", "container_state", b"container_state", "image", b"image", "restart_count", b"restart_count", "started", b"started"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["container_state", b"container_state"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["container_id", b"container_id", "container_state", b"container_state", "image", b"image", "restart_count", b"restart_count", "started", b"started"]) -> None: ...
 
 global___ContainerStatus = ContainerStatus
 
+@typing.final
 class ContainerState(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -213,12 +220,13 @@ class ContainerState(google.protobuf.message.Message):
         container_state_terminated: global___ContainerStateTerminated | None = ...,
         container_state_waiting: global___ContainerStateWaiting | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["container_state_running", b"container_state_running", "container_state_terminated", b"container_state_terminated", "container_state_waiting", b"container_state_waiting", "state", b"state"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["container_state_running", b"container_state_running", "container_state_terminated", b"container_state_terminated", "container_state_waiting", b"container_state_waiting", "state", b"state"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["state", b"state"]) -> typing_extensions.Literal["container_state_running", "container_state_terminated", "container_state_waiting"] | None: ...
+    def HasField(self, field_name: typing.Literal["container_state_running", b"container_state_running", "container_state_terminated", b"container_state_terminated", "container_state_waiting", b"container_state_waiting", "state", b"state"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["container_state_running", b"container_state_running", "container_state_terminated", b"container_state_terminated", "container_state_waiting", b"container_state_waiting", "state", b"state"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["state", b"state"]) -> typing.Literal["container_state_running", "container_state_terminated", "container_state_waiting"] | None: ...
 
 global___ContainerState = ContainerState
 
+@typing.final
 class ContainerStateRunning(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -229,10 +237,11 @@ class ContainerStateRunning(google.protobuf.message.Message):
         *,
         started_at: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["started_at", b"started_at"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["started_at", b"started_at"]) -> None: ...
 
 global___ContainerStateRunning = ContainerStateRunning
 
+@typing.final
 class ContainerStateTerminated(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -261,10 +270,11 @@ class ContainerStateTerminated(google.protobuf.message.Message):
         signal: builtins.int = ...,
         started_at: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["container_id", b"container_id", "exit_code", b"exit_code", "finished_at", b"finished_at", "message", b"message", "reason", b"reason", "signal", b"signal", "started_at", b"started_at"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["container_id", b"container_id", "exit_code", b"exit_code", "finished_at", b"finished_at", "message", b"message", "reason", b"reason", "signal", b"signal", "started_at", b"started_at"]) -> None: ...
 
 global___ContainerStateTerminated = ContainerStateTerminated
 
+@typing.final
 class ContainerStateWaiting(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -278,6 +288,6 @@ class ContainerStateWaiting(google.protobuf.message.Message):
         message: builtins.str = ...,
         reason: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["message", b"message", "reason", b"reason"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["message", b"message", "reason", b"reason"]) -> None: ...
 
 global___ContainerStateWaiting = ContainerStateWaiting

frogml/_proto/qwak/inference/feedback/feedback_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.qwak.inference/feedback/feedback.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.inference.feedback.feedback_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n\030com.qwak.ai.feedback.api'
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n\030com.qwak.ai.feedback.api'
   _globals['_ACTUALVALUESREQUEST']._serialized_start=102
   _globals['_ACTUALVALUESREQUEST']._serialized_end=290
   _globals['_ENRICHEDACTUALVALUESDATA']._serialized_start=293