weave-python 0.28.1__py3-none-any.whl → 0.30.3__py3-none-any.whl

weave/weaveapi/synthesize/v1/training_pb2.py

@@ -2,7 +2,7 @@
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # NO CHECKED-IN PROTOBUF GENCODE
 # source: weaveapi/synthesize/v1/training.proto
-# Protobuf Python Version: 6.32.0
+# Protobuf Python Version: 6.33.5
 """Generated protocol buffer code."""
 
 from google.protobuf import descriptor as _descriptor
@@ -14,8 +14,8 @@ from google.protobuf.internal import builder as _builder
 _runtime_version.ValidateProtobufRuntimeVersion(
     _runtime_version.Domain.PUBLIC,
     6,
-    32,
-    0,
+    33,
+    5,
     "",
     "weaveapi/synthesize/v1/training.proto",
 )
@@ -25,7 +25,7 @@ _sym_db = _symbol_database.Default()
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n%weaveapi/synthesize/v1/training.proto\x12\x16weaveapi.synthesize.v1\x1a\x1cgoogle/protobuf/struct.proto"\xb7\x02\n\x1dStartSynthesizerTrainingEvent\x12%\n\x0esynthesizer_id\x18\x01 \x01(\tR\rsynthesizerId\x12\x1f\n\x0b\x64\x61taset_ids\x18\x02 \x03(\tR\ndatasetIds\x12O\n\x0ftraining_driver\x18\x03 \x01(\x0e\x32&.weaveapi.synthesize.v1.TrainingDriverR\x0etrainingDriver\x12<\n\rdriver_config\x18\x04 \x01(\x0b\x32\x17.google.protobuf.StructR\x0c\x64riverConfig\x12?\n\x1coutput_storage_connection_id\x18\x05 \x01(\tR\x19outputStorageConnectionId"\xf5\x01\n\x15TrainingProgressEvent\x12@\n\njob_status\x18\x01 \x01(\x0e\x32!.weaveapi.synthesize.v1.JobStatusR\tjobStatus\x12\x1b\n\tstep_name\x18\x02 \x01(\tR\x08stepName\x12\x42\n\x0bstep_status\x18\x03 \x01(\x0e\x32!.weaveapi.synthesize.v1.JobStatusR\nstepStatus\x12#\n\rstep_progress\x18\x04 \x01(\x02R\x0cstepProgress\x12\x14\n\x05\x65rror\x18\x05 \x01(\tR\x05\x65rror*\x88\x01\n\x0eTrainingDriver\x12\x1f\n\x1bTRAINING_DRIVER_UNSPECIFIED\x10\x00\x12\x1c\n\x18TRAINING_DRIVER_MOSTLYAI\x10\x01\x12\x17\n\x13TRAINING_DRIVER_SDV\x10\x02\x12\x1e\n\x1aTRAINING_DRIVER_SMARTNOISE\x10\x03*\x88\x01\n\tJobStatus\x12\x1a\n\x16JOB_STATUS_UNSPECIFIED\x10\x00\x12\x16\n\x12JOB_STATUS_PENDING\x10\x01\x12\x16\n\x12JOB_STATUS_RUNNING\x10\x02\x12\x18\n\x14JOB_STATUS_COMPLETED\x10\x03\x12\x15\n\x11JOB_STATUS_FAILED\x10\x04\x42\xe7\x01\n\x1a\x63om.weaveapi.synthesize.v1B\rTrainingProtoP\x01Z@github.com/weave-labs/weave-go/weaveapi/synthesize/v1;synthesize\xa2\x02\x03WSX\xaa\x02\x16Weaveapi.Synthesize.V1\xca\x02\x16Weaveapi\\Synthesize\\V1\xe2\x02"Weaveapi\\Synthesize\\V1\\GPBMetadata\xea\x02\x18Weaveapi::Synthesize::V1b\x06proto3'
+    b'\n%weaveapi/synthesize/v1/training.proto\x12\x16weaveapi.synthesize.v1\x1a!google/protobuf/go_features.proto\x1a\x1cgoogle/protobuf/struct.proto"\xbb\x02\n\x1dStartSynthesizerTrainingEvent\x12%\n\x0esynthesizer_id\x18\x01 \x01(\tR\rsynthesizerId\x12\x1f\n\x0b\x64\x61taset_ids\x18\x02 \x03(\tR\ndatasetIds\x12O\n\x0ftraining_driver\x18\x03 \x01(\x0e\x32&.weaveapi.synthesize.v1.TrainingDriverR\x0etrainingDriver\x12@\n\rdriver_config\x18\x04 \x01(\x0b\x32\x17.google.protobuf.StructB\x02(\x01R\x0c\x64riverConfig\x12?\n\x1coutput_storage_connection_id\x18\x05 \x01(\tR\x19outputStorageConnectionId"\xf5\x01\n\x15TrainingProgressEvent\x12@\n\njob_status\x18\x01 \x01(\x0e\x32!.weaveapi.synthesize.v1.JobStatusR\tjobStatus\x12\x1b\n\tstep_name\x18\x02 \x01(\tR\x08stepName\x12\x42\n\x0bstep_status\x18\x03 \x01(\x0e\x32!.weaveapi.synthesize.v1.JobStatusR\nstepStatus\x12#\n\rstep_progress\x18\x04 \x01(\x02R\x0cstepProgress\x12\x14\n\x05\x65rror\x18\x05 \x01(\tR\x05\x65rror*\x88\x01\n\x0eTrainingDriver\x12\x1f\n\x1bTRAINING_DRIVER_UNSPECIFIED\x10\x00\x12\x1c\n\x18TRAINING_DRIVER_MOSTLYAI\x10\x01\x12\x17\n\x13TRAINING_DRIVER_SDV\x10\x02\x12\x1e\n\x1aTRAINING_DRIVER_SMARTNOISE\x10\x03*\x88\x01\n\tJobStatus\x12\x1a\n\x16JOB_STATUS_UNSPECIFIED\x10\x00\x12\x16\n\x12JOB_STATUS_PENDING\x10\x01\x12\x16\n\x12JOB_STATUS_RUNNING\x10\x02\x12\x18\n\x14JOB_STATUS_COMPLETED\x10\x03\x12\x15\n\x11JOB_STATUS_FAILED\x10\x04\x42\xef\x01\n\x1a\x63om.weaveapi.synthesize.v1B\rTrainingProtoP\x01Z@github.com/weave-labs/weave-go/weaveapi/synthesize/v1;synthesize\xa2\x02\x03WSX\xaa\x02\x16Weaveapi.Synthesize.V1\xca\x02\x16Weaveapi\\Synthesize\\V1\xe2\x02"Weaveapi\\Synthesize\\V1\\GPBMetadata\xea\x02\x18Weaveapi::Synthesize::V1\x92\x03\x05\xd2>\x02\x10\x03\x62\x08\x65\x64itionsp\xe8\x07'
 )
 
 _globals = globals()
@@ -37,13 +37,19 @@ if not _descriptor._USE_C_DESCRIPTORS:
     _globals["DESCRIPTOR"]._loaded_options = None
     _globals[
         "DESCRIPTOR"
-    ]._serialized_options = b'\n\032com.weaveapi.synthesize.v1B\rTrainingProtoP\001Z@github.com/weave-labs/weave-go/weaveapi/synthesize/v1;synthesize\242\002\003WSX\252\002\026Weaveapi.Synthesize.V1\312\002\026Weaveapi\\Synthesize\\V1\342\002"Weaveapi\\Synthesize\\V1\\GPBMetadata\352\002\030Weaveapi::Synthesize::V1'
-    _globals["_TRAININGDRIVER"]._serialized_start = 658
-    _globals["_TRAININGDRIVER"]._serialized_end = 794
-    _globals["_JOBSTATUS"]._serialized_start = 797
-    _globals["_JOBSTATUS"]._serialized_end = 933
-    _globals["_STARTSYNTHESIZERTRAININGEVENT"]._serialized_start = 96
-    _globals["_STARTSYNTHESIZERTRAININGEVENT"]._serialized_end = 407
-    _globals["_TRAININGPROGRESSEVENT"]._serialized_start = 410
-    _globals["_TRAININGPROGRESSEVENT"]._serialized_end = 655
+    ]._serialized_options = b'\n\032com.weaveapi.synthesize.v1B\rTrainingProtoP\001Z@github.com/weave-labs/weave-go/weaveapi/synthesize/v1;synthesize\242\002\003WSX\252\002\026Weaveapi.Synthesize.V1\312\002\026Weaveapi\\Synthesize\\V1\342\002"Weaveapi\\Synthesize\\V1\\GPBMetadata\352\002\030Weaveapi::Synthesize::V1\222\003\005\322>\002\020\003'
+    _globals["_STARTSYNTHESIZERTRAININGEVENT"].fields_by_name[
+        "driver_config"
+    ]._loaded_options = None
+    _globals["_STARTSYNTHESIZERTRAININGEVENT"].fields_by_name[
+        "driver_config"
+    ]._serialized_options = b"(\001"
+    _globals["_TRAININGDRIVER"]._serialized_start = 697
+    _globals["_TRAININGDRIVER"]._serialized_end = 833
+    _globals["_JOBSTATUS"]._serialized_start = 836
+    _globals["_JOBSTATUS"]._serialized_end = 972
+    _globals["_STARTSYNTHESIZERTRAININGEVENT"]._serialized_start = 131
+    _globals["_STARTSYNTHESIZERTRAININGEVENT"]._serialized_end = 446
+    _globals["_TRAININGPROGRESSEVENT"]._serialized_start = 449
+    _globals["_TRAININGPROGRESSEVENT"]._serialized_end = 694
 # @@protoc_insertion_point(module_scope)

weave/weaveapi/synthesize/v1/training_pb2.pyi

@@ -3,34 +3,31 @@
 isort:skip_file
 """
 
-import builtins
-import collections.abc
-import google.protobuf.descriptor
-import google.protobuf.internal.containers
-import google.protobuf.internal.enum_type_wrapper
-import google.protobuf.message
-import google.protobuf.struct_pb2
+from collections import abc as _abc
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from google.protobuf import struct_pb2 as _struct_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
+import builtins as _builtins
 import sys
-import typing
+import typing as _typing
 
 if sys.version_info >= (3, 10):
-    import typing as typing_extensions
+    from typing import TypeAlias as _TypeAlias
 else:
-    import typing_extensions
+    from typing_extensions import TypeAlias as _TypeAlias
 
-DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
+DESCRIPTOR: _descriptor.FileDescriptor
 
 class _TrainingDriver:
-    ValueType = typing.NewType("ValueType", builtins.int)
-    V: typing_extensions.TypeAlias = ValueType
+    ValueType = _typing.NewType("ValueType", _builtins.int)
+    V: _TypeAlias = ValueType  # noqa: Y015
 
 class _TrainingDriverEnumTypeWrapper(
-    google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[
-        _TrainingDriver.ValueType
-    ],
-    builtins.type,
+    _enum_type_wrapper._EnumTypeWrapper[_TrainingDriver.ValueType], _builtins.type
 ):
-    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    DESCRIPTOR: _descriptor.EnumDescriptor
     TRAINING_DRIVER_UNSPECIFIED: _TrainingDriver.ValueType  # 0
     TRAINING_DRIVER_MOSTLYAI: _TrainingDriver.ValueType  # 1
     TRAINING_DRIVER_SDV: _TrainingDriver.ValueType  # 2
@@ -42,17 +39,16 @@ TRAINING_DRIVER_UNSPECIFIED: TrainingDriver.ValueType  # 0
 TRAINING_DRIVER_MOSTLYAI: TrainingDriver.ValueType  # 1
 TRAINING_DRIVER_SDV: TrainingDriver.ValueType  # 2
 TRAINING_DRIVER_SMARTNOISE: TrainingDriver.ValueType  # 3
-global___TrainingDriver = TrainingDriver
+Global___TrainingDriver: _TypeAlias = TrainingDriver  # noqa: Y015
 
 class _JobStatus:
-    ValueType = typing.NewType("ValueType", builtins.int)
-    V: typing_extensions.TypeAlias = ValueType
+    ValueType = _typing.NewType("ValueType", _builtins.int)
+    V: _TypeAlias = ValueType  # noqa: Y015
 
 class _JobStatusEnumTypeWrapper(
-    google.protobuf.internal.enum_type_wrapper._EnumTypeWrapper[_JobStatus.ValueType],
-    builtins.type,
+    _enum_type_wrapper._EnumTypeWrapper[_JobStatus.ValueType], _builtins.type
 ):
-    DESCRIPTOR: google.protobuf.descriptor.EnumDescriptor
+    DESCRIPTOR: _descriptor.EnumDescriptor
     JOB_STATUS_UNSPECIFIED: _JobStatus.ValueType  # 0
     JOB_STATUS_PENDING: _JobStatus.ValueType  # 1
     JOB_STATUS_RUNNING: _JobStatus.ValueType  # 2
@@ -66,95 +62,112 @@ JOB_STATUS_PENDING: JobStatus.ValueType  # 1
 JOB_STATUS_RUNNING: JobStatus.ValueType  # 2
 JOB_STATUS_COMPLETED: JobStatus.ValueType  # 3
 JOB_STATUS_FAILED: JobStatus.ValueType  # 4
-global___JobStatus = JobStatus
-
-@typing.final
-class StartSynthesizerTrainingEvent(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    SYNTHESIZER_ID_FIELD_NUMBER: builtins.int
-    DATASET_IDS_FIELD_NUMBER: builtins.int
-    TRAINING_DRIVER_FIELD_NUMBER: builtins.int
-    DRIVER_CONFIG_FIELD_NUMBER: builtins.int
-    OUTPUT_STORAGE_CONNECTION_ID_FIELD_NUMBER: builtins.int
-    synthesizer_id: builtins.str
-    training_driver: global___TrainingDriver.ValueType
-    output_storage_connection_id: builtins.str
-    @property
+Global___JobStatus: _TypeAlias = JobStatus  # noqa: Y015
+
+@_typing.final
+class StartSynthesizerTrainingEvent(_message.Message):
+    DESCRIPTOR: _descriptor.Descriptor
+
+    SYNTHESIZER_ID_FIELD_NUMBER: _builtins.int
+    DATASET_IDS_FIELD_NUMBER: _builtins.int
+    TRAINING_DRIVER_FIELD_NUMBER: _builtins.int
+    DRIVER_CONFIG_FIELD_NUMBER: _builtins.int
+    OUTPUT_STORAGE_CONNECTION_ID_FIELD_NUMBER: _builtins.int
+    synthesizer_id: _builtins.str
+    training_driver: Global___TrainingDriver.ValueType
+    output_storage_connection_id: _builtins.str
+    @_builtins.property
     def dataset_ids(
         self,
-    ) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[
-        builtins.str
-    ]: ...
-    @property
-    def driver_config(self) -> google.protobuf.struct_pb2.Struct: ...
+    ) -> _containers.RepeatedScalarFieldContainer[_builtins.str]: ...
+    @_builtins.property
+    def driver_config(self) -> _struct_pb2.Struct: ...
     def __init__(
         self,
         *,
-        synthesizer_id: builtins.str = ...,
-        dataset_ids: collections.abc.Iterable[builtins.str] | None = ...,
-        training_driver: global___TrainingDriver.ValueType = ...,
-        driver_config: google.protobuf.struct_pb2.Struct | None = ...,
-        output_storage_connection_id: builtins.str = ...,
+        synthesizer_id: _builtins.str | None = ...,
+        dataset_ids: _abc.Iterable[_builtins.str] | None = ...,
+        training_driver: Global___TrainingDriver.ValueType | None = ...,
+        driver_config: _struct_pb2.Struct | None = ...,
+        output_storage_connection_id: _builtins.str | None = ...,
     ) -> None: ...
-    def HasField(
-        self, field_name: typing.Literal["driver_config", b"driver_config"]
-    ) -> builtins.bool: ...
-    def ClearField(
-        self,
-        field_name: typing.Literal[
-            "dataset_ids",
-            b"dataset_ids",
-            "driver_config",
-            b"driver_config",
-            "output_storage_connection_id",
-            b"output_storage_connection_id",
-            "synthesizer_id",
-            b"synthesizer_id",
-            "training_driver",
-            b"training_driver",
-        ],
-    ) -> None: ...
-
-global___StartSynthesizerTrainingEvent = StartSynthesizerTrainingEvent
-
-@typing.final
-class TrainingProgressEvent(google.protobuf.message.Message):
-    DESCRIPTOR: google.protobuf.descriptor.Descriptor
-
-    JOB_STATUS_FIELD_NUMBER: builtins.int
-    STEP_NAME_FIELD_NUMBER: builtins.int
-    STEP_STATUS_FIELD_NUMBER: builtins.int
-    STEP_PROGRESS_FIELD_NUMBER: builtins.int
-    ERROR_FIELD_NUMBER: builtins.int
-    job_status: global___JobStatus.ValueType
-    step_name: builtins.str
-    step_status: global___JobStatus.ValueType
-    step_progress: builtins.float
-    error: builtins.str
+    _HasFieldArgType: _TypeAlias = _typing.Literal[
+        "dataset_ids",
+        b"dataset_ids",
+        "driver_config",
+        b"driver_config",
+        "output_storage_connection_id",
+        b"output_storage_connection_id",
+        "synthesizer_id",
+        b"synthesizer_id",
+        "training_driver",
+        b"training_driver",
+    ]  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal[
+        "dataset_ids",
+        b"dataset_ids",
+        "driver_config",
+        b"driver_config",
+        "output_storage_connection_id",
+        b"output_storage_connection_id",
+        "synthesizer_id",
+        b"synthesizer_id",
+        "training_driver",
+        b"training_driver",
+    ]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
+
+Global___StartSynthesizerTrainingEvent: _TypeAlias = StartSynthesizerTrainingEvent  # noqa: Y015
+
+@_typing.final
+class TrainingProgressEvent(_message.Message):
+    DESCRIPTOR: _descriptor.Descriptor
+
+    JOB_STATUS_FIELD_NUMBER: _builtins.int
+    STEP_NAME_FIELD_NUMBER: _builtins.int
+    STEP_STATUS_FIELD_NUMBER: _builtins.int
+    STEP_PROGRESS_FIELD_NUMBER: _builtins.int
+    ERROR_FIELD_NUMBER: _builtins.int
+    job_status: Global___JobStatus.ValueType
+    step_name: _builtins.str
+    step_status: Global___JobStatus.ValueType
+    step_progress: _builtins.float
+    error: _builtins.str
     def __init__(
         self,
         *,
-        job_status: global___JobStatus.ValueType = ...,
-        step_name: builtins.str = ...,
-        step_status: global___JobStatus.ValueType = ...,
-        step_progress: builtins.float = ...,
-        error: builtins.str = ...,
+        job_status: Global___JobStatus.ValueType | None = ...,
+        step_name: _builtins.str | None = ...,
+        step_status: Global___JobStatus.ValueType | None = ...,
+        step_progress: _builtins.float | None = ...,
+        error: _builtins.str | None = ...,
     ) -> None: ...
-    def ClearField(
-        self,
-        field_name: typing.Literal[
-            "error",
-            b"error",
-            "job_status",
-            b"job_status",
-            "step_name",
-            b"step_name",
-            "step_progress",
-            b"step_progress",
-            "step_status",
-            b"step_status",
-        ],
-    ) -> None: ...
-
-global___TrainingProgressEvent = TrainingProgressEvent
+    _HasFieldArgType: _TypeAlias = _typing.Literal[
+        "error",
+        b"error",
+        "job_status",
+        b"job_status",
+        "step_name",
+        b"step_name",
+        "step_progress",
+        b"step_progress",
+        "step_status",
+        b"step_status",
+    ]  # noqa: Y015
+    def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
+    _ClearFieldArgType: _TypeAlias = _typing.Literal[
+        "error",
+        b"error",
+        "job_status",
+        b"job_status",
+        "step_name",
+        b"step_name",
+        "step_progress",
+        b"step_progress",
+        "step_status",
+        b"step_status",
+    ]  # noqa: Y015
+    def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
+
+Global___TrainingProgressEvent: _TypeAlias = TrainingProgressEvent  # noqa: Y015

weave/weaveapi/synthesize/v1/training_pb2_grpc.pyi

@@ -3,18 +3,19 @@
 isort:skip_file
 """
 
-import abc
-import collections.abc
-import grpc
-import grpc.aio
-import typing
+from collections import abc as _abc
+from grpc import aio as _aio
+import abc as _abc_1
+import grpc as _grpc
+import typing as _typing
 
-_T = typing.TypeVar("_T")
+_T = _typing.TypeVar("_T")
 
 class _MaybeAsyncIterator(
-    collections.abc.AsyncIterator[_T],
-    collections.abc.Iterator[_T],
-    metaclass=abc.ABCMeta,
+    _abc.AsyncIterator[_T], _abc.Iterator[_T], metaclass=_abc_1.ABCMeta
 ): ...
-class _ServicerContext(grpc.ServicerContext, grpc.aio.ServicerContext):  # type: ignore[misc, type-arg]
+class _ServicerContext(_grpc.ServicerContext, _aio.ServicerContext):  # type: ignore[misc, type-arg]
     ...
+
+GRPC_GENERATED_VERSION: str
+GRPC_VERSION: str