weave-python 0.12.0__tar.gz → 0.14.0__tar.gz

{weave_python-0.12.0 → weave_python-0.14.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: weave-python
-Version: 0.12.0
+Version: 0.14.0
 Requires-Python: >=3.10
 Requires-Dist: protobuf>=6.30.2
 Requires-Dist: sqlalchemy>=2.0.40

{weave_python-0.12.0 → weave_python-0.14.0}/weave/datamanagement/synthesize/v1/service_pb2.py

@@ -25,7 +25,7 @@ _sym_db = _symbol_database.Default()
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n0weave/datamanagement/synthesize/v1/service.proto\x12"weave.datamanagement.synthesize.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x35weave/datamanagement/synthesize/v1/relationship.proto"\xc2\x01\n\x18\x43reateSynthesizerRequest\x12\'\n\x0forganization_id\x18\x01 \x01(\tR\x0eorganizationId\x12\x1f\n\x0b\x64\x61taset_ids\x18\x02 \x03(\tR\ndatasetIds\x12\\\n\rrelationships\x18\x03 \x03(\x0b\x32\x36.weave.datamanagement.synthesize.v1.ColumnRelationshipR\rrelationships"<\n\x19\x43reateSynthesizerResponse\x12\x1f\n\x0btraining_id\x18\x01 \x01(\tR\ntrainingId2\xc1\x01\n\nSynthesize\x12\xb2\x01\n\x11\x43reateSynthesizer\x12<.weave.datamanagement.synthesize.v1.CreateSynthesizerRequest\x1a=.weave.datamanagement.synthesize.v1.CreateSynthesizerResponse" \x82\xd3\xe4\x93\x02\x1a"\x15/v1/synthesize/create:\x01*B\xb3\x02\n&com.weave.datamanagement.synthesize.v1B\x0cServiceProtoP\x01ZPgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/synthesize/v1;synthesize\xa2\x02\x03WDS\xaa\x02"Weave.Datamanagement.Synthesize.V1\xca\x02"Weave\\Datamanagement\\Synthesize\\V1\xe2\x02.Weave\\Datamanagement\\Synthesize\\V1\\GPBMetadata\xea\x02%Weave::Datamanagement::Synthesize::V1b\x06proto3'
+    b'\n0weave/datamanagement/synthesize/v1/service.proto\x12"weave.datamanagement.synthesize.v1\x1a\x1cgoogle/api/annotations.proto\x1a\x35weave/datamanagement/synthesize/v1/relationship.proto"\x83\x02\n\x18\x43reateSynthesizerRequest\x12\'\n\x0forganization_id\x18\x01 \x01(\tR\x0eorganizationId\x12\x1f\n\x0b\x64\x61taset_ids\x18\x02 \x03(\tR\ndatasetIds\x12\\\n\rrelationships\x18\x03 \x03(\x0b\x32\x36.weave.datamanagement.synthesize.v1.ColumnRelationshipR\rrelationships\x12?\n\x1coutput_storage_connection_id\x18\x04 \x01(\tR\x19outputStorageConnectionId"<\n\x19\x43reateSynthesizerResponse\x12\x1f\n\x0btraining_id\x18\x01 \x01(\tR\ntrainingId2\xc1\x01\n\nSynthesize\x12\xb2\x01\n\x11\x43reateSynthesizer\x12<.weave.datamanagement.synthesize.v1.CreateSynthesizerRequest\x1a=.weave.datamanagement.synthesize.v1.CreateSynthesizerResponse" \x82\xd3\xe4\x93\x02\x1a"\x15/v1/synthesize/create:\x01*B\xb3\x02\n&com.weave.datamanagement.synthesize.v1B\x0cServiceProtoP\x01ZPgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/synthesize/v1;synthesize\xa2\x02\x03WDS\xaa\x02"Weave.Datamanagement.Synthesize.V1\xca\x02"Weave\\Datamanagement\\Synthesize\\V1\xe2\x02.Weave\\Datamanagement\\Synthesize\\V1\\GPBMetadata\xea\x02%Weave::Datamanagement::Synthesize::V1b\x06proto3'
 )
 
 _globals = globals()
@@ -43,9 +43,9 @@ if not _descriptor._USE_C_DESCRIPTORS:
         "CreateSynthesizer"
     ]._serialized_options = b'\202\323\344\223\002\032"\025/v1/synthesize/create:\001*'
     _globals["_CREATESYNTHESIZERREQUEST"]._serialized_start = 174
-    _globals["_CREATESYNTHESIZERREQUEST"]._serialized_end = 368
-    _globals["_CREATESYNTHESIZERRESPONSE"]._serialized_start = 370
-    _globals["_CREATESYNTHESIZERRESPONSE"]._serialized_end = 430
-    _globals["_SYNTHESIZE"]._serialized_start = 433
-    _globals["_SYNTHESIZE"]._serialized_end = 626
+    _globals["_CREATESYNTHESIZERREQUEST"]._serialized_end = 433
+    _globals["_CREATESYNTHESIZERRESPONSE"]._serialized_start = 435
+    _globals["_CREATESYNTHESIZERRESPONSE"]._serialized_end = 495
+    _globals["_SYNTHESIZE"]._serialized_start = 498
+    _globals["_SYNTHESIZE"]._serialized_end = 691
 # @@protoc_insertion_point(module_scope)

{weave_python-0.12.0 → weave_python-0.14.0}/weave/datamanagement/synthesize/v1/service_pb2.pyi

@@ -20,7 +20,9 @@ class CreateSynthesizerRequest(google.protobuf.message.Message):
     ORGANIZATION_ID_FIELD_NUMBER: builtins.int
     DATASET_IDS_FIELD_NUMBER: builtins.int
     RELATIONSHIPS_FIELD_NUMBER: builtins.int
+    OUTPUT_STORAGE_CONNECTION_ID_FIELD_NUMBER: builtins.int
     organization_id: builtins.str
+    output_storage_connection_id: builtins.str
     @property
     def dataset_ids(
         self,
@@ -42,6 +44,7 @@ class CreateSynthesizerRequest(google.protobuf.message.Message):
             weave.datamanagement.synthesize.v1.relationship_pb2.ColumnRelationship
         ]
         | None = ...,
+        output_storage_connection_id: builtins.str = ...,
     ) -> None: ...
     def ClearField(
         self,
@@ -50,6 +53,8 @@ class CreateSynthesizerRequest(google.protobuf.message.Message):
             b"dataset_ids",
             "organization_id",
             b"organization_id",
+            "output_storage_connection_id",
+            b"output_storage_connection_id",
             "relationships",
             b"relationships",
         ],

{weave_python-0.12.0 → weave_python-0.14.0}/weave/datamanagement/synthesize/v1/training_pb2.py

@@ -25,7 +25,7 @@ _sym_db = _symbol_database.Default()
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n1weave/datamanagement/synthesize/v1/training.proto\x12"weave.datamanagement.synthesize.v1"\xc7\x01\n\x12StartTrainingEvent\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1f\n\x0b\x64\x61taset_ids\x18\x02 \x03(\tR\ndatasetIds\x12[\n\x0ftraining_driver\x18\x03 \x01(\x0e\x32\x32.weave.datamanagement.synthesize.v1.TrainingDriverR\x0etrainingDriver\x12#\n\rdriver_config\x18\x04 \x01(\x0cR\x0c\x64riverConfig"\xaf\x01\n\x15TrainingProgressEvent\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x45\n\x06status\x18\x02 \x01(\x0e\x32-.weave.datamanagement.synthesize.v1.JobStatusR\x06status\x12#\n\rerror_message\x18\x03 \x01(\tR\x0c\x65rrorMessage\x12\x1a\n\x08progress\x18\x04 \x01(\x02R\x08progress*\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\xb4\x02\n&com.weave.datamanagement.synthesize.v1B\rTrainingProtoP\x01ZPgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/synthesize/v1;synthesize\xa2\x02\x03WDS\xaa\x02"Weave.Datamanagement.Synthesize.V1\xca\x02"Weave\\Datamanagement\\Synthesize\\V1\xe2\x02.Weave\\Datamanagement\\Synthesize\\V1\\GPBMetadata\xea\x02%Weave::Datamanagement::Synthesize::V1b\x06proto3'
+    b'\n1weave/datamanagement/synthesize/v1/training.proto\x12"weave.datamanagement.synthesize.v1"\x88\x02\n\x12StartTrainingEvent\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x1f\n\x0b\x64\x61taset_ids\x18\x02 \x03(\tR\ndatasetIds\x12[\n\x0ftraining_driver\x18\x03 \x01(\x0e\x32\x32.weave.datamanagement.synthesize.v1.TrainingDriverR\x0etrainingDriver\x12#\n\rdriver_config\x18\x04 \x01(\x0cR\x0c\x64riverConfig\x12?\n\x1coutput_storage_connection_id\x18\x05 \x01(\tR\x19outputStorageConnectionId"\x8d\x02\n\x15TrainingProgressEvent\x12L\n\njob_status\x18\x01 \x01(\x0e\x32-.weave.datamanagement.synthesize.v1.JobStatusR\tjobStatus\x12\x1b\n\tstep_name\x18\x02 \x01(\tR\x08stepName\x12N\n\x0bstep_status\x18\x03 \x01(\x0e\x32-.weave.datamanagement.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\xb4\x02\n&com.weave.datamanagement.synthesize.v1B\rTrainingProtoP\x01ZPgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/synthesize/v1;synthesize\xa2\x02\x03WDS\xaa\x02"Weave.Datamanagement.Synthesize.V1\xca\x02"Weave\\Datamanagement\\Synthesize\\V1\xe2\x02.Weave\\Datamanagement\\Synthesize\\V1\\GPBMetadata\xea\x02%Weave::Datamanagement::Synthesize::V1b\x06proto3'
 )
 
 _globals = globals()
@@ -38,12 +38,12 @@ if not _descriptor._USE_C_DESCRIPTORS:
     _globals[
         "DESCRIPTOR"
     ]._serialized_options = b'\n&com.weave.datamanagement.synthesize.v1B\rTrainingProtoP\001ZPgithub.com/weave-labs/weave-go/pkg/weave/datamanagement/synthesize/v1;synthesize\242\002\003WDS\252\002"Weave.Datamanagement.Synthesize.V1\312\002"Weave\\Datamanagement\\Synthesize\\V1\342\002.Weave\\Datamanagement\\Synthesize\\V1\\GPBMetadata\352\002%Weave::Datamanagement::Synthesize::V1'
-    _globals["_TRAININGDRIVER"]._serialized_start = 470
-    _globals["_TRAININGDRIVER"]._serialized_end = 606
-    _globals["_JOBSTATUS"]._serialized_start = 609
-    _globals["_JOBSTATUS"]._serialized_end = 745
+    _globals["_TRAININGDRIVER"]._serialized_start = 629
+    _globals["_TRAININGDRIVER"]._serialized_end = 765
+    _globals["_JOBSTATUS"]._serialized_start = 768
+    _globals["_JOBSTATUS"]._serialized_end = 904
     _globals["_STARTTRAININGEVENT"]._serialized_start = 90
-    _globals["_STARTTRAININGEVENT"]._serialized_end = 289
-    _globals["_TRAININGPROGRESSEVENT"]._serialized_start = 292
-    _globals["_TRAININGPROGRESSEVENT"]._serialized_end = 467
+    _globals["_STARTTRAININGEVENT"]._serialized_end = 354
+    _globals["_TRAININGPROGRESSEVENT"]._serialized_start = 357
+    _globals["_TRAININGPROGRESSEVENT"]._serialized_end = 626
 # @@protoc_insertion_point(module_scope)

{weave_python-0.12.0 → weave_python-0.14.0}/weave/datamanagement/synthesize/v1/training_pb2.pyi

@@ -75,9 +75,11 @@ class StartTrainingEvent(google.protobuf.message.Message):
     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
     id: builtins.str
     training_driver: global___TrainingDriver.ValueType
     driver_config: builtins.bytes
+    output_storage_connection_id: builtins.str
     @property
     def dataset_ids(
         self,
@@ -91,6 +93,7 @@ class StartTrainingEvent(google.protobuf.message.Message):
         dataset_ids: collections.abc.Iterable[builtins.str] | None = ...,
         training_driver: global___TrainingDriver.ValueType = ...,
         driver_config: builtins.bytes = ...,
+        output_storage_connection_id: builtins.str = ...,
     ) -> None: ...
     def ClearField(
         self,
@@ -101,6 +104,8 @@ class StartTrainingEvent(google.protobuf.message.Message):
             b"driver_config",
             "id",
             b"id",
+            "output_storage_connection_id",
+            b"output_storage_connection_id",
             "training_driver",
             b"training_driver",
         ],
@@ -112,33 +117,38 @@ global___StartTrainingEvent = StartTrainingEvent
 class TrainingProgressEvent(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
 
-    ID_FIELD_NUMBER: builtins.int
-    STATUS_FIELD_NUMBER: builtins.int
-    ERROR_MESSAGE_FIELD_NUMBER: builtins.int
-    PROGRESS_FIELD_NUMBER: builtins.int
-    id: builtins.str
-    status: global___JobStatus.ValueType
-    error_message: builtins.str
-    progress: builtins.float
+    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,
         *,
-        id: builtins.str = ...,
-        status: global___JobStatus.ValueType = ...,
-        error_message: builtins.str = ...,
-        progress: builtins.float = ...,
+        job_status: global___JobStatus.ValueType = ...,
+        step_name: builtins.str = ...,
+        step_status: global___JobStatus.ValueType = ...,
+        step_progress: builtins.float = ...,
+        error: builtins.str = ...,
     ) -> None: ...
     def ClearField(
         self,
         field_name: typing.Literal[
-            "error_message",
-            b"error_message",
-            "id",
-            b"id",
-            "progress",
-            b"progress",
-            "status",
-            b"status",
+            "error",
+            b"error",
+            "job_status",
+            b"job_status",
+            "step_name",
+            b"step_name",
+            "step_progress",
+            b"step_progress",
+            "step_status",
+            b"step_status",
         ],
     ) -> None: ...