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

frogml/_proto/qwak/features_operator/v3/features_operator_async_service_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.message
@@ -9,27 +10,23 @@ import google.protobuf.timestamp_pb2
 import frogml._proto.qwak.feature_store.features.feature_set_pb2
 import frogml._proto.qwak.feature_store.sources.data_source_pb2
 import frogml._proto.qwak.features_operator.v3.features_operator_pb2
-import sys
 import typing
 
-if sys.version_info >= (3, 8):
-    import typing as typing_extensions
-else:
-    import typing_extensions
-
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 
+@typing.final
 class ValidateDataSourceRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
     DATA_SOURCE_SPEC_FIELD_NUMBER: builtins.int
     NUM_SAMPLES_FIELD_NUMBER: builtins.int
     VALIDATION_OPTIONS_FIELD_NUMBER: builtins.int
+    num_samples: builtins.int
+    """Number of samples to return from the validation"""
     @property
     def data_source_spec(self) -> frogml._proto.qwak.feature_store.sources.data_source_pb2.DataSourceSpec:
         """Data source spec to validate"""
-    num_samples: builtins.int
-    """Number of samples to return from the validation"""
+
     @property
     def validation_options(self) -> global___DataSourceValidationOptions: ...
     def __init__(
@@ -39,12 +36,13 @@ class ValidateDataSourceRequest(google.protobuf.message.Message):
         num_samples: builtins.int = ...,
         validation_options: global___DataSourceValidationOptions | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["data_source_spec", b"data_source_spec", "optional_validation_options", b"optional_validation_options", "validation_options", b"validation_options"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["data_source_spec", b"data_source_spec", "num_samples", b"num_samples", "optional_validation_options", b"optional_validation_options", "validation_options", b"validation_options"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_validation_options", b"optional_validation_options"]) -> typing_extensions.Literal["validation_options"] | None: ...
+    def HasField(self, field_name: typing.Literal["data_source_spec", b"data_source_spec", "optional_validation_options", b"optional_validation_options", "validation_options", b"validation_options"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["data_source_spec", b"data_source_spec", "num_samples", b"num_samples", "optional_validation_options", b"optional_validation_options", "validation_options", b"validation_options"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["optional_validation_options", b"optional_validation_options"]) -> typing.Literal["validation_options"] | None: ...
 
 global___ValidateDataSourceRequest = ValidateDataSourceRequest
 
+@typing.final
 class ValidateFeatureSetRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -52,13 +50,14 @@ class ValidateFeatureSetRequest(google.protobuf.message.Message):
     ZIP_PATH_FIELD_NUMBER: builtins.int
     NUM_SAMPLES_FIELD_NUMBER: builtins.int
     VALIDATION_OPTIONS_FIELD_NUMBER: builtins.int
-    @property
-    def feature_set_spec(self) -> frogml._proto.qwak.feature_store.features.feature_set_pb2.FeatureSetSpec:
-        """Featureset spec to validate"""
     zip_path: builtins.str
     """Dependencies zip file path"""
     num_samples: builtins.int
     """Number of samples to return from the validation"""
+    @property
+    def feature_set_spec(self) -> frogml._proto.qwak.feature_store.features.feature_set_pb2.FeatureSetSpec:
+        """Featureset spec to validate"""
+
     @property
     def validation_options(self) -> global___FeatureSetValidationOptions: ...
     def __init__(
@@ -69,12 +68,13 @@ class ValidateFeatureSetRequest(google.protobuf.message.Message):
         num_samples: builtins.int = ...,
         validation_options: global___FeatureSetValidationOptions | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["feature_set_spec", b"feature_set_spec", "optional_validation_options", b"optional_validation_options", "validation_options", b"validation_options"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["feature_set_spec", b"feature_set_spec", "num_samples", b"num_samples", "optional_validation_options", b"optional_validation_options", "validation_options", b"validation_options", "zip_path", b"zip_path"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_validation_options", b"optional_validation_options"]) -> typing_extensions.Literal["validation_options"] | None: ...
+    def HasField(self, field_name: typing.Literal["feature_set_spec", b"feature_set_spec", "optional_validation_options", b"optional_validation_options", "validation_options", b"validation_options"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["feature_set_spec", b"feature_set_spec", "num_samples", b"num_samples", "optional_validation_options", b"optional_validation_options", "validation_options", b"validation_options", "zip_path", b"zip_path"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["optional_validation_options", b"optional_validation_options"]) -> typing.Literal["validation_options"] | None: ...
 
 global___ValidateFeatureSetRequest = ValidateFeatureSetRequest
 
+@typing.final
 class ValidationResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -86,10 +86,11 @@ class ValidationResponse(google.protobuf.message.Message):
         *,
         request_id: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["request_id", b"request_id"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["request_id", b"request_id"]) -> None: ...
 
 global___ValidationResponse = ValidationResponse
 
+@typing.final
 class GetValidationResultRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -101,10 +102,11 @@ class GetValidationResultRequest(google.protobuf.message.Message):
         *,
         request_id: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["request_id", b"request_id"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["request_id", b"request_id"]) -> None: ...
 
 global___GetValidationResultRequest = GetValidationResultRequest
 
+@typing.final
 class GetValidationResultResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -124,12 +126,13 @@ class GetValidationResultResponse(google.protobuf.message.Message):
         failure_response: frogml._proto.qwak.features_operator.v3.features_operator_pb2.ValidationFailureResponse | None = ...,
         not_ready_response: frogml._proto.qwak.features_operator.v3.features_operator_pb2.ValidationNotReadyResponse | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["failure_response", b"failure_response", "not_ready_response", b"not_ready_response", "success_response", b"success_response", "type", b"type"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["failure_response", b"failure_response", "not_ready_response", b"not_ready_response", "success_response", b"success_response", "type", b"type"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["type", b"type"]) -> typing_extensions.Literal["success_response", "failure_response", "not_ready_response"] | None: ...
+    def HasField(self, field_name: typing.Literal["failure_response", b"failure_response", "not_ready_response", b"not_ready_response", "success_response", b"success_response", "type", b"type"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["failure_response", b"failure_response", "not_ready_response", b"not_ready_response", "success_response", b"success_response", "type", b"type"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["type", b"type"]) -> typing.Literal["success_response", "failure_response", "not_ready_response"] | None: ...
 
 global___GetValidationResultResponse = GetValidationResultResponse
 
+@typing.final
 class CancelValidationRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -140,10 +143,11 @@ class CancelValidationRequest(google.protobuf.message.Message):
         *,
         request_id: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["request_id", b"request_id"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["request_id", b"request_id"]) -> None: ...
 
 global___CancelValidationRequest = CancelValidationRequest
 
+@typing.final
 class CancelValidationResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -153,6 +157,7 @@ class CancelValidationResponse(google.protobuf.message.Message):
 
 global___CancelValidationResponse = CancelValidationResponse
 
+@typing.final
 class FeatureSetValidationOptions(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -167,15 +172,16 @@ class FeatureSetValidationOptions(google.protobuf.message.Message):
         data_source_limit: builtins.int = ...,
         validation_time_range: global___ValidationTimeRange | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["data_source_limit", b"data_source_limit", "optional_data_source_limit", b"optional_data_source_limit", "optional_validation_time_range", b"optional_validation_time_range", "validation_time_range", b"validation_time_range"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["data_source_limit", b"data_source_limit", "optional_data_source_limit", b"optional_data_source_limit", "optional_validation_time_range", b"optional_validation_time_range", "validation_time_range", b"validation_time_range"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["data_source_limit", b"data_source_limit", "optional_data_source_limit", b"optional_data_source_limit", "optional_validation_time_range", b"optional_validation_time_range", "validation_time_range", b"validation_time_range"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["data_source_limit", b"data_source_limit", "optional_data_source_limit", b"optional_data_source_limit", "optional_validation_time_range", b"optional_validation_time_range", "validation_time_range", b"validation_time_range"]) -> None: ...
     @typing.overload
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_data_source_limit", b"optional_data_source_limit"]) -> typing_extensions.Literal["data_source_limit"] | None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["optional_data_source_limit", b"optional_data_source_limit"]) -> typing.Literal["data_source_limit"] | None: ...
     @typing.overload
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_validation_time_range", b"optional_validation_time_range"]) -> typing_extensions.Literal["validation_time_range"] | None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["optional_validation_time_range", b"optional_validation_time_range"]) -> typing.Literal["validation_time_range"] | None: ...
 
 global___FeatureSetValidationOptions = FeatureSetValidationOptions
 
+@typing.final
 class DataSourceValidationOptions(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -187,12 +193,13 @@ class DataSourceValidationOptions(google.protobuf.message.Message):
         *,
         validation_time_range: global___ValidationTimeRange | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["optional_validation_time_range", b"optional_validation_time_range", "validation_time_range", b"validation_time_range"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["optional_validation_time_range", b"optional_validation_time_range", "validation_time_range", b"validation_time_range"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_validation_time_range", b"optional_validation_time_range"]) -> typing_extensions.Literal["validation_time_range"] | None: ...
+    def HasField(self, field_name: typing.Literal["optional_validation_time_range", b"optional_validation_time_range", "validation_time_range", b"validation_time_range"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["optional_validation_time_range", b"optional_validation_time_range", "validation_time_range", b"validation_time_range"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["optional_validation_time_range", b"optional_validation_time_range"]) -> typing.Literal["validation_time_range"] | None: ...
 
 global___DataSourceValidationOptions = DataSourceValidationOptions
 
+@typing.final
 class ValidationTimeRange(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -208,11 +215,11 @@ class ValidationTimeRange(google.protobuf.message.Message):
         lower_time_bound: google.protobuf.timestamp_pb2.Timestamp | None = ...,
         upper_time_bound: google.protobuf.timestamp_pb2.Timestamp | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["lower_time_bound", b"lower_time_bound", "optional_lower_time_bound", b"optional_lower_time_bound", "optional_upper_time_bound", b"optional_upper_time_bound", "upper_time_bound", b"upper_time_bound"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["lower_time_bound", b"lower_time_bound", "optional_lower_time_bound", b"optional_lower_time_bound", "optional_upper_time_bound", b"optional_upper_time_bound", "upper_time_bound", b"upper_time_bound"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["lower_time_bound", b"lower_time_bound", "optional_lower_time_bound", b"optional_lower_time_bound", "optional_upper_time_bound", b"optional_upper_time_bound", "upper_time_bound", b"upper_time_bound"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["lower_time_bound", b"lower_time_bound", "optional_lower_time_bound", b"optional_lower_time_bound", "optional_upper_time_bound", b"optional_upper_time_bound", "upper_time_bound", b"upper_time_bound"]) -> None: ...
     @typing.overload
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_lower_time_bound", b"optional_lower_time_bound"]) -> typing_extensions.Literal["lower_time_bound"] | None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["optional_lower_time_bound", b"optional_lower_time_bound"]) -> typing.Literal["lower_time_bound"] | None: ...
     @typing.overload
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["optional_upper_time_bound", b"optional_upper_time_bound"]) -> typing_extensions.Literal["upper_time_bound"] | None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["optional_upper_time_bound", b"optional_upper_time_bound"]) -> typing.Literal["upper_time_bound"] | None: ...
 
 global___ValidationTimeRange = ValidationTimeRange

frogml/_proto/qwak/features_operator/v3/features_operator_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.qwak.features_operator/v3/features_operator.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.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['DESCRIPTOR']._options = None
+  _globals['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

frogml/_proto/qwak/features_operator/v3/features_operator_pb2.pyi

@@ -2,20 +2,17 @@
 @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 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 ValidationSuccessResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -28,12 +25,15 @@ class ValidationSuccessResponse(google.protobuf.message.Message):
     @property
     def pandas_column_description(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___PandasColumnDescription]:
         """The pandas column description of the success response"""
+
     @property
     def spark_column_description(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___SparkColumnDescription]:
         """The spark column description of the success response"""
+
     @property
     def outputs(self) -> global___Outputs:
         """the outputs printed during the run"""
+
     def __init__(
         self,
         *,
@@ -42,11 +42,12 @@ class ValidationSuccessResponse(google.protobuf.message.Message):
         spark_column_description: collections.abc.Iterable[global___SparkColumnDescription] | None = ...,
         outputs: global___Outputs | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["outputs", b"outputs"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["outputs", b"outputs", "pandas_column_description", b"pandas_column_description", "sample", b"sample", "spark_column_description", b"spark_column_description"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["outputs", b"outputs"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["outputs", b"outputs", "pandas_column_description", b"pandas_column_description", "sample", b"sample", "spark_column_description", b"spark_column_description"]) -> None: ...
 
 global___ValidationSuccessResponse = ValidationSuccessResponse
 
+@typing.final
 class PandasColumnDescription(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -62,10 +63,11 @@ class PandasColumnDescription(google.protobuf.message.Message):
         column_name: builtins.str = ...,
         pandas_type: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["column_name", b"column_name", "pandas_type", b"pandas_type"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["column_name", b"column_name", "pandas_type", b"pandas_type"]) -> None: ...
 
 global___PandasColumnDescription = PandasColumnDescription
 
+@typing.final
 class SparkColumnDescription(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -81,10 +83,11 @@ class SparkColumnDescription(google.protobuf.message.Message):
         column_name: builtins.str = ...,
         spark_type: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["column_name", b"column_name", "spark_type", b"spark_type"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["column_name", b"column_name", "spark_type", b"spark_type"]) -> None: ...
 
 global___SparkColumnDescription = SparkColumnDescription
 
+@typing.final
 class ValidationFailureResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -101,6 +104,7 @@ class ValidationFailureResponse(google.protobuf.message.Message):
     @property
     def outputs(self) -> global___Outputs:
         """the outputs printed during the run"""
+
     def __init__(
         self,
         *,
@@ -109,11 +113,12 @@ class ValidationFailureResponse(google.protobuf.message.Message):
         description: builtins.str = ...,
         outputs: global___Outputs | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["outputs", b"outputs"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["description", b"description", "error_message", b"error_message", "outputs", b"outputs", "phase", b"phase"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["outputs", b"outputs"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["description", b"description", "error_message", b"error_message", "outputs", b"outputs", "phase", b"phase"]) -> None: ...
 
 global___ValidationFailureResponse = ValidationFailureResponse
 
+@typing.final
 class Outputs(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -129,10 +134,11 @@ class Outputs(google.protobuf.message.Message):
         stdout: builtins.str = ...,
         stderr: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["stderr", b"stderr", "stdout", b"stdout"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["stderr", b"stderr", "stdout", b"stdout"]) -> None: ...
 
 global___Outputs = Outputs
 
+@typing.final
 class ValidationNotReadyResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 

frogml/_proto/qwak/file_versioning/file_versioning_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.qwak.file_versioning/file_versioning.proto
+# Protobuf Python Version: 4.25.1
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
@@ -19,11 +20,10 @@ _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['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n+com.qwak.ai.management.model_versioning.apiP\001'
+  _globals['_FILETAGSPEC'].fields_by_name['extension']._options = None
+  _globals['_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

frogml/_proto/qwak/file_versioning/file_versioning_pb2.pyi

@@ -2,6 +2,7 @@
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
 """
+
 import builtins
 import google.protobuf.descriptor
 import google.protobuf.internal.enum_type_wrapper
@@ -20,7 +21,7 @@ class _FileExtensionType:
     ValueType = typing.NewType("ValueType", builtins.int)
     V: typing_extensions.TypeAlias = ValueType
 
-class _FileExtensionTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_FileExtensionType.ValueType], builtins.type):  # noqa: F821
+class _FileExtensionTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_FileExtensionType.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     INVALID: _FileExtensionType.ValueType  # 0
     """An invalid extension type"""
@@ -43,6 +44,7 @@ JPEG: FileExtensionType.ValueType  # 3
 """JPEG extension type"""
 global___FileExtensionType = FileExtensionType
 
+@typing.final
 class FileTagSpec(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -77,10 +79,11 @@ class FileTagSpec(google.protobuf.message.Message):
         environment_id: builtins.str = ...,
         path: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_id", b"build_id", "environment_id", b"environment_id", "extension", b"extension", "extension_type", b"extension_type", "model_id", b"model_id", "path", b"path", "tag", b"tag"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["build_id", b"build_id", "environment_id", b"environment_id", "extension", b"extension", "extension_type", b"extension_type", "model_id", b"model_id", "path", b"path", "tag", b"tag"]) -> None: ...
 
 global___FileTagSpec = FileTagSpec
 
+@typing.final
 class FileTagFilter(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -96,8 +99,8 @@ class FileTagFilter(google.protobuf.message.Message):
         tag_contains: builtins.str = ...,
         tag_prefix: builtins.str = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["filter", b"filter", "tag_contains", b"tag_contains", "tag_prefix", b"tag_prefix"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["filter", b"filter", "tag_contains", b"tag_contains", "tag_prefix", b"tag_prefix"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["filter", b"filter"]) -> typing_extensions.Literal["tag_contains", "tag_prefix"] | None: ...
+    def HasField(self, field_name: typing.Literal["filter", b"filter", "tag_contains", b"tag_contains", "tag_prefix", b"tag_prefix"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["filter", b"filter", "tag_contains", b"tag_contains", "tag_prefix", b"tag_prefix"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["filter", b"filter"]) -> typing.Literal["tag_contains", "tag_prefix"] | None: ...
 
 global___FileTagFilter = FileTagFilter

frogml/_proto/qwak/file_versioning/file_versioning_service_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.qwak.file_versioning/file_versioning_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
@@ -20,9 +21,8 @@ _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['DESCRIPTOR']._options = None
+  _globals['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

frogml/_proto/qwak/file_versioning/file_versioning_service_pb2.pyi

@@ -2,21 +2,18 @@
 @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
+import typing
 
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 
+@typing.final
 class RegisterFileTagRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -28,11 +25,12 @@ class RegisterFileTagRequest(google.protobuf.message.Message):
         *,
         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: ...
+    def HasField(self, field_name: typing.Literal["file_tag_spec", b"file_tag_spec"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["file_tag_spec", b"file_tag_spec"]) -> None: ...
 
 global___RegisterFileTagRequest = RegisterFileTagRequest
 
+@typing.final
 class RegisterFileTagResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -42,6 +40,7 @@ class RegisterFileTagResponse(google.protobuf.message.Message):
 
 global___RegisterFileTagResponse = RegisterFileTagResponse
 
+@typing.final
 class GetModelFileTagsRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -55,6 +54,7 @@ class GetModelFileTagsRequest(google.protobuf.message.Message):
     @property
     def filter(self) -> frogml._proto.qwak.file_versioning.file_versioning_pb2.FileTagFilter:
         """The filter to apply to the file tags"""
+
     def __init__(
         self,
         *,
@@ -62,11 +62,12 @@ class GetModelFileTagsRequest(google.protobuf.message.Message):
         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: ...
+    def HasField(self, field_name: typing.Literal["filter", b"filter"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["build_id", b"build_id", "filter", b"filter", "model_id", b"model_id"]) -> None: ...
 
 global___GetModelFileTagsRequest = GetModelFileTagsRequest
 
+@typing.final
 class GetModelFileTagsResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -78,6 +79,6 @@ class GetModelFileTagsResponse(google.protobuf.message.Message):
         *,
         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: ...
+    def ClearField(self, field_name: typing.Literal["file_tags", b"file_tags"]) -> None: ...
 
 global___GetModelFileTagsResponse = GetModelFileTagsResponse

frogml/_proto/qwak/fitness_service/constructs_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.qwak.fitness_service/constructs.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
@@ -13,19 +14,18 @@ _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')
+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*\x82\x03\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\x12\x13\n\x0fNVIDIA_H100_NVL\x10\r\x12\x0e\n\nNVIDIA_A10\x10\x0e\x12\x13\n\x0fNVIDIA_H100_SXM\x10\x0f\x12\x14\n\x10NVIDIA_H100_40GB\x10\x10\x12\x14\n\x10NVIDIA_H100_80GB\x10\x11\x12\x0e\n\nNVIDIA_M60\x10\x12\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['DESCRIPTOR']._options = None
+  _globals['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['_GPUTYPE']._serialized_end=1487
   _globals['_BUILDPROPERTIES']._serialized_start=64
   _globals['_BUILDPROPERTIES']._serialized_end=226
   _globals['_MODELURISPEC']._serialized_start=229