update neptune_retrieval_api using the generate_proto script

src/neptune_retrieval_api/proto/neptune_pb/api/model/__init__.pyi

@@ -3,3 +3,6 @@ from . import (
     leaderboard_entries_pb2,
     series_values_pb2,
 )
+from . import attributes_pb2
+from . import leaderboard_entries_pb2
+from . import series_values_pb2

src/neptune_retrieval_api/proto/neptune_pb/api/model/attributes_pb2.py

@@ -1,32 +1,24 @@
 """Generated protocol buffer code."""
-
 from google.protobuf import descriptor as _descriptor
+from neptune_retrieval_api.proto import descriptor_pool as _descriptor_pool
 from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
-
-from neptune_retrieval_api.proto import descriptor_pool as _descriptor_pool
-
 _sym_db = _symbol_database.Default()
-from ....neptune_pb.api.model import (
-    leaderboard_entries_pb2 as neptune__pb_dot_api_dot_model_dot_leaderboard__entries__pb2,
-)
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n%neptune_pb/api/model/attributes.proto\x12\x11neptune.api.model\x1a.neptune_pb/api/model/leaderboard_entries.proto"a\n\x1eProtoAttributesSearchResultDTO\x12?\n\x07entries\x18\x01 \x03(\x0b2..neptune.api.model.ProtoAttributeDefinitionDTO"9\n\x1bProtoAttributeDefinitionDTO\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0c\n\x04type\x18\x02 \x01(\t"\xa3\x01\n\x1dProtoQueryAttributesResultDTO\x12K\n\x07entries\x18\x01 \x03(\x0b2:.neptune.api.model.ProtoQueryAttributesExperimentResultDTO\x125\n\x08nextPage\x18\x02 \x01(\x0b2#.neptune.api.model.ProtoNextPageDTO"^\n\x10ProtoNextPageDTO\x12\x1a\n\rnextPageToken\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x12\n\x05limit\x18\x02 \x01(\rH\x01\x88\x01\x01B\x10\n\x0e_nextPageTokenB\x08\n\x06_limit"\x94\x01\n\'ProtoQueryAttributesExperimentResultDTO\x12\x14\n\x0cexperimentId\x18\x01 \x01(\t\x12\x19\n\x11experimentShortId\x18\x02 \x01(\t\x128\n\nattributes\x18\x03 \x03(\x0b2$.neptune.api.model.ProtoAttributeDTOB4\n0ml.neptune.leaderboard.api.model.proto.generatedP\x01b\x06proto3'
-)
+from ....neptune_pb.api.model import leaderboard_entries_pb2 as neptune__pb_dot_api_dot_model_dot_leaderboard__entries__pb2
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n%neptune_pb/api/model/attributes.proto\x12\x11neptune.api.model\x1a.neptune_pb/api/model/leaderboard_entries.proto"a\n\x1eProtoAttributesSearchResultDTO\x12?\n\x07entries\x18\x01 \x03(\x0b2..neptune.api.model.ProtoAttributeDefinitionDTO"9\n\x1bProtoAttributeDefinitionDTO\x12\x0c\n\x04name\x18\x01 \x01(\t\x12\x0c\n\x04type\x18\x02 \x01(\t"\xa3\x01\n\x1dProtoQueryAttributesResultDTO\x12K\n\x07entries\x18\x01 \x03(\x0b2:.neptune.api.model.ProtoQueryAttributesExperimentResultDTO\x125\n\x08nextPage\x18\x02 \x01(\x0b2#.neptune.api.model.ProtoNextPageDTO"^\n\x10ProtoNextPageDTO\x12\x1a\n\rnextPageToken\x18\x01 \x01(\tH\x00\x88\x01\x01\x12\x12\n\x05limit\x18\x02 \x01(\rH\x01\x88\x01\x01B\x10\n\x0e_nextPageTokenB\x08\n\x06_limit"\x94\x01\n\'ProtoQueryAttributesExperimentResultDTO\x12\x14\n\x0cexperimentId\x18\x01 \x01(\t\x12\x19\n\x11experimentShortId\x18\x02 \x01(\t\x128\n\nattributes\x18\x03 \x03(\x0b2$.neptune.api.model.ProtoAttributeDTOB4\n0ml.neptune.leaderboard.api.model.proto.generatedP\x01b\x06proto3')
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "neptune_pb.api.model.attributes_pb2", _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'neptune_pb.api.model.attributes_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
-    _globals["DESCRIPTOR"]._options = None
-    _globals["DESCRIPTOR"]._serialized_options = b"\n0ml.neptune.leaderboard.api.model.proto.generatedP\x01"
-    _globals["_PROTOATTRIBUTESSEARCHRESULTDTO"]._serialized_start = 108
-    _globals["_PROTOATTRIBUTESSEARCHRESULTDTO"]._serialized_end = 205
-    _globals["_PROTOATTRIBUTEDEFINITIONDTO"]._serialized_start = 207
-    _globals["_PROTOATTRIBUTEDEFINITIONDTO"]._serialized_end = 264
-    _globals["_PROTOQUERYATTRIBUTESRESULTDTO"]._serialized_start = 267
-    _globals["_PROTOQUERYATTRIBUTESRESULTDTO"]._serialized_end = 430
-    _globals["_PROTONEXTPAGEDTO"]._serialized_start = 432
-    _globals["_PROTONEXTPAGEDTO"]._serialized_end = 526
-    _globals["_PROTOQUERYATTRIBUTESEXPERIMENTRESULTDTO"]._serialized_start = 529
-    _globals["_PROTOQUERYATTRIBUTESEXPERIMENTRESULTDTO"]._serialized_end = 677
+    _globals['DESCRIPTOR']._options = None
+    _globals['DESCRIPTOR']._serialized_options = b'\n0ml.neptune.leaderboard.api.model.proto.generatedP\x01'
+    _globals['_PROTOATTRIBUTESSEARCHRESULTDTO']._serialized_start = 108
+    _globals['_PROTOATTRIBUTESSEARCHRESULTDTO']._serialized_end = 205
+    _globals['_PROTOATTRIBUTEDEFINITIONDTO']._serialized_start = 207
+    _globals['_PROTOATTRIBUTEDEFINITIONDTO']._serialized_end = 264
+    _globals['_PROTOQUERYATTRIBUTESRESULTDTO']._serialized_start = 267
+    _globals['_PROTOQUERYATTRIBUTESRESULTDTO']._serialized_end = 430
+    _globals['_PROTONEXTPAGEDTO']._serialized_start = 432
+    _globals['_PROTONEXTPAGEDTO']._serialized_end = 526
+    _globals['_PROTOQUERYATTRIBUTESEXPERIMENTRESULTDTO']._serialized_start = 529
+    _globals['_PROTOQUERYATTRIBUTESEXPERIMENTRESULTDTO']._serialized_end = 677

src/neptune_retrieval_api/proto/neptune_pb/api/model/attributes_pb2.pyi

@@ -2,15 +2,13 @@
 @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
 from .... import neptune_pb
 import typing
-
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 
 @typing.final
@@ -19,14 +17,14 @@ class ProtoAttributesSearchResultDTO(google.protobuf.message.Message):
     ENTRIES_FIELD_NUMBER: builtins.int
 
     @property
-    def entries(
-        self,
-    ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ProtoAttributeDefinitionDTO]: ...
-    def __init__(
-        self, *, entries: collections.abc.Iterable[global___ProtoAttributeDefinitionDTO] | None = ...
-    ) -> None: ...
-    def ClearField(self, field_name: typing.Literal["entries", b"entries"]) -> None: ...
+    def entries(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ProtoAttributeDefinitionDTO]:
+        ...
 
+    def __init__(self, *, entries: collections.abc.Iterable[global___ProtoAttributeDefinitionDTO] | None=...) -> None:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['entries', b'entries']) -> None:
+        ...
 global___ProtoAttributesSearchResultDTO = ProtoAttributesSearchResultDTO
 
 @typing.final
@@ -37,9 +35,11 @@ class ProtoAttributeDefinitionDTO(google.protobuf.message.Message):
     name: builtins.str
     type: builtins.str
 
-    def __init__(self, *, name: builtins.str = ..., type: builtins.str = ...) -> None: ...
-    def ClearField(self, field_name: typing.Literal["name", b"name", "type", b"type"]) -> None: ...
+    def __init__(self, *, name: builtins.str=..., type: builtins.str=...) -> None:
+        ...
 
+    def ClearField(self, field_name: typing.Literal['name', b'name', 'type', b'type']) -> None:
+        ...
 global___ProtoAttributeDefinitionDTO = ProtoAttributeDefinitionDTO
 
 @typing.final
@@ -49,22 +49,21 @@ class ProtoQueryAttributesResultDTO(google.protobuf.message.Message):
     NEXTPAGE_FIELD_NUMBER: builtins.int
 
     @property
-    def entries(
-        self,
-    ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[
-        global___ProtoQueryAttributesExperimentResultDTO
-    ]: ...
+    def entries(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ProtoQueryAttributesExperimentResultDTO]:
+        ...
+
     @property
-    def nextPage(self) -> global___ProtoNextPageDTO: ...
-    def __init__(
-        self,
-        *,
-        entries: collections.abc.Iterable[global___ProtoQueryAttributesExperimentResultDTO] | None = ...,
-        nextPage: global___ProtoNextPageDTO | None = ...,
-    ) -> None: ...
-    def HasField(self, field_name: typing.Literal["nextPage", b"nextPage"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing.Literal["entries", b"entries", "nextPage", b"nextPage"]) -> None: ...
+    def nextPage(self) -> global___ProtoNextPageDTO:
+        ...
 
+    def __init__(self, *, entries: collections.abc.Iterable[global___ProtoQueryAttributesExperimentResultDTO] | None=..., nextPage: global___ProtoNextPageDTO | None=...) -> None:
+        ...
+
+    def HasField(self, field_name: typing.Literal['nextPage', b'nextPage']) -> builtins.bool:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['entries', b'entries', 'nextPage', b'nextPage']) -> None:
+        ...
 global___ProtoQueryAttributesResultDTO = ProtoQueryAttributesResultDTO
 
 @typing.final
@@ -75,40 +74,22 @@ class ProtoNextPageDTO(google.protobuf.message.Message):
     nextPageToken: builtins.str
     limit: builtins.int
 
-    def __init__(self, *, nextPageToken: builtins.str | None = ..., limit: builtins.int | None = ...) -> None: ...
-    def HasField(
-        self,
-        field_name: typing.Literal[
-            "_limit",
-            b"_limit",
-            "_nextPageToken",
-            b"_nextPageToken",
-            "limit",
-            b"limit",
-            "nextPageToken",
-            b"nextPageToken",
-        ],
-    ) -> builtins.bool: ...
-    def ClearField(
-        self,
-        field_name: typing.Literal[
-            "_limit",
-            b"_limit",
-            "_nextPageToken",
-            b"_nextPageToken",
-            "limit",
-            b"limit",
-            "nextPageToken",
-            b"nextPageToken",
-        ],
-    ) -> None: ...
-    @typing.overload
-    def WhichOneof(self, oneof_group: typing.Literal["_limit", b"_limit"]) -> typing.Literal["limit"] | None: ...
+    def __init__(self, *, nextPageToken: builtins.str | None=..., limit: builtins.int | None=...) -> None:
+        ...
+
+    def HasField(self, field_name: typing.Literal['_limit', b'_limit', '_nextPageToken', b'_nextPageToken', 'limit', b'limit', 'nextPageToken', b'nextPageToken']) -> builtins.bool:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['_limit', b'_limit', '_nextPageToken', b'_nextPageToken', 'limit', b'limit', 'nextPageToken', b'nextPageToken']) -> None:
+        ...
+
     @typing.overload
-    def WhichOneof(
-        self, oneof_group: typing.Literal["_nextPageToken", b"_nextPageToken"]
-    ) -> typing.Literal["nextPageToken"] | None: ...
+    def WhichOneof(self, oneof_group: typing.Literal['_limit', b'_limit']) -> typing.Literal['limit'] | None:
+        ...
 
+    @typing.overload
+    def WhichOneof(self, oneof_group: typing.Literal['_nextPageToken', b'_nextPageToken']) -> typing.Literal['nextPageToken'] | None:
+        ...
 global___ProtoNextPageDTO = ProtoNextPageDTO
 
 @typing.final
@@ -121,25 +102,12 @@ class ProtoQueryAttributesExperimentResultDTO(google.protobuf.message.Message):
     experimentShortId: builtins.str
 
     @property
-    def attributes(
-        self,
-    ) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[
-        neptune_pb.api.model.leaderboard_entries_pb2.ProtoAttributeDTO
-    ]: ...
-    def __init__(
-        self,
-        *,
-        experimentId: builtins.str = ...,
-        experimentShortId: builtins.str = ...,
-        attributes: (
-            collections.abc.Iterable[neptune_pb.api.model.leaderboard_entries_pb2.ProtoAttributeDTO] | None
-        ) = ...,
-    ) -> None: ...
-    def ClearField(
-        self,
-        field_name: typing.Literal[
-            "attributes", b"attributes", "experimentId", b"experimentId", "experimentShortId", b"experimentShortId"
-        ],
-    ) -> None: ...
-
-global___ProtoQueryAttributesExperimentResultDTO = ProtoQueryAttributesExperimentResultDTO
+    def attributes(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[neptune_pb.api.model.leaderboard_entries_pb2.ProtoAttributeDTO]:
+        ...
+
+    def __init__(self, *, experimentId: builtins.str=..., experimentShortId: builtins.str=..., attributes: collections.abc.Iterable[neptune_pb.api.model.leaderboard_entries_pb2.ProtoAttributeDTO] | None=...) -> None:
+        ...
+
+    def ClearField(self, field_name: typing.Literal['attributes', b'attributes', 'experimentId', b'experimentId', 'experimentShortId', b'experimentShortId']) -> None:
+        ...
+global___ProtoQueryAttributesExperimentResultDTO = ProtoQueryAttributesExperimentResultDTO