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

frogml/__init__.py

@@ -1,8 +1,10 @@
 __author__ = "jfrog"
-__version__ = "1.1.62"
+__version__ = "1.2.19"
 
 from frogml.sdk.model.decorators.api import api_decorator as api
 from frogml.sdk.model_loggers.model_logger import load_model, log_model
+from frogml.sdk.frogml_client.client import FrogMLClient
+from frogml.sdk.model.base import BaseModel as FrogMlModel
 
 from frogml.core.model_loggers.artifact_logger import (
     load_file,

frogml/_proto/jfml/hosting_gateway/v1/build_upload_url_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.jfml.hosting_gateway/v1/build_upload_url.proto
+# Protobuf Python Version: 4.25.1
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
@@ -19,9 +20,8 @@ _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'jfml.hosting_gateway.v1.build_upload_url_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n#com.jfrog.ml.hosting_gateway.api.v1P\001'
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n#com.jfrog.ml.hosting_gateway.api.v1P\001'
   _globals['_BUILDARTIFACTTYPE']._serialized_start=289
   _globals['_BUILDARTIFACTTYPE']._serialized_end=407
   _globals['_BUILDUPLOADURLPARAMETERS']._serialized_start=76

frogml/_proto/jfml/hosting_gateway/v1/build_upload_url_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 _BuildArtifactType:
     ValueType = typing.NewType("ValueType", builtins.int)
     V: typing_extensions.TypeAlias = ValueType
 
-class _BuildArtifactTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_BuildArtifactType.ValueType], builtins.type):  # noqa: F821
+class _BuildArtifactTypeEnumTypeWrapper(google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_BuildArtifactType.ValueType], builtins.type):
     DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
     INVALID_ARTIFACT_TYPE: _BuildArtifactType.ValueType  # 0
     DATA_ARTIFACT_TYPE: _BuildArtifactType.ValueType  # 1
@@ -36,6 +37,7 @@ FILE_ARTIFACT_TYPE: BuildArtifactType.ValueType  # 2
 CODE_ARTIFACT_TYPE: BuildArtifactType.ValueType  # 3
 global___BuildArtifactType = BuildArtifactType
 
+@typing.final
 class BuildUploadUrlParameters(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -63,8 +65,8 @@ class BuildUploadUrlParameters(google.protobuf.message.Message):
         artifact_type: global___BuildArtifactType.ValueType = ...,
         use_cloud_provider_storage: builtins.bool = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["cloud_provider_option", b"cloud_provider_option", "use_cloud_provider_storage", b"use_cloud_provider_storage"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["artifact_type", b"artifact_type", "build_id", b"build_id", "cloud_provider_option", b"cloud_provider_option", "model_id", b"model_id", "tag_value", b"tag_value", "use_cloud_provider_storage", b"use_cloud_provider_storage"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["cloud_provider_option", b"cloud_provider_option"]) -> typing_extensions.Literal["use_cloud_provider_storage"] | None: ...
+    def HasField(self, field_name: typing.Literal["cloud_provider_option", b"cloud_provider_option", "use_cloud_provider_storage", b"use_cloud_provider_storage"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["artifact_type", b"artifact_type", "build_id", b"build_id", "cloud_provider_option", b"cloud_provider_option", "model_id", b"model_id", "tag_value", b"tag_value", "use_cloud_provider_storage", b"use_cloud_provider_storage"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["cloud_provider_option", b"cloud_provider_option"]) -> typing.Literal["use_cloud_provider_storage"] | None: ...
 
 global___BuildUploadUrlParameters = BuildUploadUrlParameters

frogml/_proto/jfml/hosting_gateway/v1/hosting_gateway_service_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.jfml.hosting_gateway/v1/hosting_gateway_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, 'jfml.hosting_gateway.v1.hosting_gateway_service_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n#com.jfrog.ml.hosting_gateway.api.v1P\001'
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n#com.jfrog.ml.hosting_gateway.api.v1P\001'
   _globals['_GETBUILDVERSIONINGUPLOADURLREQUEST']._serialized_start=130
   _globals['_GETBUILDVERSIONINGUPLOADURLREQUEST']._serialized_end=233
   _globals['_GETBUILDVERSIONINGUPLOADURLRESPONSE']._serialized_start=235

frogml/_proto/jfml/hosting_gateway/v1/hosting_gateway_service_pb2.pyi

@@ -2,19 +2,16 @@
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
 """
+
 import builtins
 import google.protobuf.descriptor
 import google.protobuf.message
 import frogml._proto.jfml.hosting_gateway.v1.build_upload_url_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 GetBuildVersioningUploadURLRequest(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -22,16 +19,18 @@ class GetBuildVersioningUploadURLRequest(google.protobuf.message.Message):
     @property
     def params(self) -> frogml._proto.jfml.hosting_gateway.v1.build_upload_url_pb2.BuildUploadUrlParameters:
         """The params to define the url."""
+
     def __init__(
         self,
         *,
         params: frogml._proto.jfml.hosting_gateway.v1.build_upload_url_pb2.BuildUploadUrlParameters | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["params", b"params"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["params", b"params"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["params", b"params"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["params", b"params"]) -> None: ...
 
 global___GetBuildVersioningUploadURLRequest = GetBuildVersioningUploadURLRequest
 
+@typing.final
 class GetBuildVersioningUploadURLResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -43,6 +42,6 @@ class GetBuildVersioningUploadURLResponse(google.protobuf.message.Message):
         *,
         upload_url: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["upload_url", b"upload_url"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["upload_url", b"upload_url"]) -> None: ...
 
 global___GetBuildVersioningUploadURLResponse = GetBuildVersioningUploadURLResponse

frogml/_proto/jfml/model_version/v1/artifact_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.jfml.model_version/v1/artifact.proto
+# Protobuf Python Version: 4.25.1
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
@@ -19,9 +20,8 @@ _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'jfml.model_version.v1.artifact_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
   _globals['_ARTIFACT']._serialized_start=63
   _globals['_ARTIFACT']._serialized_end=163
   _globals['_CHECKSUMS']._serialized_start=165

frogml/_proto/jfml/model_version/v1/artifact_pb2.pyi

@@ -2,18 +2,15 @@
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
 """
+
 import builtins
 import google.protobuf.descriptor
 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 Artifact(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -27,6 +24,7 @@ class Artifact(google.protobuf.message.Message):
     @property
     def checksums(self) -> global___Checksums:
         """Checksums of the artifact"""
+
     def __init__(
         self,
         *,
@@ -34,11 +32,12 @@ class Artifact(google.protobuf.message.Message):
         size: builtins.int = ...,
         checksums: global___Checksums | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["checksums", b"checksums"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["artifact_path", b"artifact_path", "checksums", b"checksums", "size", b"size"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["checksums", b"checksums"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["artifact_path", b"artifact_path", "checksums", b"checksums", "size", b"size"]) -> None: ...
 
 global___Artifact = Artifact
 
+@typing.final
 class Checksums(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -50,6 +49,6 @@ class Checksums(google.protobuf.message.Message):
         *,
         sha2: builtins.str = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["sha2", b"sha2"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["sha2", b"sha2"]) -> None: ...
 
 global___Checksums = Checksums

frogml/_proto/jfml/model_version/v1/build_spec_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.jfml.model_version/v1/build_spec.proto
+# Protobuf Python Version: 4.25.1
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
@@ -21,9 +22,8 @@ _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'jfml.model_version.v1.build_spec_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
   _globals['_BUILDSPEC']._serialized_start=151
   _globals['_BUILDSPEC']._serialized_end=467
   _globals['_BUILDPROPERTIES']._serialized_start=469

frogml/_proto/jfml/model_version/v1/build_spec_pb2.pyi

@@ -2,6 +2,7 @@
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
 """
+
 import builtins
 import collections.abc
 import google.protobuf.descriptor
@@ -9,15 +10,11 @@ import google.protobuf.internal.containers
 import google.protobuf.message
 import frogml._proto.qwak.fitness_service.fitness_pb2
 import frogml._proto.qwak.user_application.common.v0.resources_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 BuildSpec(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -27,20 +24,22 @@ class BuildSpec(google.protobuf.message.Message):
     VERBOSE_FIELD_NUMBER: builtins.int
     ENVIRONMENT_ID_FIELD_NUMBER: builtins.int
     PURCHASE_OPTION_FIELD_NUMBER: builtins.int
+    verbose: builtins.int
+    """The build verbosity level (0: warning, 1: info, 2: debug)"""
+    environment_id: builtins.str
+    """The environment to build on"""
+    purchase_option: frogml._proto.qwak.fitness_service.fitness_pb2.PurchaseOption.ValueType
+    """Purchase option for the build"""
     @property
     def build_properties(self) -> global___BuildProperties:
         """The build properties"""
+
     @property
     def build_env(self) -> global___BuildEnv:
         """The build environment"""
+
     @property
     def pod_resources(self) -> frogml._proto.qwak.user_application.common.v0.resources_pb2.ClientPodComputeResources: ...
-    verbose: builtins.int
-    """The build verbosity level (0: warning, 1: info, 2: debug)"""
-    environment_id: builtins.str
-    """The environment to build on"""
-    purchase_option: frogml._proto.qwak.fitness_service.fitness_pb2.PurchaseOption.ValueType
-    """Purchase option for the build"""
     def __init__(
         self,
         *,
@@ -51,11 +50,12 @@ class BuildSpec(google.protobuf.message.Message):
         environment_id: builtins.str = ...,
         purchase_option: frogml._proto.qwak.fitness_service.fitness_pb2.PurchaseOption.ValueType = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["build_env", b"build_env", "build_properties", b"build_properties", "pod_resources", b"pod_resources"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_env", b"build_env", "build_properties", b"build_properties", "environment_id", b"environment_id", "pod_resources", b"pod_resources", "purchase_option", b"purchase_option", "verbose", b"verbose"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["build_env", b"build_env", "build_properties", b"build_properties", "pod_resources", b"pod_resources"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["build_env", b"build_env", "build_properties", b"build_properties", "environment_id", b"environment_id", "pod_resources", b"pod_resources", "purchase_option", b"purchase_option", "verbose", b"verbose"]) -> None: ...
 
 global___BuildSpec = BuildSpec
 
+@typing.final
 class BuildProperties(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -64,11 +64,12 @@ class BuildProperties(google.protobuf.message.Message):
     GPU_COMPATIBLE_FIELD_NUMBER: builtins.int
     build_name: builtins.str
     """Optional - The name of the build we want"""
+    gpu_compatible: builtins.bool
+    """Optional - Should the build be GPU compatible"""
     @property
     def tags(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
         """Optional - The Build's tags"""
-    gpu_compatible: builtins.bool
-    """Optional - Should the build be GPU compatible"""
+
     def __init__(
         self,
         *,
@@ -76,10 +77,11 @@ class BuildProperties(google.protobuf.message.Message):
         tags: collections.abc.Iterable[builtins.str] | None = ...,
         gpu_compatible: builtins.bool = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["build_name", b"build_name", "gpu_compatible", b"gpu_compatible", "tags", b"tags"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["build_name", b"build_name", "gpu_compatible", b"gpu_compatible", "tags", b"tags"]) -> None: ...
 
 global___BuildProperties = BuildProperties
 
+@typing.final
 class BuildEnv(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -87,16 +89,18 @@ class BuildEnv(google.protobuf.message.Message):
     @property
     def docker_env(self) -> global___DockerEnv:
         """Optional - The docker specs to use for the build"""
+
     def __init__(
         self,
         *,
         docker_env: global___DockerEnv | None = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["docker_env", b"docker_env"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["docker_env", b"docker_env"]) -> None: ...
+    def HasField(self, field_name: typing.Literal["docker_env", b"docker_env"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["docker_env", b"docker_env"]) -> None: ...
 
 global___BuildEnv = BuildEnv
 
+@typing.final
 class DockerEnv(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -115,6 +119,6 @@ class DockerEnv(google.protobuf.message.Message):
         base_image: builtins.str = ...,
         no_cache: builtins.bool = ...,
     ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["base_image", b"base_image", "no_cache", b"no_cache"]) -> None: ...
+    def ClearField(self, field_name: typing.Literal["base_image", b"base_image", "no_cache", b"no_cache"]) -> None: ...
 
 global___DockerEnv = DockerEnv

frogml/_proto/jfml/model_version/v1/model_repository_spec_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.jfml.model_version/v1/model_repository_spec.proto
+# Protobuf Python Version: 4.25.1
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
@@ -19,9 +20,8 @@ _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'jfml.model_version.v1.model_repository_spec_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
   _globals['_MODELREPOSITORYSPEC']._serialized_start=77
   _globals['_MODELREPOSITORYSPEC']._serialized_end=205
 # @@protoc_insertion_point(module_scope)

frogml/_proto/jfml/model_version/v1/model_repository_spec_pb2.pyi

@@ -2,18 +2,15 @@
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
 """
+
 import builtins
 import google.protobuf.descriptor
 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 ModelRepositorySpec(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -34,8 +31,8 @@ class ModelRepositorySpec(google.protobuf.message.Message):
         model_id: builtins.str = ...,
         model_name: builtins.str = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["project_identifier", b"project_identifier", "project_key", b"project_key"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["model_id", b"model_id", "model_name", b"model_name", "project_identifier", b"project_identifier", "project_key", b"project_key", "repository_key", b"repository_key"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["project_identifier", b"project_identifier"]) -> typing_extensions.Literal["project_key"] | None: ...
+    def HasField(self, field_name: typing.Literal["project_identifier", b"project_identifier", "project_key", b"project_key"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["model_id", b"model_id", "model_name", b"model_name", "project_identifier", b"project_identifier", "project_key", b"project_key", "repository_key", b"repository_key"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["project_identifier", b"project_identifier"]) -> typing.Literal["project_key"] | None: ...
 
 global___ModelRepositorySpec = ModelRepositorySpec

frogml/_proto/jfml/model_version/v1/model_version_framework_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.jfml.model_version/v1/model_version_framework.proto
+# Protobuf Python Version: 4.25.1
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
@@ -19,9 +20,8 @@ _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'jfml.model_version.v1.model_version_framework_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
   _globals['_MODELVERSIONFRAMEWORK']._serialized_start=79
   _globals['_MODELVERSIONFRAMEWORK']._serialized_end=512
   _globals['_CATBOOSTFRAMEWORK']._serialized_start=514

frogml/_proto/jfml/model_version/v1/model_version_framework_pb2.pyi

@@ -2,18 +2,15 @@
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
 """
+
 import builtins
 import google.protobuf.descriptor
 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 ModelVersionFramework(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -24,6 +21,7 @@ class ModelVersionFramework(google.protobuf.message.Message):
     PYTORCH_FIELD_NUMBER: builtins.int
     SCIKIT_LEARN_FIELD_NUMBER: builtins.int
     VERSION_FIELD_NUMBER: builtins.int
+    version: builtins.str
     @property
     def catboost(self) -> global___CatboostFramework: ...
     @property
@@ -36,7 +34,6 @@ class ModelVersionFramework(google.protobuf.message.Message):
     def pytorch(self) -> global___PytorchFramework: ...
     @property
     def scikit_learn(self) -> global___ScikitLearnFramework: ...
-    version: builtins.str
     def __init__(
         self,
         *,
@@ -48,12 +45,13 @@ class ModelVersionFramework(google.protobuf.message.Message):
         scikit_learn: global___ScikitLearnFramework | None = ...,
         version: builtins.str = ...,
     ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["catboost", b"catboost", "hugging_face", b"hugging_face", "name", b"name", "onnx", b"onnx", "python_pickle", b"python_pickle", "pytorch", b"pytorch", "scikit_learn", b"scikit_learn"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["catboost", b"catboost", "hugging_face", b"hugging_face", "name", b"name", "onnx", b"onnx", "python_pickle", b"python_pickle", "pytorch", b"pytorch", "scikit_learn", b"scikit_learn", "version", b"version"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["name", b"name"]) -> typing_extensions.Literal["catboost", "hugging_face", "onnx", "python_pickle", "pytorch", "scikit_learn"] | None: ...
+    def HasField(self, field_name: typing.Literal["catboost", b"catboost", "hugging_face", b"hugging_face", "name", b"name", "onnx", b"onnx", "python_pickle", b"python_pickle", "pytorch", b"pytorch", "scikit_learn", b"scikit_learn"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["catboost", b"catboost", "hugging_face", b"hugging_face", "name", b"name", "onnx", b"onnx", "python_pickle", b"python_pickle", "pytorch", b"pytorch", "scikit_learn", b"scikit_learn", "version", b"version"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["name", b"name"]) -> typing.Literal["catboost", "hugging_face", "onnx", "python_pickle", "pytorch", "scikit_learn"] | None: ...
 
 global___ModelVersionFramework = ModelVersionFramework
 
+@typing.final
 class CatboostFramework(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -63,6 +61,7 @@ class CatboostFramework(google.protobuf.message.Message):
 
 global___CatboostFramework = CatboostFramework
 
+@typing.final
 class HuggingFaceFramework(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -72,6 +71,7 @@ class HuggingFaceFramework(google.protobuf.message.Message):
 
 global___HuggingFaceFramework = HuggingFaceFramework
 
+@typing.final
 class OnnxFramework(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -81,6 +81,7 @@ class OnnxFramework(google.protobuf.message.Message):
 
 global___OnnxFramework = OnnxFramework
 
+@typing.final
 class PythonPickleFramework(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -90,6 +91,7 @@ class PythonPickleFramework(google.protobuf.message.Message):
 
 global___PythonPickleFramework = PythonPickleFramework
 
+@typing.final
 class PytorchFramework(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
@@ -99,6 +101,7 @@ class PytorchFramework(google.protobuf.message.Message):
 
 global___PytorchFramework = PytorchFramework
 
+@typing.final
 class ScikitLearnFramework(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 

frogml/_proto/jfml/model_version/v1/model_version_manager_service_pb2.py

@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: frogml._proto.jfml.model_version/v1/model_version_manager_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
@@ -22,9 +23,8 @@ _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'jfml.model_version.v1.model_version_manager_service_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
-
-  DESCRIPTOR._options = None
-  DESCRIPTOR._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
+  _globals['DESCRIPTOR']._options = None
+  _globals['DESCRIPTOR']._serialized_options = b'\n!com.jfrog.ml.model_version.api.v1P\001'
   _globals['_CREATEMODELVERSIONREQUEST']._serialized_start=219
   _globals['_CREATEMODELVERSIONREQUEST']._serialized_end=382
   _globals['_CREATEMODELVERSIONRESPONSE']._serialized_start=384