frogml-core 0.0.1__py3-none-any.whl

_frogml_proto/qwak/fitness_service/constructs_pb2.py

@@ -0,0 +1,120 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: _frogml_proto.qwak.fitness_service/constructs.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n%qwak/fitness_service/constructs.proto\x12\x14qwak.fitness.service\"\xa2\x01\n\x0f\x42uildProperties\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\x12\x0e\n\x06\x62ranch\x18\x02 \x01(\t\x12\x10\n\x08model_id\x18\x03 \x01(\t\x12\x0c\n\x04tags\x18\x04 \x03(\t\x12\x35\n\tmodel_uri\x18\x05 \x01(\x0b\x32\".qwak.fitness.service.ModelUriSpec\x12\x16\n\x0egpu_compatible\x18\x06 \x01(\x08\"\x8d\x01\n\x0cModelUriSpec\x12\x0b\n\x03uri\x18\x01 \x01(\t\x12\x12\n\ngit_branch\x18\x02 \x01(\t\x12\x17\n\x0fgit_credentials\x18\x03 \x01(\t\x12\x1e\n\x16git_credentials_secret\x18\x04 \x01(\t\x12\x11\n\tcommit_id\x18\x05 \x01(\t\x12\x10\n\x08main_dir\x18\x06 \x01(\t\"t\n\x08\x42uildEnv\x12\x33\n\ndocker_env\x18\x01 \x01(\x0b\x32\x1f.qwak.fitness.service.DockerEnv\x12\x33\n\npython_env\x18\x02 \x01(\x0b\x32\x1f.qwak.fitness.service.PythonEnv\"\x8a\x01\n\tDockerEnv\x12\x12\n\nbase_image\x18\x01 \x01(\t\x12\x10\n\x08\x65nv_vars\x18\x02 \x03(\t\x12\x10\n\x08no_cache\x18\x03 \x01(\x08\x12\x1c\n\x14\x61ssumed_iam_role_arn\x18\x04 \x01(\t\x12\'\n\x1fservice_account_key_secret_name\x18\x05 \x01(\t\"|\n\tPythonEnv\x12\x17\n\x0fgit_credentials\x18\x01 \x01(\t\x12\x1e\n\x16git_credentials_secret\x18\x02 \x01(\t\x12\x18\n\x10qwak_sdk_version\x18\x03 \x01(\t\x12\x1c\n\x14qwak_sdk_extra_index\x18\x04 \x01(\t\"j\n\x0c\x43puResources\x12\x0b\n\x03\x63pu\x18\x01 \x01(\x02\x12\x15\n\rmemory_amount\x18\x02 \x01(\x05\x12\x36\n\x0cmemory_units\x18\x03 \x01(\x0e\x32 .qwak.fitness.service.MemoryUnit\"\xaf\x01\n\x0cGpuResources\x12/\n\x08gpu_type\x18\x01 \x01(\x0e\x32\x1d.qwak.fitness.service.GpuType\x12\x12\n\ngpu_amount\x18\x02 \x01(\x05\x12\x0b\n\x03\x63pu\x18\x03 \x01(\x02\x12\x15\n\rmemory_amount\x18\x04 \x01(\x05\x12\x36\n\x0cmemory_units\x18\x05 \x01(\x0e\x32 .qwak.fitness.service.MemoryUnit*7\n\nMemoryUnit\x12\x17\n\x13UNKNOWN_MEMORY_UNIT\x10\x00\x12\x07\n\x03MIB\x10\x01\x12\x07\n\x03GIB\x10\x02*\x8c\x02\n\x07GpuType\x12\x0b\n\x07INVALID\x10\x00\x12\x0e\n\nNVIDIA_K80\x10\x01\x12\x0f\n\x0bNVIDIA_V100\x10\x02\x12\x0f\n\x0bNVIDIA_A100\x10\x03\x12\r\n\tNVIDIA_T4\x10\x04\x12\x0f\n\x0bNVIDIA_A10G\x10\x05\x12\r\n\tNVIDIA_L4\x10\x06\x12\x14\n\x10NVIDIA_T4_1_4_15\x10\x07\x12\x14\n\x10NVIDIA_T4_1_8_30\x10\x08\x12\x15\n\x11NVIDIA_T4_1_16_60\x10\t\x12\x1e\n\x1aNVIDIA_A100_80GB_8_96_1360\x10\n\x12\x16\n\x12NVIDIA_V100_1_8_52\x10\x0b\x12\x18\n\x14NVIDIA_V100_4_32_208\x10\x0c\x42#\n\x1f\x63om.qwak.ai.fitness.service.apiP\x01\x62\x06proto3')
+
+_MEMORYUNIT = DESCRIPTOR.enum_types_by_name['MemoryUnit']
+MemoryUnit = enum_type_wrapper.EnumTypeWrapper(_MEMORYUNIT)
+_GPUTYPE = DESCRIPTOR.enum_types_by_name['GpuType']
+GpuType = enum_type_wrapper.EnumTypeWrapper(_GPUTYPE)
+UNKNOWN_MEMORY_UNIT = 0
+MIB = 1
+GIB = 2
+INVALID = 0
+NVIDIA_K80 = 1
+NVIDIA_V100 = 2
+NVIDIA_A100 = 3
+NVIDIA_T4 = 4
+NVIDIA_A10G = 5
+NVIDIA_L4 = 6
+NVIDIA_T4_1_4_15 = 7
+NVIDIA_T4_1_8_30 = 8
+NVIDIA_T4_1_16_60 = 9
+NVIDIA_A100_80GB_8_96_1360 = 10
+NVIDIA_V100_1_8_52 = 11
+NVIDIA_V100_4_32_208 = 12
+
+
+_BUILDPROPERTIES = DESCRIPTOR.message_types_by_name['BuildProperties']
+_MODELURISPEC = DESCRIPTOR.message_types_by_name['ModelUriSpec']
+_BUILDENV = DESCRIPTOR.message_types_by_name['BuildEnv']
+_DOCKERENV = DESCRIPTOR.message_types_by_name['DockerEnv']
+_PYTHONENV = DESCRIPTOR.message_types_by_name['PythonEnv']
+_CPURESOURCES = DESCRIPTOR.message_types_by_name['CpuResources']
+_GPURESOURCES = DESCRIPTOR.message_types_by_name['GpuResources']
+BuildProperties = _reflection.GeneratedProtocolMessageType('BuildProperties', (_message.Message,), {
+  'DESCRIPTOR' : _BUILDPROPERTIES,
+  '__module__' : 'qwak.fitness_service.constructs_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.fitness.service.BuildProperties)
+  })
+_sym_db.RegisterMessage(BuildProperties)
+
+ModelUriSpec = _reflection.GeneratedProtocolMessageType('ModelUriSpec', (_message.Message,), {
+  'DESCRIPTOR' : _MODELURISPEC,
+  '__module__' : 'qwak.fitness_service.constructs_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.fitness.service.ModelUriSpec)
+  })
+_sym_db.RegisterMessage(ModelUriSpec)
+
+BuildEnv = _reflection.GeneratedProtocolMessageType('BuildEnv', (_message.Message,), {
+  'DESCRIPTOR' : _BUILDENV,
+  '__module__' : 'qwak.fitness_service.constructs_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.fitness.service.BuildEnv)
+  })
+_sym_db.RegisterMessage(BuildEnv)
+
+DockerEnv = _reflection.GeneratedProtocolMessageType('DockerEnv', (_message.Message,), {
+  'DESCRIPTOR' : _DOCKERENV,
+  '__module__' : 'qwak.fitness_service.constructs_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.fitness.service.DockerEnv)
+  })
+_sym_db.RegisterMessage(DockerEnv)
+
+PythonEnv = _reflection.GeneratedProtocolMessageType('PythonEnv', (_message.Message,), {
+  'DESCRIPTOR' : _PYTHONENV,
+  '__module__' : 'qwak.fitness_service.constructs_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.fitness.service.PythonEnv)
+  })
+_sym_db.RegisterMessage(PythonEnv)
+
+CpuResources = _reflection.GeneratedProtocolMessageType('CpuResources', (_message.Message,), {
+  'DESCRIPTOR' : _CPURESOURCES,
+  '__module__' : 'qwak.fitness_service.constructs_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.fitness.service.CpuResources)
+  })
+_sym_db.RegisterMessage(CpuResources)
+
+GpuResources = _reflection.GeneratedProtocolMessageType('GpuResources', (_message.Message,), {
+  'DESCRIPTOR' : _GPURESOURCES,
+  '__module__' : 'qwak.fitness_service.constructs_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.fitness.service.GpuResources)
+  })
+_sym_db.RegisterMessage(GpuResources)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\037com.qwak.ai.fitness.service.apiP\001'
+  _MEMORYUNIT._serialized_start=1043
+  _MEMORYUNIT._serialized_end=1098
+  _GPUTYPE._serialized_start=1101
+  _GPUTYPE._serialized_end=1369
+  _BUILDPROPERTIES._serialized_start=64
+  _BUILDPROPERTIES._serialized_end=226
+  _MODELURISPEC._serialized_start=229
+  _MODELURISPEC._serialized_end=370
+  _BUILDENV._serialized_start=372
+  _BUILDENV._serialized_end=488
+  _DOCKERENV._serialized_start=491
+  _DOCKERENV._serialized_end=629
+  _PYTHONENV._serialized_start=631
+  _PYTHONENV._serialized_end=755
+  _CPURESOURCES._serialized_start=757
+  _CPURESOURCES._serialized_end=863
+  _GPURESOURCES._serialized_start=866
+  _GPURESOURCES._serialized_end=1041
+# @@protoc_insertion_point(module_scope)

_frogml_proto/qwak/fitness_service/constructs_pb2.pyi

@@ -0,0 +1,297 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import collections.abc
+import google.protobuf.descriptor
+import google.protobuf.internal.containers
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+import sys
+import typing
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class _MemoryUnit:
+    ValueType = typing.NewType("ValueType", builtins.int)
+    V: typing_extensions.TypeAlias = ValueType
+
+class _MemoryUnitEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_MemoryUnit.ValueType], builtins.type):  # noqa: F821
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    UNKNOWN_MEMORY_UNIT: _MemoryUnit.ValueType  # 0
+    MIB: _MemoryUnit.ValueType  # 1
+    GIB: _MemoryUnit.ValueType  # 2
+
+class MemoryUnit(_MemoryUnit, metaclass=_MemoryUnitEnumTypeWrapper): ...
+
+UNKNOWN_MEMORY_UNIT: MemoryUnit.ValueType  # 0
+MIB: MemoryUnit.ValueType  # 1
+GIB: MemoryUnit.ValueType  # 2
+global___MemoryUnit = MemoryUnit
+
+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
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    INVALID: _GpuType.ValueType  # 0
+    NVIDIA_K80: _GpuType.ValueType  # 1
+    NVIDIA_V100: _GpuType.ValueType  # 2
+    NVIDIA_A100: _GpuType.ValueType  # 3
+    NVIDIA_T4: _GpuType.ValueType  # 4
+    NVIDIA_A10G: _GpuType.ValueType  # 5
+    NVIDIA_L4: _GpuType.ValueType  # 6
+    NVIDIA_T4_1_4_15: _GpuType.ValueType  # 7
+    """More specific node types
+    t4.xl
+    """
+    NVIDIA_T4_1_8_30: _GpuType.ValueType  # 8
+    """t4.2xl"""
+    NVIDIA_T4_1_16_60: _GpuType.ValueType  # 9
+    """t4.4xl"""
+    NVIDIA_A100_80GB_8_96_1360: _GpuType.ValueType  # 10
+    """a100.8xl"""
+    NVIDIA_V100_1_8_52: _GpuType.ValueType  # 11
+    """a100.xl"""
+    NVIDIA_V100_4_32_208: _GpuType.ValueType  # 12
+    """v100.4xl"""
+
+class GpuType(_GpuType, metaclass=_GpuTypeEnumTypeWrapper): ...
+
+INVALID: GpuType.ValueType  # 0
+NVIDIA_K80: GpuType.ValueType  # 1
+NVIDIA_V100: GpuType.ValueType  # 2
+NVIDIA_A100: GpuType.ValueType  # 3
+NVIDIA_T4: GpuType.ValueType  # 4
+NVIDIA_A10G: GpuType.ValueType  # 5
+NVIDIA_L4: GpuType.ValueType  # 6
+NVIDIA_T4_1_4_15: GpuType.ValueType  # 7
+"""More specific node types
+t4.xl
+"""
+NVIDIA_T4_1_8_30: GpuType.ValueType  # 8
+"""t4.2xl"""
+NVIDIA_T4_1_16_60: GpuType.ValueType  # 9
+"""t4.4xl"""
+NVIDIA_A100_80GB_8_96_1360: GpuType.ValueType  # 10
+"""a100.8xl"""
+NVIDIA_V100_1_8_52: GpuType.ValueType  # 11
+"""a100.xl"""
+NVIDIA_V100_4_32_208: GpuType.ValueType  # 12
+"""v100.4xl"""
+global___GpuType = GpuType
+
+class BuildProperties(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    BUILD_ID_FIELD_NUMBER: builtins.int
+    BRANCH_FIELD_NUMBER: builtins.int
+    MODEL_ID_FIELD_NUMBER: builtins.int
+    TAGS_FIELD_NUMBER: builtins.int
+    MODEL_URI_FIELD_NUMBER: builtins.int
+    GPU_COMPATIBLE_FIELD_NUMBER: builtins.int
+    build_id: builtins.str
+    """Build id"""
+    branch: builtins.str
+    """Branch name. When none provided - 'main' will be the selected branch"""
+    model_id: builtins.str
+    """Model id"""
+    @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,
+        *,
+        build_id: builtins.str = ...,
+        branch: builtins.str = ...,
+        model_id: builtins.str = ...,
+        tags: collections.abc.Iterable[builtins.str] | None = ...,
+        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: ...
+
+global___BuildProperties = BuildProperties
+
+class ModelUriSpec(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    URI_FIELD_NUMBER: builtins.int
+    GIT_BRANCH_FIELD_NUMBER: builtins.int
+    GIT_CREDENTIALS_FIELD_NUMBER: builtins.int
+    GIT_CREDENTIALS_SECRET_FIELD_NUMBER: builtins.int
+    COMMIT_ID_FIELD_NUMBER: builtins.int
+    MAIN_DIR_FIELD_NUMBER: builtins.int
+    uri: builtins.str
+    """Model code to build upon, can be a download link or a git repo"""
+    git_branch: builtins.str
+    """Model code branch to clone"""
+    git_credentials: builtins.str
+    """Git credentials"""
+    git_credentials_secret: builtins.str
+    """Secret to fetch git credentials, overrides git_credentials if set"""
+    commit_id: builtins.str
+    """Commit id - filled when using local model code which is part of a git repo"""
+    main_dir: builtins.str
+    """Overrides the models main dir"""
+    def __init__(
+        self,
+        *,
+        uri: builtins.str = ...,
+        git_branch: builtins.str = ...,
+        git_credentials: builtins.str = ...,
+        git_credentials_secret: builtins.str = ...,
+        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: ...
+
+global___ModelUriSpec = ModelUriSpec
+
+class BuildEnv(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    DOCKER_ENV_FIELD_NUMBER: builtins.int
+    PYTHON_ENV_FIELD_NUMBER: builtins.int
+    @property
+    def docker_env(self) -> global___DockerEnv: ...
+    @property
+    def python_env(self) -> global___PythonEnv: ...
+    def __init__(
+        self,
+        *,
+        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: ...
+
+global___BuildEnv = BuildEnv
+
+class DockerEnv(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    BASE_IMAGE_FIELD_NUMBER: builtins.int
+    ENV_VARS_FIELD_NUMBER: builtins.int
+    NO_CACHE_FIELD_NUMBER: builtins.int
+    ASSUMED_IAM_ROLE_ARN_FIELD_NUMBER: builtins.int
+    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
+    """
+    assumed_iam_role_arn: builtins.str
+    """Custom IAM role to be assumed by model"""
+    service_account_key_secret_name: builtins.str
+    """Service account key secret name for gcp"""
+    def __init__(
+        self,
+        *,
+        base_image: builtins.str = ...,
+        env_vars: collections.abc.Iterable[builtins.str] | None = ...,
+        no_cache: builtins.bool = ...,
+        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: ...
+
+global___DockerEnv = DockerEnv
+
+class PythonEnv(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    GIT_CREDENTIALS_FIELD_NUMBER: builtins.int
+    GIT_CREDENTIALS_SECRET_FIELD_NUMBER: builtins.int
+    QWAK_SDK_VERSION_FIELD_NUMBER: builtins.int
+    QWAK_SDK_EXTRA_INDEX_FIELD_NUMBER: builtins.int
+    git_credentials: builtins.str
+    """Git credentials"""
+    git_credentials_secret: builtins.str
+    """Secret to fetch git credentials, overrides git_credentials if set"""
+    qwak_sdk_version: builtins.str
+    """Qwak sdk version of calling user"""
+    qwak_sdk_extra_index: builtins.str
+    """Pip extra index url to install qwak-sdk from"""
+    def __init__(
+        self,
+        *,
+        git_credentials: builtins.str = ...,
+        git_credentials_secret: builtins.str = ...,
+        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: ...
+
+global___PythonEnv = PythonEnv
+
+class CpuResources(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    CPU_FIELD_NUMBER: builtins.int
+    MEMORY_AMOUNT_FIELD_NUMBER: builtins.int
+    MEMORY_UNITS_FIELD_NUMBER: builtins.int
+    cpu: builtins.float
+    """Amount of CPU cores"""
+    memory_amount: builtins.int
+    """Amount of memory"""
+    memory_units: global___MemoryUnit.ValueType
+    """Units type of memory"""
+    def __init__(
+        self,
+        *,
+        cpu: builtins.float = ...,
+        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: ...
+
+global___CpuResources = CpuResources
+
+class GpuResources(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    GPU_TYPE_FIELD_NUMBER: builtins.int
+    GPU_AMOUNT_FIELD_NUMBER: builtins.int
+    CPU_FIELD_NUMBER: builtins.int
+    MEMORY_AMOUNT_FIELD_NUMBER: builtins.int
+    MEMORY_UNITS_FIELD_NUMBER: builtins.int
+    gpu_type: global___GpuType.ValueType
+    """The type of the GPU"""
+    gpu_amount: builtins.int
+    """Amount of GPUs"""
+    cpu: builtins.float
+    """Amount of CPU cores"""
+    memory_amount: builtins.int
+    """Amount of memory"""
+    memory_units: global___MemoryUnit.ValueType
+    """Units type of memory"""
+    def __init__(
+        self,
+        *,
+        gpu_type: global___GpuType.ValueType = ...,
+        gpu_amount: builtins.int = ...,
+        cpu: builtins.float = ...,
+        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: ...
+
+global___GpuResources = GpuResources

_frogml_proto/qwak/fitness_service/constructs_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+

_frogml_proto/qwak/fitness_service/fitness_pb2.py

@@ -0,0 +1,54 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: _frogml_proto.qwak.fitness_service/fitness.proto
+"""Generated protocol buffer code."""
+from google.protobuf.internal import enum_type_wrapper
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import message as _message
+from google.protobuf import reflection as _reflection
+from google.protobuf import symbol_database as _symbol_database
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from _frogml_proto.qwak.fitness_service import constructs_pb2 as qwak_dot_fitness__service_dot_constructs__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\"qwak/fitness_service/fitness.proto\x12\x14qwak.fitness.service\x1a%qwak/fitness_service/constructs.proto\"\xca\x04\n\tBuildSpec\x12?\n\x10\x62uild_properties\x18\x01 \x01(\x0b\x32%.qwak.fitness.service.BuildProperties\x12\x31\n\tbuild_env\x18\x02 \x01(\x0b\x32\x1e.qwak.fitness.service.BuildEnv\x12;\n\rcpu_resources\x18\x03 \x01(\x0b\x32\".qwak.fitness.service.CpuResourcesH\x00\x12;\n\rgpu_resources\x18\x07 \x01(\x0b\x32\".qwak.fitness.service.GpuResourcesH\x00\x12\x0f\n\x07verbose\x18\x04 \x01(\x05\x12\x14\n\x0c\x62uild_config\x18\x05 \x01(\t\x12\x15\n\rbuild_v1_flag\x18\x06 \x01(\x08\x12\x44\n\x13\x62uild_properties_v1\x18\x08 \x01(\x0b\x32\'.qwak.fitness.service.BuildPropertiesV1\x12=\n\x0fpurchase_option\x18\t \x01(\x0e\x32$.qwak.fitness.service.PurchaseOption\x12\x1c\n\x14\x62uild_destined_image\x18\n \x01(\t\x12\"\n\x1aprovision_instance_timeout\x18\x0b \x01(\x05\x12\x1f\n\x17\x61rtifactory_project_key\x18\x0c \x01(\t\x12\x1c\n\x14\x61rtifactory_base_url\x18\r \x01(\tB\x0b\n\tResources\"g\n\x11\x42uildPropertiesV1\x12\x18\n\x10\x62uild_config_url\x18\x01 \x01(\t\x12\x1a\n\x12qwak_sdk_wheel_url\x18\x02 \x01(\t\x12\x1c\n\x14qwak_sdk_version_url\x18\x03 \x01(\t*e\n\x0ePurchaseOption\x12\x1b\n\x17INVALID_PURCHASE_OPTION\x10\x00\x12\x18\n\x14SPOT_PURCHASE_OPTION\x10\x01\x12\x1c\n\x18ONDEMAND_PURCHASE_OPTION\x10\x02\x42#\n\x1f\x63om.qwak.ai.fitness.service.apiP\x01\x62\x06proto3')
+
+_PURCHASEOPTION = DESCRIPTOR.enum_types_by_name['PurchaseOption']
+PurchaseOption = enum_type_wrapper.EnumTypeWrapper(_PURCHASEOPTION)
+INVALID_PURCHASE_OPTION = 0
+SPOT_PURCHASE_OPTION = 1
+ONDEMAND_PURCHASE_OPTION = 2
+
+
+_BUILDSPEC = DESCRIPTOR.message_types_by_name['BuildSpec']
+_BUILDPROPERTIESV1 = DESCRIPTOR.message_types_by_name['BuildPropertiesV1']
+BuildSpec = _reflection.GeneratedProtocolMessageType('BuildSpec', (_message.Message,), {
+  'DESCRIPTOR' : _BUILDSPEC,
+  '__module__' : 'qwak.fitness_service.fitness_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.fitness.service.BuildSpec)
+  })
+_sym_db.RegisterMessage(BuildSpec)
+
+BuildPropertiesV1 = _reflection.GeneratedProtocolMessageType('BuildPropertiesV1', (_message.Message,), {
+  'DESCRIPTOR' : _BUILDPROPERTIESV1,
+  '__module__' : 'qwak.fitness_service.fitness_pb2'
+  # @@protoc_insertion_point(class_scope:qwak.fitness.service.BuildPropertiesV1)
+  })
+_sym_db.RegisterMessage(BuildPropertiesV1)
+
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\037com.qwak.ai.fitness.service.apiP\001'
+  _PURCHASEOPTION._serialized_start=793
+  _PURCHASEOPTION._serialized_end=894
+  _BUILDSPEC._serialized_start=100
+  _BUILDSPEC._serialized_end=686
+  _BUILDPROPERTIESV1._serialized_start=688
+  _BUILDPROPERTIESV1._serialized_end=791
+# @@protoc_insertion_point(module_scope)

_frogml_proto/qwak/fitness_service/fitness_pb2.pyi

@@ -0,0 +1,126 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""
+import builtins
+import google.protobuf.descriptor
+import google.protobuf.internal.enum_type_wrapper
+import google.protobuf.message
+import _frogml_proto.qwak.fitness_service.constructs_pb2
+import sys
+import typing
+
+if sys.version_info >= (3, 10):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class _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
+    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    INVALID_PURCHASE_OPTION: _PurchaseOption.ValueType  # 0
+    SPOT_PURCHASE_OPTION: _PurchaseOption.ValueType  # 1
+    ONDEMAND_PURCHASE_OPTION: _PurchaseOption.ValueType  # 2
+
+class PurchaseOption(_PurchaseOption, metaclass=_PurchaseOptionEnumTypeWrapper): ...
+
+INVALID_PURCHASE_OPTION: PurchaseOption.ValueType  # 0
+SPOT_PURCHASE_OPTION: PurchaseOption.ValueType  # 1
+ONDEMAND_PURCHASE_OPTION: PurchaseOption.ValueType  # 2
+global___PurchaseOption = PurchaseOption
+
+class BuildSpec(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    BUILD_PROPERTIES_FIELD_NUMBER: builtins.int
+    BUILD_ENV_FIELD_NUMBER: builtins.int
+    CPU_RESOURCES_FIELD_NUMBER: builtins.int
+    GPU_RESOURCES_FIELD_NUMBER: builtins.int
+    VERBOSE_FIELD_NUMBER: builtins.int
+    BUILD_CONFIG_FIELD_NUMBER: builtins.int
+    BUILD_V1_FLAG_FIELD_NUMBER: builtins.int
+    BUILD_PROPERTIES_V1_FIELD_NUMBER: builtins.int
+    PURCHASE_OPTION_FIELD_NUMBER: builtins.int
+    BUILD_DESTINED_IMAGE_FIELD_NUMBER: builtins.int
+    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
+    """The image name to save build to"""
+    provision_instance_timeout: builtins.int
+    """The image name to save build to"""
+    artifactory_project_key: builtins.str
+    """artifactory project key"""
+    artifactory_base_url: builtins.str
+    """artifactory base url"""
+    def __init__(
+        self,
+        *,
+        build_properties: _frogml_proto.qwak.fitness_service.constructs_pb2.BuildProperties | None = ...,
+        build_env: _frogml_proto.qwak.fitness_service.constructs_pb2.BuildEnv | None = ...,
+        cpu_resources: _frogml_proto.qwak.fitness_service.constructs_pb2.CpuResources | None = ...,
+        gpu_resources: _frogml_proto.qwak.fitness_service.constructs_pb2.GpuResources | None = ...,
+        verbose: builtins.int = ...,
+        build_config: builtins.str = ...,
+        build_v1_flag: builtins.bool = ...,
+        build_properties_v1: global___BuildPropertiesV1 | None = ...,
+        purchase_option: global___PurchaseOption.ValueType = ...,
+        build_destined_image: builtins.str = ...,
+        provision_instance_timeout: builtins.int = ...,
+        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: ...
+
+global___BuildSpec = BuildSpec
+
+class BuildPropertiesV1(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    BUILD_CONFIG_URL_FIELD_NUMBER: builtins.int
+    QWAK_SDK_WHEEL_URL_FIELD_NUMBER: builtins.int
+    QWAK_SDK_VERSION_URL_FIELD_NUMBER: builtins.int
+    build_config_url: builtins.str
+    """Build V1 - build config url"""
+    qwak_sdk_wheel_url: builtins.str
+    """Build V1 - qwak-sdk wheel url"""
+    qwak_sdk_version_url: builtins.str
+    """Build V1 - qwak-sdk version"""
+    def __init__(
+        self,
+        *,
+        build_config_url: builtins.str = ...,
+        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: ...
+
+global___BuildPropertiesV1 = BuildPropertiesV1

_frogml_proto/qwak/fitness_service/fitness_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+