weave-python 0.28.1__py3-none-any.whl → 0.30.3__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- weave/weaveapi/auth/v1/service_pb2.py +194 -98
- weave/weaveapi/auth/v1/service_pb2.pyi +1288 -1134
- weave/weaveapi/auth/v1/service_pb2_grpc.pyi +248 -335
- weave/weaveapi/auth/v1/session_pb2.py +14 -6
- weave/weaveapi/auth/v1/session_pb2.pyi +109 -89
- weave/weaveapi/auth/v1/session_pb2_grpc.pyi +11 -10
- weave/weaveapi/auth/v1/usage_pb2.py +18 -14
- weave/weaveapi/auth/v1/usage_pb2.pyi +251 -197
- weave/weaveapi/auth/v1/usage_pb2_grpc.pyi +11 -10
- weave/weaveapi/auth/v1/user_pb2.py +60 -18
- weave/weaveapi/auth/v1/user_pb2.pyi +588 -505
- weave/weaveapi/auth/v1/user_pb2_grpc.pyi +11 -10
- weave/weaveapi/generate/v1/configuration_pb2.py +31 -19
- weave/weaveapi/generate/v1/configuration_pb2.pyi +277 -241
- weave/weaveapi/generate/v1/configuration_pb2_grpc.pyi +11 -10
- weave/weaveapi/generate/v1/generate_pb2.py +32 -6
- weave/weaveapi/generate/v1/generate_pb2.pyi +138 -40
- weave/weaveapi/generate/v1/generate_pb2_grpc.pyi +11 -10
- weave/weaveapi/generate/v1/service_pb2.py +52 -22
- weave/weaveapi/generate/v1/service_pb2.pyi +264 -135
- weave/weaveapi/generate/v1/service_pb2_grpc.py +47 -0
- weave/weaveapi/generate/v1/service_pb2_grpc.pyi +89 -78
- weave/weaveapi/llmx/v1/architecture_pb2.py +42 -42
- weave/weaveapi/llmx/v1/architecture_pb2.pyi +664 -555
- weave/weaveapi/llmx/v1/architecture_pb2_grpc.pyi +11 -10
- weave/weaveapi/llmx/v1/capabilities_pb2.py +88 -56
- weave/weaveapi/llmx/v1/capabilities_pb2.pyi +967 -852
- weave/weaveapi/llmx/v1/capabilities_pb2_grpc.pyi +11 -10
- weave/weaveapi/llmx/v1/model_pb2.py +56 -20
- weave/weaveapi/llmx/v1/model_pb2.pyi +533 -455
- weave/weaveapi/llmx/v1/model_pb2_grpc.pyi +11 -10
- weave/weaveapi/llmx/v1/pricing_pb2.py +24 -20
- weave/weaveapi/llmx/v1/pricing_pb2.pyi +194 -172
- weave/weaveapi/llmx/v1/pricing_pb2_grpc.pyi +11 -10
- weave/weaveapi/llmx/v1/provider_pb2.py +6 -6
- weave/weaveapi/llmx/v1/provider_pb2.pyi +84 -59
- weave/weaveapi/llmx/v1/provider_pb2_grpc.pyi +11 -10
- weave/weaveapi/llmx/v1/service_pb2.py +228 -90
- weave/weaveapi/llmx/v1/service_pb2.pyi +1651 -1403
- weave/weaveapi/llmx/v1/service_pb2_grpc.pyi +160 -203
- weave/weaveapi/mcpregistry/v1/server_pb2.py +16 -10
- weave/weaveapi/mcpregistry/v1/server_pb2.pyi +143 -121
- weave/weaveapi/mcpregistry/v1/server_pb2_grpc.pyi +11 -10
- weave/weaveapi/mcpregistry/v1/service_pb2.py +51 -31
- weave/weaveapi/mcpregistry/v1/service_pb2.pyi +132 -122
- weave/weaveapi/mcpregistry/v1/service_pb2_grpc.pyi +87 -112
- weave/weaveapi/payment/v1/invoice_pb2.py +38 -12
- weave/weaveapi/payment/v1/invoice_pb2.pyi +352 -291
- weave/weaveapi/payment/v1/invoice_pb2_grpc.pyi +11 -10
- weave/weaveapi/payment/v1/service_pb2.py +264 -98
- weave/weaveapi/payment/v1/service_pb2.pyi +1381 -1242
- weave/weaveapi/payment/v1/service_pb2_grpc.pyi +229 -319
- weave/weaveapi/payment/v1/subscription_pb2.py +99 -23
- weave/weaveapi/payment/v1/subscription_pb2.pyi +727 -611
- weave/weaveapi/payment/v1/subscription_pb2_grpc.pyi +11 -10
- weave/weaveapi/storage/v1/auth_pb2.py +6 -6
- weave/weaveapi/storage/v1/auth_pb2.pyi +42 -29
- weave/weaveapi/storage/v1/auth_pb2_grpc.pyi +11 -10
- weave/weaveapi/storage/v1/nosql_database_pb2.py +47 -23
- weave/weaveapi/storage/v1/nosql_database_pb2.pyi +438 -372
- weave/weaveapi/storage/v1/nosql_database_pb2_grpc.pyi +11 -10
- weave/weaveapi/storage/v1/object_store_pb2.py +27 -13
- weave/weaveapi/storage/v1/object_store_pb2.pyi +203 -187
- weave/weaveapi/storage/v1/object_store_pb2_grpc.pyi +11 -10
- weave/weaveapi/storage/v1/service_pb2.py +96 -36
- weave/weaveapi/storage/v1/service_pb2.pyi +414 -357
- weave/weaveapi/storage/v1/service_pb2_grpc.pyi +88 -107
- weave/weaveapi/storage/v1/sql_database_pb2.py +39 -23
- weave/weaveapi/storage/v1/sql_database_pb2.pyi +481 -400
- weave/weaveapi/storage/v1/sql_database_pb2_grpc.pyi +11 -10
- weave/weaveapi/storage/v1/storage_pb2.py +20 -6
- weave/weaveapi/storage/v1/storage_pb2.pyi +79 -70
- weave/weaveapi/storage/v1/storage_pb2_grpc.pyi +11 -10
- weave/weaveapi/synthesize/v1/dataset_pb2.py +14 -12
- weave/weaveapi/synthesize/v1/dataset_pb2.pyi +158 -128
- weave/weaveapi/synthesize/v1/dataset_pb2_grpc.pyi +11 -10
- weave/weaveapi/synthesize/v1/inline_data_pb2.py +9 -9
- weave/weaveapi/synthesize/v1/inline_data_pb2.pyi +31 -27
- weave/weaveapi/synthesize/v1/inline_data_pb2_grpc.pyi +11 -10
- weave/weaveapi/synthesize/v1/relationship_pb2.py +19 -11
- weave/weaveapi/synthesize/v1/relationship_pb2.pyi +67 -64
- weave/weaveapi/synthesize/v1/relationship_pb2_grpc.pyi +11 -10
- weave/weaveapi/synthesize/v1/service_pb2.py +42 -24
- weave/weaveapi/synthesize/v1/service_pb2.pyi +202 -168
- weave/weaveapi/synthesize/v1/service_pb2_grpc.pyi +67 -79
- weave/weaveapi/synthesize/v1/training_pb2.py +19 -13
- weave/weaveapi/synthesize/v1/training_pb2.pyi +119 -106
- weave/weaveapi/synthesize/v1/training_pb2_grpc.pyi +11 -10
- weave/weavesql/llmxdb/capabilities.py +487 -0
- weave/weavesql/llmxdb/changes.py +297 -0
- weave/weavesql/llmxdb/models.py +594 -0
- weave/weavesql/llmxdb/providers.py +348 -0
- weave/weavesql/llmxdb/scraper_runs.py +287 -0
- weave/weavesql/llmxdb/search.py +721 -0
- weave/weavesql/weavedb/dataset.py +75 -0
- weave/weavesql/weavedb/models.py +135 -0
- weave/weavesql/weavedb/relationships.py +72 -0
- weave/weavesql/weavedb/storage.py +113 -0
- weave/weavesql/weavedb/synthesizer.py +107 -0
- weave_python-0.30.3.dist-info/METADATA +52 -0
- weave_python-0.30.3.dist-info/RECORD +131 -0
- {weave_python-0.28.1.dist-info → weave_python-0.30.3.dist-info}/WHEEL +1 -1
- weave_python-0.30.3.dist-info/licenses/LICENSE +120 -0
- weave_python-0.28.1.dist-info/METADATA +0 -6
- weave_python-0.28.1.dist-info/RECORD +0 -119
|
@@ -3,31 +3,30 @@
|
|
|
3
3
|
isort:skip_file
|
|
4
4
|
"""
|
|
5
5
|
|
|
6
|
-
import
|
|
7
|
-
import
|
|
8
|
-
|
|
9
|
-
|
|
10
|
-
|
|
11
|
-
import
|
|
6
|
+
from collections import abc as _abc
|
|
7
|
+
from google.protobuf import descriptor as _descriptor
|
|
8
|
+
from google.protobuf import message as _message
|
|
9
|
+
from google.protobuf.internal import containers as _containers
|
|
10
|
+
from google.protobuf.internal import enum_type_wrapper as _enum_type_wrapper
|
|
11
|
+
import builtins as _builtins
|
|
12
12
|
import sys
|
|
13
|
-
import typing
|
|
13
|
+
import typing as _typing
|
|
14
14
|
|
|
15
15
|
if sys.version_info >= (3, 10):
|
|
16
|
-
|
|
16
|
+
from typing import TypeAlias as _TypeAlias
|
|
17
17
|
else:
|
|
18
|
-
import
|
|
18
|
+
from typing_extensions import TypeAlias as _TypeAlias
|
|
19
19
|
|
|
20
|
-
DESCRIPTOR:
|
|
20
|
+
DESCRIPTOR: _descriptor.FileDescriptor
|
|
21
21
|
|
|
22
22
|
class _ModelType:
|
|
23
|
-
ValueType =
|
|
24
|
-
V:
|
|
23
|
+
ValueType = _typing.NewType("ValueType", _builtins.int)
|
|
24
|
+
V: _TypeAlias = ValueType # noqa: Y015
|
|
25
25
|
|
|
26
26
|
class _ModelTypeEnumTypeWrapper(
|
|
27
|
-
|
|
28
|
-
builtins.type,
|
|
27
|
+
_enum_type_wrapper._EnumTypeWrapper[_ModelType.ValueType], _builtins.type
|
|
29
28
|
):
|
|
30
|
-
DESCRIPTOR:
|
|
29
|
+
DESCRIPTOR: _descriptor.EnumDescriptor
|
|
31
30
|
MODEL_TYPE_UNSPECIFIED: _ModelType.ValueType # 0
|
|
32
31
|
MODEL_TYPE_CTGAN: _ModelType.ValueType # 1
|
|
33
32
|
"""Conditional Tabular GAN"""
|
|
@@ -70,301 +69,338 @@ MODEL_TYPE_BAYESIAN: ModelType.ValueType # 8
|
|
|
70
69
|
"""Bayesian Network"""
|
|
71
70
|
MODEL_TYPE_HMA: ModelType.ValueType # 9
|
|
72
71
|
"""Hierarchical Modeling Algorithm"""
|
|
73
|
-
|
|
72
|
+
Global___ModelType: _TypeAlias = ModelType # noqa: Y015
|
|
74
73
|
|
|
75
|
-
@
|
|
76
|
-
class Configuration(
|
|
74
|
+
@_typing.final
|
|
75
|
+
class Configuration(_message.Message):
|
|
77
76
|
"""SynthesizerConfig defines customization options for synthetic data generation"""
|
|
78
77
|
|
|
79
|
-
DESCRIPTOR:
|
|
78
|
+
DESCRIPTOR: _descriptor.Descriptor
|
|
80
79
|
|
|
81
|
-
NUM_RECORDS_FIELD_NUMBER:
|
|
82
|
-
DATA_QUALITY_FIELD_NUMBER:
|
|
83
|
-
PRIVACY_FIELD_NUMBER:
|
|
84
|
-
MODEL_FIELD_NUMBER:
|
|
85
|
-
num_records:
|
|
80
|
+
NUM_RECORDS_FIELD_NUMBER: _builtins.int
|
|
81
|
+
DATA_QUALITY_FIELD_NUMBER: _builtins.int
|
|
82
|
+
PRIVACY_FIELD_NUMBER: _builtins.int
|
|
83
|
+
MODEL_FIELD_NUMBER: _builtins.int
|
|
84
|
+
num_records: _builtins.int
|
|
86
85
|
"""Basic configuration
|
|
87
86
|
Number of synthetic records to generate
|
|
88
87
|
"""
|
|
89
|
-
@property
|
|
90
|
-
def data_quality(self) ->
|
|
91
|
-
"""Data quality settings
|
|
92
|
-
|
|
93
|
-
|
|
94
|
-
|
|
95
|
-
|
|
96
|
-
|
|
97
|
-
|
|
98
|
-
|
|
99
|
-
"""
|
|
100
|
-
|
|
101
|
-
@property
|
|
102
|
-
def model(self) -> global___ModelConfig:
|
|
103
|
-
"""Advanced model configuration
|
|
104
|
-
ML model configuration options
|
|
105
|
-
"""
|
|
88
|
+
@_builtins.property
|
|
89
|
+
def data_quality(self) -> Global___DataQualityConfig:
|
|
90
|
+
"""Data quality settings"""
|
|
91
|
+
|
|
92
|
+
@_builtins.property
|
|
93
|
+
def privacy(self) -> Global___PrivacyConfig:
|
|
94
|
+
"""Privacy settings"""
|
|
95
|
+
|
|
96
|
+
@_builtins.property
|
|
97
|
+
def model(self) -> Global___ModelConfig:
|
|
98
|
+
"""Advanced model configuration"""
|
|
106
99
|
|
|
107
100
|
def __init__(
|
|
108
101
|
self,
|
|
109
102
|
*,
|
|
110
|
-
num_records:
|
|
111
|
-
data_quality:
|
|
112
|
-
privacy:
|
|
113
|
-
model:
|
|
103
|
+
num_records: _builtins.int | None = ...,
|
|
104
|
+
data_quality: Global___DataQualityConfig | None = ...,
|
|
105
|
+
privacy: Global___PrivacyConfig | None = ...,
|
|
106
|
+
model: Global___ModelConfig | None = ...,
|
|
114
107
|
) -> None: ...
|
|
115
|
-
|
|
116
|
-
|
|
117
|
-
|
|
118
|
-
|
|
119
|
-
|
|
120
|
-
|
|
121
|
-
|
|
122
|
-
|
|
123
|
-
|
|
124
|
-
|
|
125
|
-
|
|
126
|
-
|
|
127
|
-
|
|
128
|
-
|
|
129
|
-
|
|
130
|
-
|
|
131
|
-
|
|
132
|
-
|
|
133
|
-
|
|
134
|
-
|
|
135
|
-
|
|
136
|
-
|
|
137
|
-
|
|
138
|
-
|
|
108
|
+
_HasFieldArgType: _TypeAlias = _typing.Literal[
|
|
109
|
+
"data_quality",
|
|
110
|
+
b"data_quality",
|
|
111
|
+
"model",
|
|
112
|
+
b"model",
|
|
113
|
+
"num_records",
|
|
114
|
+
b"num_records",
|
|
115
|
+
"privacy",
|
|
116
|
+
b"privacy",
|
|
117
|
+
] # noqa: Y015
|
|
118
|
+
def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
|
|
119
|
+
_ClearFieldArgType: _TypeAlias = _typing.Literal[
|
|
120
|
+
"data_quality",
|
|
121
|
+
b"data_quality",
|
|
122
|
+
"model",
|
|
123
|
+
b"model",
|
|
124
|
+
"num_records",
|
|
125
|
+
b"num_records",
|
|
126
|
+
"privacy",
|
|
127
|
+
b"privacy",
|
|
128
|
+
] # noqa: Y015
|
|
129
|
+
def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
|
|
130
|
+
|
|
131
|
+
Global___Configuration: _TypeAlias = Configuration # noqa: Y015
|
|
132
|
+
|
|
133
|
+
@_typing.final
|
|
134
|
+
class DataQualityConfig(_message.Message):
|
|
139
135
|
"""DataQualityConfig controls the quality of generated data"""
|
|
140
136
|
|
|
141
|
-
DESCRIPTOR:
|
|
137
|
+
DESCRIPTOR: _descriptor.Descriptor
|
|
142
138
|
|
|
143
|
-
CORRELATION_ACCURACY_FIELD_NUMBER:
|
|
144
|
-
STATISTICAL_SIMILARITY_FIELD_NUMBER:
|
|
145
|
-
PRESERVE_OUTLIERS_FIELD_NUMBER:
|
|
146
|
-
ENFORCE_CONSTRAINTS_FIELD_NUMBER:
|
|
147
|
-
CONSTRAINTS_FIELD_NUMBER:
|
|
148
|
-
correlation_accuracy:
|
|
139
|
+
CORRELATION_ACCURACY_FIELD_NUMBER: _builtins.int
|
|
140
|
+
STATISTICAL_SIMILARITY_FIELD_NUMBER: _builtins.int
|
|
141
|
+
PRESERVE_OUTLIERS_FIELD_NUMBER: _builtins.int
|
|
142
|
+
ENFORCE_CONSTRAINTS_FIELD_NUMBER: _builtins.int
|
|
143
|
+
CONSTRAINTS_FIELD_NUMBER: _builtins.int
|
|
144
|
+
correlation_accuracy: _builtins.float
|
|
149
145
|
"""How closely to maintain correlations (0.0-1.0)"""
|
|
150
|
-
statistical_similarity:
|
|
146
|
+
statistical_similarity: _builtins.float
|
|
151
147
|
"""Statistical similarity to original data (0.0-1.0)"""
|
|
152
|
-
preserve_outliers:
|
|
148
|
+
preserve_outliers: _builtins.bool
|
|
153
149
|
"""Whether to preserve outlier patterns"""
|
|
154
|
-
enforce_constraints:
|
|
150
|
+
enforce_constraints: _builtins.bool
|
|
155
151
|
"""Whether to enforce data constraints"""
|
|
156
|
-
@property
|
|
152
|
+
@_builtins.property
|
|
157
153
|
def constraints(
|
|
158
154
|
self,
|
|
159
|
-
) ->
|
|
160
|
-
global___ColumnConstraint
|
|
161
|
-
]:
|
|
162
|
-
"""Column-specific constraints"""
|
|
163
|
-
|
|
155
|
+
) -> _containers.RepeatedCompositeFieldContainer[Global___ColumnConstraint]: ...
|
|
164
156
|
def __init__(
|
|
165
157
|
self,
|
|
166
158
|
*,
|
|
167
|
-
correlation_accuracy:
|
|
168
|
-
statistical_similarity:
|
|
169
|
-
preserve_outliers:
|
|
170
|
-
enforce_constraints:
|
|
171
|
-
constraints:
|
|
172
|
-
) -> None: ...
|
|
173
|
-
def ClearField(
|
|
174
|
-
self,
|
|
175
|
-
field_name: typing.Literal[
|
|
176
|
-
"constraints",
|
|
177
|
-
b"constraints",
|
|
178
|
-
"correlation_accuracy",
|
|
179
|
-
b"correlation_accuracy",
|
|
180
|
-
"enforce_constraints",
|
|
181
|
-
b"enforce_constraints",
|
|
182
|
-
"preserve_outliers",
|
|
183
|
-
b"preserve_outliers",
|
|
184
|
-
"statistical_similarity",
|
|
185
|
-
b"statistical_similarity",
|
|
186
|
-
],
|
|
159
|
+
correlation_accuracy: _builtins.float | None = ...,
|
|
160
|
+
statistical_similarity: _builtins.float | None = ...,
|
|
161
|
+
preserve_outliers: _builtins.bool | None = ...,
|
|
162
|
+
enforce_constraints: _builtins.bool | None = ...,
|
|
163
|
+
constraints: _abc.Iterable[Global___ColumnConstraint] | None = ...,
|
|
187
164
|
) -> None: ...
|
|
188
|
-
|
|
189
|
-
|
|
190
|
-
|
|
191
|
-
|
|
192
|
-
|
|
165
|
+
_HasFieldArgType: _TypeAlias = _typing.Literal[
|
|
166
|
+
"constraints",
|
|
167
|
+
b"constraints",
|
|
168
|
+
"correlation_accuracy",
|
|
169
|
+
b"correlation_accuracy",
|
|
170
|
+
"enforce_constraints",
|
|
171
|
+
b"enforce_constraints",
|
|
172
|
+
"preserve_outliers",
|
|
173
|
+
b"preserve_outliers",
|
|
174
|
+
"statistical_similarity",
|
|
175
|
+
b"statistical_similarity",
|
|
176
|
+
] # noqa: Y015
|
|
177
|
+
def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
|
|
178
|
+
_ClearFieldArgType: _TypeAlias = _typing.Literal[
|
|
179
|
+
"constraints",
|
|
180
|
+
b"constraints",
|
|
181
|
+
"correlation_accuracy",
|
|
182
|
+
b"correlation_accuracy",
|
|
183
|
+
"enforce_constraints",
|
|
184
|
+
b"enforce_constraints",
|
|
185
|
+
"preserve_outliers",
|
|
186
|
+
b"preserve_outliers",
|
|
187
|
+
"statistical_similarity",
|
|
188
|
+
b"statistical_similarity",
|
|
189
|
+
] # noqa: Y015
|
|
190
|
+
def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
|
|
191
|
+
|
|
192
|
+
Global___DataQualityConfig: _TypeAlias = DataQualityConfig # noqa: Y015
|
|
193
|
+
|
|
194
|
+
@_typing.final
|
|
195
|
+
class ColumnConstraint(_message.Message):
|
|
193
196
|
"""ColumnConstraint defines constraints for specific columns"""
|
|
194
197
|
|
|
195
|
-
DESCRIPTOR:
|
|
198
|
+
DESCRIPTOR: _descriptor.Descriptor
|
|
196
199
|
|
|
197
|
-
COLUMN_NAME_FIELD_NUMBER:
|
|
198
|
-
CONSTRAINT_TYPE_FIELD_NUMBER:
|
|
199
|
-
MIN_VALUE_FIELD_NUMBER:
|
|
200
|
-
MAX_VALUE_FIELD_NUMBER:
|
|
201
|
-
REGEX_PATTERN_FIELD_NUMBER:
|
|
202
|
-
ALLOWED_VALUES_FIELD_NUMBER:
|
|
203
|
-
column_name:
|
|
200
|
+
COLUMN_NAME_FIELD_NUMBER: _builtins.int
|
|
201
|
+
CONSTRAINT_TYPE_FIELD_NUMBER: _builtins.int
|
|
202
|
+
MIN_VALUE_FIELD_NUMBER: _builtins.int
|
|
203
|
+
MAX_VALUE_FIELD_NUMBER: _builtins.int
|
|
204
|
+
REGEX_PATTERN_FIELD_NUMBER: _builtins.int
|
|
205
|
+
ALLOWED_VALUES_FIELD_NUMBER: _builtins.int
|
|
206
|
+
column_name: _builtins.str
|
|
204
207
|
"""Name of the column"""
|
|
205
|
-
constraint_type:
|
|
208
|
+
constraint_type: _builtins.str
|
|
206
209
|
"""Type of constraint (e.g., "range", "regex", "categorical")"""
|
|
207
|
-
min_value:
|
|
210
|
+
min_value: _builtins.str
|
|
208
211
|
"""Minimum value (for numeric constraints)"""
|
|
209
|
-
max_value:
|
|
212
|
+
max_value: _builtins.str
|
|
210
213
|
"""Maximum value (for numeric constraints)"""
|
|
211
|
-
regex_pattern:
|
|
214
|
+
regex_pattern: _builtins.str
|
|
212
215
|
"""Regex pattern (for string constraints)"""
|
|
213
|
-
@property
|
|
214
|
-
def allowed_values(
|
|
215
|
-
self,
|
|
216
|
-
) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]:
|
|
216
|
+
@_builtins.property
|
|
217
|
+
def allowed_values(self) -> _containers.RepeatedScalarFieldContainer[_builtins.str]:
|
|
217
218
|
"""List of allowed values (for categorical constraints)"""
|
|
218
219
|
|
|
219
220
|
def __init__(
|
|
220
221
|
self,
|
|
221
222
|
*,
|
|
222
|
-
column_name:
|
|
223
|
-
constraint_type:
|
|
224
|
-
min_value:
|
|
225
|
-
max_value:
|
|
226
|
-
regex_pattern:
|
|
227
|
-
allowed_values:
|
|
223
|
+
column_name: _builtins.str | None = ...,
|
|
224
|
+
constraint_type: _builtins.str | None = ...,
|
|
225
|
+
min_value: _builtins.str | None = ...,
|
|
226
|
+
max_value: _builtins.str | None = ...,
|
|
227
|
+
regex_pattern: _builtins.str | None = ...,
|
|
228
|
+
allowed_values: _abc.Iterable[_builtins.str] | None = ...,
|
|
228
229
|
) -> None: ...
|
|
229
|
-
|
|
230
|
-
|
|
231
|
-
|
|
232
|
-
|
|
233
|
-
|
|
234
|
-
|
|
235
|
-
|
|
236
|
-
|
|
237
|
-
|
|
238
|
-
|
|
239
|
-
|
|
240
|
-
|
|
241
|
-
|
|
242
|
-
|
|
243
|
-
|
|
244
|
-
|
|
245
|
-
|
|
246
|
-
|
|
247
|
-
|
|
248
|
-
|
|
249
|
-
|
|
250
|
-
|
|
230
|
+
_HasFieldArgType: _TypeAlias = _typing.Literal[
|
|
231
|
+
"allowed_values",
|
|
232
|
+
b"allowed_values",
|
|
233
|
+
"column_name",
|
|
234
|
+
b"column_name",
|
|
235
|
+
"constraint_type",
|
|
236
|
+
b"constraint_type",
|
|
237
|
+
"max_value",
|
|
238
|
+
b"max_value",
|
|
239
|
+
"min_value",
|
|
240
|
+
b"min_value",
|
|
241
|
+
"regex_pattern",
|
|
242
|
+
b"regex_pattern",
|
|
243
|
+
] # noqa: Y015
|
|
244
|
+
def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
|
|
245
|
+
_ClearFieldArgType: _TypeAlias = _typing.Literal[
|
|
246
|
+
"allowed_values",
|
|
247
|
+
b"allowed_values",
|
|
248
|
+
"column_name",
|
|
249
|
+
b"column_name",
|
|
250
|
+
"constraint_type",
|
|
251
|
+
b"constraint_type",
|
|
252
|
+
"max_value",
|
|
253
|
+
b"max_value",
|
|
254
|
+
"min_value",
|
|
255
|
+
b"min_value",
|
|
256
|
+
"regex_pattern",
|
|
257
|
+
b"regex_pattern",
|
|
258
|
+
] # noqa: Y015
|
|
259
|
+
def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
|
|
260
|
+
|
|
261
|
+
Global___ColumnConstraint: _TypeAlias = ColumnConstraint # noqa: Y015
|
|
262
|
+
|
|
263
|
+
@_typing.final
|
|
264
|
+
class PrivacyConfig(_message.Message):
|
|
251
265
|
"""PrivacyConfig controls privacy aspects of synthetic data"""
|
|
252
266
|
|
|
253
|
-
DESCRIPTOR:
|
|
267
|
+
DESCRIPTOR: _descriptor.Descriptor
|
|
254
268
|
|
|
255
|
-
EPSILON_FIELD_NUMBER:
|
|
256
|
-
K_ANONYMITY_CHECK_FIELD_NUMBER:
|
|
257
|
-
K_VALUE_FIELD_NUMBER:
|
|
258
|
-
SENSITIVE_COLUMNS_FIELD_NUMBER:
|
|
259
|
-
ENABLE_PRIVACY_METRICS_FIELD_NUMBER:
|
|
260
|
-
epsilon:
|
|
269
|
+
EPSILON_FIELD_NUMBER: _builtins.int
|
|
270
|
+
K_ANONYMITY_CHECK_FIELD_NUMBER: _builtins.int
|
|
271
|
+
K_VALUE_FIELD_NUMBER: _builtins.int
|
|
272
|
+
SENSITIVE_COLUMNS_FIELD_NUMBER: _builtins.int
|
|
273
|
+
ENABLE_PRIVACY_METRICS_FIELD_NUMBER: _builtins.int
|
|
274
|
+
epsilon: _builtins.float
|
|
261
275
|
"""Differential privacy epsilon parameter"""
|
|
262
|
-
k_anonymity_check:
|
|
276
|
+
k_anonymity_check: _builtins.bool
|
|
263
277
|
"""Whether to enforce k-anonymity"""
|
|
264
|
-
k_value:
|
|
278
|
+
k_value: _builtins.int
|
|
265
279
|
"""k value for k-anonymity (if enabled)"""
|
|
266
|
-
enable_privacy_metrics:
|
|
280
|
+
enable_privacy_metrics: _builtins.bool
|
|
267
281
|
"""Whether to compute privacy metrics"""
|
|
268
|
-
@property
|
|
282
|
+
@_builtins.property
|
|
269
283
|
def sensitive_columns(
|
|
270
284
|
self,
|
|
271
|
-
) ->
|
|
285
|
+
) -> _containers.RepeatedScalarFieldContainer[_builtins.str]:
|
|
272
286
|
"""Columns that contain sensitive information"""
|
|
273
287
|
|
|
274
288
|
def __init__(
|
|
275
289
|
self,
|
|
276
290
|
*,
|
|
277
|
-
epsilon:
|
|
278
|
-
k_anonymity_check:
|
|
279
|
-
k_value:
|
|
280
|
-
sensitive_columns:
|
|
281
|
-
enable_privacy_metrics:
|
|
291
|
+
epsilon: _builtins.float | None = ...,
|
|
292
|
+
k_anonymity_check: _builtins.bool | None = ...,
|
|
293
|
+
k_value: _builtins.int | None = ...,
|
|
294
|
+
sensitive_columns: _abc.Iterable[_builtins.str] | None = ...,
|
|
295
|
+
enable_privacy_metrics: _builtins.bool | None = ...,
|
|
282
296
|
) -> None: ...
|
|
283
|
-
|
|
284
|
-
|
|
285
|
-
|
|
286
|
-
|
|
287
|
-
|
|
288
|
-
|
|
289
|
-
|
|
290
|
-
|
|
291
|
-
|
|
292
|
-
|
|
293
|
-
|
|
294
|
-
|
|
295
|
-
|
|
296
|
-
|
|
297
|
-
|
|
298
|
-
|
|
299
|
-
|
|
300
|
-
|
|
301
|
-
|
|
302
|
-
|
|
297
|
+
_HasFieldArgType: _TypeAlias = _typing.Literal[
|
|
298
|
+
"enable_privacy_metrics",
|
|
299
|
+
b"enable_privacy_metrics",
|
|
300
|
+
"epsilon",
|
|
301
|
+
b"epsilon",
|
|
302
|
+
"k_anonymity_check",
|
|
303
|
+
b"k_anonymity_check",
|
|
304
|
+
"k_value",
|
|
305
|
+
b"k_value",
|
|
306
|
+
"sensitive_columns",
|
|
307
|
+
b"sensitive_columns",
|
|
308
|
+
] # noqa: Y015
|
|
309
|
+
def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
|
|
310
|
+
_ClearFieldArgType: _TypeAlias = _typing.Literal[
|
|
311
|
+
"enable_privacy_metrics",
|
|
312
|
+
b"enable_privacy_metrics",
|
|
313
|
+
"epsilon",
|
|
314
|
+
b"epsilon",
|
|
315
|
+
"k_anonymity_check",
|
|
316
|
+
b"k_anonymity_check",
|
|
317
|
+
"k_value",
|
|
318
|
+
b"k_value",
|
|
319
|
+
"sensitive_columns",
|
|
320
|
+
b"sensitive_columns",
|
|
321
|
+
] # noqa: Y015
|
|
322
|
+
def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
|
|
323
|
+
|
|
324
|
+
Global___PrivacyConfig: _TypeAlias = PrivacyConfig # noqa: Y015
|
|
325
|
+
|
|
326
|
+
@_typing.final
|
|
327
|
+
class ModelConfig(_message.Message):
|
|
303
328
|
"""ModelConfig controls the ML model used for synthesis"""
|
|
304
329
|
|
|
305
|
-
DESCRIPTOR:
|
|
330
|
+
DESCRIPTOR: _descriptor.Descriptor
|
|
306
331
|
|
|
307
|
-
@
|
|
308
|
-
class CustomParametersEntry(
|
|
309
|
-
DESCRIPTOR:
|
|
332
|
+
@_typing.final
|
|
333
|
+
class CustomParametersEntry(_message.Message):
|
|
334
|
+
DESCRIPTOR: _descriptor.Descriptor
|
|
310
335
|
|
|
311
|
-
KEY_FIELD_NUMBER:
|
|
312
|
-
VALUE_FIELD_NUMBER:
|
|
313
|
-
key:
|
|
314
|
-
value:
|
|
336
|
+
KEY_FIELD_NUMBER: _builtins.int
|
|
337
|
+
VALUE_FIELD_NUMBER: _builtins.int
|
|
338
|
+
key: _builtins.str
|
|
339
|
+
value: _builtins.str
|
|
315
340
|
def __init__(
|
|
316
341
|
self,
|
|
317
342
|
*,
|
|
318
|
-
key:
|
|
319
|
-
value:
|
|
343
|
+
key: _builtins.str | None = ...,
|
|
344
|
+
value: _builtins.str | None = ...,
|
|
320
345
|
) -> None: ...
|
|
321
|
-
|
|
322
|
-
|
|
323
|
-
|
|
324
|
-
|
|
325
|
-
|
|
326
|
-
|
|
327
|
-
|
|
328
|
-
|
|
329
|
-
|
|
330
|
-
|
|
346
|
+
_HasFieldArgType: _TypeAlias = _typing.Literal["key", b"key", "value", b"value"] # noqa: Y015
|
|
347
|
+
def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
|
|
348
|
+
_ClearFieldArgType: _TypeAlias = _typing.Literal[
|
|
349
|
+
"key", b"key", "value", b"value"
|
|
350
|
+
] # noqa: Y015
|
|
351
|
+
def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
|
|
352
|
+
|
|
353
|
+
MODEL_TYPE_FIELD_NUMBER: _builtins.int
|
|
354
|
+
EPOCHS_FIELD_NUMBER: _builtins.int
|
|
355
|
+
LEARNING_RATE_FIELD_NUMBER: _builtins.int
|
|
356
|
+
BATCH_SIZE_FIELD_NUMBER: _builtins.int
|
|
357
|
+
CUSTOM_PARAMETERS_FIELD_NUMBER: _builtins.int
|
|
358
|
+
model_type: Global___ModelType.ValueType
|
|
331
359
|
"""Type of model to use for synthesis"""
|
|
332
|
-
epochs:
|
|
360
|
+
epochs: _builtins.int
|
|
333
361
|
"""Training epochs"""
|
|
334
|
-
learning_rate:
|
|
362
|
+
learning_rate: _builtins.float
|
|
335
363
|
"""Learning rate for training"""
|
|
336
|
-
batch_size:
|
|
364
|
+
batch_size: _builtins.int
|
|
337
365
|
"""Batch size for training"""
|
|
338
|
-
@property
|
|
339
|
-
def custom_parameters(
|
|
340
|
-
self,
|
|
341
|
-
) -> google.protobuf.internal.containers.ScalarMap[builtins.str, builtins.str]:
|
|
366
|
+
@_builtins.property
|
|
367
|
+
def custom_parameters(self) -> _containers.ScalarMap[_builtins.str, _builtins.str]:
|
|
342
368
|
"""Additional model-specific parameters"""
|
|
343
369
|
|
|
344
370
|
def __init__(
|
|
345
371
|
self,
|
|
346
372
|
*,
|
|
347
|
-
model_type:
|
|
348
|
-
epochs:
|
|
349
|
-
learning_rate:
|
|
350
|
-
batch_size:
|
|
351
|
-
custom_parameters:
|
|
352
|
-
| None = ...,
|
|
373
|
+
model_type: Global___ModelType.ValueType | None = ...,
|
|
374
|
+
epochs: _builtins.int | None = ...,
|
|
375
|
+
learning_rate: _builtins.float | None = ...,
|
|
376
|
+
batch_size: _builtins.int | None = ...,
|
|
377
|
+
custom_parameters: _abc.Mapping[_builtins.str, _builtins.str] | None = ...,
|
|
353
378
|
) -> None: ...
|
|
354
|
-
|
|
355
|
-
|
|
356
|
-
|
|
357
|
-
|
|
358
|
-
|
|
359
|
-
|
|
360
|
-
|
|
361
|
-
|
|
362
|
-
|
|
363
|
-
|
|
364
|
-
|
|
365
|
-
|
|
366
|
-
|
|
367
|
-
|
|
368
|
-
|
|
369
|
-
|
|
370
|
-
|
|
379
|
+
_HasFieldArgType: _TypeAlias = _typing.Literal[
|
|
380
|
+
"batch_size",
|
|
381
|
+
b"batch_size",
|
|
382
|
+
"custom_parameters",
|
|
383
|
+
b"custom_parameters",
|
|
384
|
+
"epochs",
|
|
385
|
+
b"epochs",
|
|
386
|
+
"learning_rate",
|
|
387
|
+
b"learning_rate",
|
|
388
|
+
"model_type",
|
|
389
|
+
b"model_type",
|
|
390
|
+
] # noqa: Y015
|
|
391
|
+
def HasField(self, field_name: _HasFieldArgType) -> _builtins.bool: ...
|
|
392
|
+
_ClearFieldArgType: _TypeAlias = _typing.Literal[
|
|
393
|
+
"batch_size",
|
|
394
|
+
b"batch_size",
|
|
395
|
+
"custom_parameters",
|
|
396
|
+
b"custom_parameters",
|
|
397
|
+
"epochs",
|
|
398
|
+
b"epochs",
|
|
399
|
+
"learning_rate",
|
|
400
|
+
b"learning_rate",
|
|
401
|
+
"model_type",
|
|
402
|
+
b"model_type",
|
|
403
|
+
] # noqa: Y015
|
|
404
|
+
def ClearField(self, field_name: _ClearFieldArgType) -> None: ...
|
|
405
|
+
|
|
406
|
+
Global___ModelConfig: _TypeAlias = ModelConfig # noqa: Y015
|
|
@@ -3,18 +3,19 @@
|
|
|
3
3
|
isort:skip_file
|
|
4
4
|
"""
|
|
5
5
|
|
|
6
|
-
import abc
|
|
7
|
-
import
|
|
8
|
-
import
|
|
9
|
-
import grpc
|
|
10
|
-
import typing
|
|
6
|
+
from collections import abc as _abc
|
|
7
|
+
from grpc import aio as _aio
|
|
8
|
+
import abc as _abc_1
|
|
9
|
+
import grpc as _grpc
|
|
10
|
+
import typing as _typing
|
|
11
11
|
|
|
12
|
-
_T =
|
|
12
|
+
_T = _typing.TypeVar("_T")
|
|
13
13
|
|
|
14
14
|
class _MaybeAsyncIterator(
|
|
15
|
-
|
|
16
|
-
collections.abc.Iterator[_T],
|
|
17
|
-
metaclass=abc.ABCMeta,
|
|
15
|
+
_abc.AsyncIterator[_T], _abc.Iterator[_T], metaclass=_abc_1.ABCMeta
|
|
18
16
|
): ...
|
|
19
|
-
class _ServicerContext(
|
|
17
|
+
class _ServicerContext(_grpc.ServicerContext, _aio.ServicerContext): # type: ignore[misc, type-arg]
|
|
20
18
|
...
|
|
19
|
+
|
|
20
|
+
GRPC_GENERATED_VERSION: str
|
|
21
|
+
GRPC_VERSION: str
|