frogml-core 0.0.115__py3-none-any.whl → 0.0.117__py3-none-any.whl

_proto/qwak/features_operator/v3/features_operator_async_service_pb2_grpc.py

@@ -0,0 +1,165 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from frogml._proto.qwak.features_operator.v3 import features_operator_async_service_pb2 as qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2
+
+
+class FeaturesOperatorAsyncServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.ValidateDataSource = channel.unary_unary(
+                '/qwak.features_operator.v3.FeaturesOperatorAsyncService/ValidateDataSource',
+                request_serializer=qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.ValidateDataSourceRequest.SerializeToString,
+                response_deserializer=qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.ValidationResponse.FromString,
+                )
+        self.ValidateFeatureSet = channel.unary_unary(
+                '/qwak.features_operator.v3.FeaturesOperatorAsyncService/ValidateFeatureSet',
+                request_serializer=qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.ValidateFeatureSetRequest.SerializeToString,
+                response_deserializer=qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.ValidationResponse.FromString,
+                )
+        self.GetValidationResult = channel.unary_unary(
+                '/qwak.features_operator.v3.FeaturesOperatorAsyncService/GetValidationResult',
+                request_serializer=qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.GetValidationResultRequest.SerializeToString,
+                response_deserializer=qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.GetValidationResultResponse.FromString,
+                )
+        self.CancelValidation = channel.unary_unary(
+                '/qwak.features_operator.v3.FeaturesOperatorAsyncService/CancelValidation',
+                request_serializer=qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.CancelValidationRequest.SerializeToString,
+                response_deserializer=qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.CancelValidationResponse.FromString,
+                )
+
+
+class FeaturesOperatorAsyncServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def ValidateDataSource(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def ValidateFeatureSet(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetValidationResult(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def CancelValidation(self, request, context):
+        """Missing associated documentation comment in .proto file."""
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_FeaturesOperatorAsyncServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'ValidateDataSource': grpc.unary_unary_rpc_method_handler(
+                    servicer.ValidateDataSource,
+                    request_deserializer=qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.ValidateDataSourceRequest.FromString,
+                    response_serializer=qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.ValidationResponse.SerializeToString,
+            ),
+            'ValidateFeatureSet': grpc.unary_unary_rpc_method_handler(
+                    servicer.ValidateFeatureSet,
+                    request_deserializer=qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.ValidateFeatureSetRequest.FromString,
+                    response_serializer=qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.ValidationResponse.SerializeToString,
+            ),
+            'GetValidationResult': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetValidationResult,
+                    request_deserializer=qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.GetValidationResultRequest.FromString,
+                    response_serializer=qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.GetValidationResultResponse.SerializeToString,
+            ),
+            'CancelValidation': grpc.unary_unary_rpc_method_handler(
+                    servicer.CancelValidation,
+                    request_deserializer=qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.CancelValidationRequest.FromString,
+                    response_serializer=qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.CancelValidationResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'qwak.features_operator.v3.FeaturesOperatorAsyncService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class FeaturesOperatorAsyncService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def ValidateDataSource(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.features_operator.v3.FeaturesOperatorAsyncService/ValidateDataSource',
+            qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.ValidateDataSourceRequest.SerializeToString,
+            qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.ValidationResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def ValidateFeatureSet(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.features_operator.v3.FeaturesOperatorAsyncService/ValidateFeatureSet',
+            qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.ValidateFeatureSetRequest.SerializeToString,
+            qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.ValidationResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetValidationResult(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.features_operator.v3.FeaturesOperatorAsyncService/GetValidationResult',
+            qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.GetValidationResultRequest.SerializeToString,
+            qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.GetValidationResultResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def CancelValidation(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.features_operator.v3.FeaturesOperatorAsyncService/CancelValidation',
+            qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.CancelValidationRequest.SerializeToString,
+            qwak_dot_features__operator_dot_v3_dot_features__operator__async__service__pb2.CancelValidationResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

_proto/qwak/features_operator/v3/features_operator_pb2.py

@@ -0,0 +1,38 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.features_operator/v3/features_operator.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n1qwak/features_operator/v3/features_operator.proto\x12\x19qwak.features_operator.v3\x1a\x1fgoogle/protobuf/timestamp.proto\"\x8c\x02\n\x19ValidationSuccessResponse\x12\x0e\n\x06sample\x18\x01 \x01(\t\x12U\n\x19pandas_column_description\x18\x02 \x03(\x0b\x32\x32.qwak.features_operator.v3.PandasColumnDescription\x12S\n\x18spark_column_description\x18\x03 \x03(\x0b\x32\x31.qwak.features_operator.v3.SparkColumnDescription\x12\x33\n\x07outputs\x18\x04 \x01(\x0b\x32\".qwak.features_operator.v3.Outputs\"C\n\x17PandasColumnDescription\x12\x13\n\x0b\x63olumn_name\x18\x01 \x01(\t\x12\x13\n\x0bpandas_type\x18\x02 \x01(\t\"A\n\x16SparkColumnDescription\x12\x13\n\x0b\x63olumn_name\x18\x01 \x01(\t\x12\x12\n\nspark_type\x18\x02 \x01(\t\"\x8b\x01\n\x19ValidationFailureResponse\x12\r\n\x05phase\x18\x01 \x01(\t\x12\x15\n\rerror_message\x18\x02 \x01(\t\x12\x13\n\x0b\x64\x65scription\x18\x03 \x01(\t\x12\x33\n\x07outputs\x18\x04 \x01(\x0b\x32\".qwak.features_operator.v3.Outputs\")\n\x07Outputs\x12\x0e\n\x06stdout\x18\x01 \x01(\t\x12\x0e\n\x06stderr\x18\x02 \x01(\t\"\x1c\n\x1aValidationNotReadyResponseB(\n$com.qwak.ai.features.operator.api.v3P\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.features_operator.v3.features_operator_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n$com.qwak.ai.features.operator.api.v3P\001'
+  _globals['_VALIDATIONSUCCESSRESPONSE']._serialized_start=114
+  _globals['_VALIDATIONSUCCESSRESPONSE']._serialized_end=382
+  _globals['_PANDASCOLUMNDESCRIPTION']._serialized_start=384
+  _globals['_PANDASCOLUMNDESCRIPTION']._serialized_end=451
+  _globals['_SPARKCOLUMNDESCRIPTION']._serialized_start=453
+  _globals['_SPARKCOLUMNDESCRIPTION']._serialized_end=518
+  _globals['_VALIDATIONFAILURERESPONSE']._serialized_start=521
+  _globals['_VALIDATIONFAILURERESPONSE']._serialized_end=660
+  _globals['_OUTPUTS']._serialized_start=662
+  _globals['_OUTPUTS']._serialized_end=703
+  _globals['_VALIDATIONNOTREADYRESPONSE']._serialized_start=705
+  _globals['_VALIDATIONNOTREADYRESPONSE']._serialized_end=733
+# @@protoc_insertion_point(module_scope)

_proto/qwak/file_versioning/file_versioning_pb2.py

@@ -0,0 +1,33 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.file_versioning/file_versioning.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n*qwak/file_versioning/file_versioning.proto\x12\x1fqwak.management.file_versioning\"\xc7\x01\n\x0b\x46ileTagSpec\x12\x10\n\x08\x62uild_id\x18\x01 \x01(\t\x12\x10\n\x08model_id\x18\x02 \x01(\t\x12\x0b\n\x03tag\x18\x03 \x01(\t\x12I\n\textension\x18\x04 \x01(\x0e\x32\x32.qwak.management.file_versioning.FileExtensionTypeB\x02\x18\x01\x12\x16\n\x0e\x65xtension_type\x18\x05 \x01(\t\x12\x16\n\x0e\x65nvironment_id\x18\x06 \x01(\t\x12\x0c\n\x04path\x18\x07 \x01(\t\"G\n\rFileTagFilter\x12\x16\n\x0ctag_contains\x18\x01 \x01(\tH\x00\x12\x14\n\ntag_prefix\x18\x02 \x01(\tH\x00\x42\x08\n\x06\x66ilter*<\n\x11\x46ileExtensionType\x12\x0b\n\x07INVALID\x10\x00\x12\x07\n\x03PNG\x10\x01\x12\x07\n\x03JPG\x10\x02\x12\x08\n\x04JPEG\x10\x03\x42/\n+com.qwak.ai.management.model_versioning.apiP\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.file_versioning.file_versioning_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n+com.qwak.ai.management.model_versioning.apiP\001'
+  _FILETAGSPEC.fields_by_name['extension']._options = None
+  _FILETAGSPEC.fields_by_name['extension']._serialized_options = b'\030\001'
+  _globals['_FILEEXTENSIONTYPE']._serialized_start=354
+  _globals['_FILEEXTENSIONTYPE']._serialized_end=414
+  _globals['_FILETAGSPEC']._serialized_start=80
+  _globals['_FILETAGSPEC']._serialized_end=279
+  _globals['_FILETAGFILTER']._serialized_start=281
+  _globals['_FILETAGFILTER']._serialized_end=352
+# @@protoc_insertion_point(module_scope)

_proto/qwak/file_versioning/file_versioning_service_pb2.py

@@ -0,0 +1,36 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: frogml._proto.qwak.file_versioning/file_versioning_service.proto
+"""Generated protocol buffer code."""
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from frogml._proto.qwak.file_versioning import file_versioning_pb2 as qwak_dot_file__versioning_dot_file__versioning__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n2qwak/file_versioning/file_versioning_service.proto\x12\x1fqwak.management.file_versioning\x1a*qwak/file_versioning/file_versioning.proto\"]\n\x16RegisterFileTagRequest\x12\x43\n\rfile_tag_spec\x18\x01 \x01(\x0b\x32,.qwak.management.file_versioning.FileTagSpec\"\x19\n\x17RegisterFileTagResponse\"}\n\x17GetModelFileTagsRequest\x12\x10\n\x08model_id\x18\x01 \x01(\t\x12\x10\n\x08\x62uild_id\x18\x02 \x01(\t\x12>\n\x06\x66ilter\x18\x03 \x01(\x0b\x32..qwak.management.file_versioning.FileTagFilter\"[\n\x18GetModelFileTagsResponse\x12?\n\tfile_tags\x18\x01 \x03(\x0b\x32,.qwak.management.file_versioning.FileTagSpec2\xb2\x02\n\x1f\x46ileVersioningManagementService\x12\x84\x01\n\x0fRegisterFileTag\x12\x37.qwak.management.file_versioning.RegisterFileTagRequest\x1a\x38.qwak.management.file_versioning.RegisterFileTagResponse\x12\x87\x01\n\x10GetModelFileTags\x12\x38.qwak.management.file_versioning.GetModelFileTagsRequest\x1a\x39.qwak.management.file_versioning.GetModelFileTagsResponseB/\n+com.qwak.ai.management.model_versioning.apiP\x01\x62\x06proto3')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.file_versioning.file_versioning_service_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n+com.qwak.ai.management.model_versioning.apiP\001'
+  _globals['_REGISTERFILETAGREQUEST']._serialized_start=131
+  _globals['_REGISTERFILETAGREQUEST']._serialized_end=224
+  _globals['_REGISTERFILETAGRESPONSE']._serialized_start=226
+  _globals['_REGISTERFILETAGRESPONSE']._serialized_end=251
+  _globals['_GETMODELFILETAGSREQUEST']._serialized_start=253
+  _globals['_GETMODELFILETAGSREQUEST']._serialized_end=378
+  _globals['_GETMODELFILETAGSRESPONSE']._serialized_start=380
+  _globals['_GETMODELFILETAGSRESPONSE']._serialized_end=471
+  _globals['_FILEVERSIONINGMANAGEMENTSERVICE']._serialized_start=474
+  _globals['_FILEVERSIONINGMANAGEMENTSERVICE']._serialized_end=780
+# @@protoc_insertion_point(module_scope)

_proto/qwak/file_versioning/file_versioning_service_pb2.pyi

@@ -0,0 +1,83 @@
+"""
+@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.message
+import frogml._proto.qwak.file_versioning.file_versioning_pb2
+import sys
+
+if sys.version_info >= (3, 8):
+    import typing as typing_extensions
+else:
+    import typing_extensions
+
+DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+
+class RegisterFileTagRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FILE_TAG_SPEC_FIELD_NUMBER: builtins.int
+    @property
+    def file_tag_spec(self) -> frogml._proto.qwak.file_versioning.file_versioning_pb2.FileTagSpec: ...
+    def __init__(
+        self,
+        *,
+        file_tag_spec: frogml._proto.qwak.file_versioning.file_versioning_pb2.FileTagSpec | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["file_tag_spec", b"file_tag_spec"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["file_tag_spec", b"file_tag_spec"]) -> None: ...
+
+global___RegisterFileTagRequest = RegisterFileTagRequest
+
+class RegisterFileTagResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    def __init__(
+        self,
+    ) -> None: ...
+
+global___RegisterFileTagResponse = RegisterFileTagResponse
+
+class GetModelFileTagsRequest(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    MODEL_ID_FIELD_NUMBER: builtins.int
+    BUILD_ID_FIELD_NUMBER: builtins.int
+    FILTER_FIELD_NUMBER: builtins.int
+    model_id: builtins.str
+    """The model id to get file tags for"""
+    build_id: builtins.str
+    """The build id to get file tags for"""
+    @property
+    def filter(self) -> frogml._proto.qwak.file_versioning.file_versioning_pb2.FileTagFilter:
+        """The filter to apply to the file tags"""
+    def __init__(
+        self,
+        *,
+        model_id: builtins.str = ...,
+        build_id: builtins.str = ...,
+        filter: frogml._proto.qwak.file_versioning.file_versioning_pb2.FileTagFilter | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["filter", b"filter"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["build_id", b"build_id", "filter", b"filter", "model_id", b"model_id"]) -> None: ...
+
+global___GetModelFileTagsRequest = GetModelFileTagsRequest
+
+class GetModelFileTagsResponse(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    FILE_TAGS_FIELD_NUMBER: builtins.int
+    @property
+    def file_tags(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[qwak.file_versioning.file_versioning_pb2.FileTagSpec]: ...
+    def __init__(
+        self,
+        *,
+        file_tags: collections.abc.Iterable[qwak.file_versioning.file_versioning_pb2.FileTagSpec] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["file_tags", b"file_tags"]) -> None: ...
+
+global___GetModelFileTagsResponse = GetModelFileTagsResponse

_proto/qwak/file_versioning/file_versioning_service_pb2_grpc.py

@@ -0,0 +1,101 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+import grpc
+
+from frogml._proto.qwak.file_versioning import file_versioning_service_pb2 as qwak_dot_file__versioning_dot_file__versioning__service__pb2
+
+
+class FileVersioningManagementServiceStub(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def __init__(self, channel):
+        """Constructor.
+
+        Args:
+            channel: A grpc.Channel.
+        """
+        self.RegisterFileTag = channel.unary_unary(
+                '/qwak.management.file_versioning.FileVersioningManagementService/RegisterFileTag',
+                request_serializer=qwak_dot_file__versioning_dot_file__versioning__service__pb2.RegisterFileTagRequest.SerializeToString,
+                response_deserializer=qwak_dot_file__versioning_dot_file__versioning__service__pb2.RegisterFileTagResponse.FromString,
+                )
+        self.GetModelFileTags = channel.unary_unary(
+                '/qwak.management.file_versioning.FileVersioningManagementService/GetModelFileTags',
+                request_serializer=qwak_dot_file__versioning_dot_file__versioning__service__pb2.GetModelFileTagsRequest.SerializeToString,
+                response_deserializer=qwak_dot_file__versioning_dot_file__versioning__service__pb2.GetModelFileTagsResponse.FromString,
+                )
+
+
+class FileVersioningManagementServiceServicer(object):
+    """Missing associated documentation comment in .proto file."""
+
+    def RegisterFileTag(self, request, context):
+        """Register data tag
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+    def GetModelFileTags(self, request, context):
+        """Get the models tags
+        """
+        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
+        context.set_details('Method not implemented!')
+        raise NotImplementedError('Method not implemented!')
+
+
+def add_FileVersioningManagementServiceServicer_to_server(servicer, server):
+    rpc_method_handlers = {
+            'RegisterFileTag': grpc.unary_unary_rpc_method_handler(
+                    servicer.RegisterFileTag,
+                    request_deserializer=qwak_dot_file__versioning_dot_file__versioning__service__pb2.RegisterFileTagRequest.FromString,
+                    response_serializer=qwak_dot_file__versioning_dot_file__versioning__service__pb2.RegisterFileTagResponse.SerializeToString,
+            ),
+            'GetModelFileTags': grpc.unary_unary_rpc_method_handler(
+                    servicer.GetModelFileTags,
+                    request_deserializer=qwak_dot_file__versioning_dot_file__versioning__service__pb2.GetModelFileTagsRequest.FromString,
+                    response_serializer=qwak_dot_file__versioning_dot_file__versioning__service__pb2.GetModelFileTagsResponse.SerializeToString,
+            ),
+    }
+    generic_handler = grpc.method_handlers_generic_handler(
+            'qwak.management.file_versioning.FileVersioningManagementService', rpc_method_handlers)
+    server.add_generic_rpc_handlers((generic_handler,))
+
+
+ # This class is part of an EXPERIMENTAL API.
+class FileVersioningManagementService(object):
+    """Missing associated documentation comment in .proto file."""
+
+    @staticmethod
+    def RegisterFileTag(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.management.file_versioning.FileVersioningManagementService/RegisterFileTag',
+            qwak_dot_file__versioning_dot_file__versioning__service__pb2.RegisterFileTagRequest.SerializeToString,
+            qwak_dot_file__versioning_dot_file__versioning__service__pb2.RegisterFileTagResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+
+    @staticmethod
+    def GetModelFileTags(request,
+            target,
+            options=(),
+            channel_credentials=None,
+            call_credentials=None,
+            insecure=False,
+            compression=None,
+            wait_for_ready=None,
+            timeout=None,
+            metadata=None):
+        return grpc.experimental.unary_unary(request, target, '/qwak.management.file_versioning.FileVersioningManagementService/GetModelFileTags',
+            qwak_dot_file__versioning_dot_file__versioning__service__pb2.GetModelFileTagsRequest.SerializeToString,
+            qwak_dot_file__versioning_dot_file__versioning__service__pb2.GetModelFileTagsResponse.FromString,
+            options, channel_credentials,
+            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

_proto/qwak/fitness_service/constructs_pb2.py

@@ -0,0 +1,43 @@
+# -*- 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 import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n%qwak/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')
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'qwak.fitness_service.constructs_pb2', _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+
+  DESCRIPTOR._options = None
+  DESCRIPTOR._serialized_options = b'\n\037com.qwak.ai.fitness.service.apiP\001'
+  _globals['_MEMORYUNIT']._serialized_start=1043
+  _globals['_MEMORYUNIT']._serialized_end=1098
+  _globals['_GPUTYPE']._serialized_start=1101
+  _globals['_GPUTYPE']._serialized_end=1369
+  _globals['_BUILDPROPERTIES']._serialized_start=64
+  _globals['_BUILDPROPERTIES']._serialized_end=226
+  _globals['_MODELURISPEC']._serialized_start=229
+  _globals['_MODELURISPEC']._serialized_end=370
+  _globals['_BUILDENV']._serialized_start=372
+  _globals['_BUILDENV']._serialized_end=488
+  _globals['_DOCKERENV']._serialized_start=491
+  _globals['_DOCKERENV']._serialized_end=629
+  _globals['_PYTHONENV']._serialized_start=631
+  _globals['_PYTHONENV']._serialized_end=755
+  _globals['_CPURESOURCES']._serialized_start=757
+  _globals['_CPURESOURCES']._serialized_end=863
+  _globals['_GPURESOURCES']._serialized_start=866
+  _globals['_GPURESOURCES']._serialized_end=1041
+# @@protoc_insertion_point(module_scope)

_proto/qwak/fitness_service/fitness_pb2.py

@@ -0,0 +1,32 @@
+# -*- 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 import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from frogml._proto.qwak.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')
+
+_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['_PURCHASEOPTION']._serialized_start=793
+  _globals['_PURCHASEOPTION']._serialized_end=894
+  _globals['_BUILDSPEC']._serialized_start=100
+  _globals['_BUILDSPEC']._serialized_end=686
+  _globals['_BUILDPROPERTIESV1']._serialized_start=688
+  _globals['_BUILDPROPERTIESV1']._serialized_end=791
+# @@protoc_insertion_point(module_scope)