chalkpy 2.96.3__py3-none-any.whl → 2.96.5__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.
- chalk/_gen/chalk/auth/v1/permissions_pb2.py +11 -3
- chalk/_gen/chalk/auth/v1/permissions_pb2.pyi +4 -0
- chalk/_gen/chalk/graph/v1/graph_pb2.py +92 -92
- chalk/_gen/chalk/python/v1/types_pb2.py +31 -25
- chalk/_gen/chalk/python/v1/types_pb2.pyi +23 -4
- chalk/_gen/chalk/server/v1/benchmark_pb2.py +33 -31
- chalk/_gen/chalk/server/v1/benchmark_pb2.pyi +12 -1
- chalk/_gen/chalk/server/v1/builder_pb2.py +240 -238
- chalk/_gen/chalk/server/v1/builder_pb2.pyi +20 -2
- chalk/_gen/chalk/server/v1/deployment_pb2.py +8 -6
- chalk/_gen/chalk/server/v1/infrastructure_pipelines_pb2.py +76 -0
- chalk/_gen/chalk/server/v1/infrastructure_pipelines_pb2.pyi +92 -0
- chalk/_gen/chalk/server/v1/infrastructure_pipelines_pb2_grpc.py +262 -0
- chalk/_gen/chalk/server/v1/infrastructure_pipelines_pb2_grpc.pyi +86 -0
- chalk/_gen/chalk/server/v1/scheduler_pb2.py +14 -12
- chalk/_gen/chalk/server/v1/scheduler_pb2.pyi +19 -9
- chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.py +7 -5
- chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.pyi +13 -0
- chalk/_version.py +1 -1
- chalk/features/resolver.py +99 -11
- chalk/functions/__init__.py +69 -0
- chalk/parsed/to_proto.py +3 -1
- {chalkpy-2.96.3.dist-info → chalkpy-2.96.5.dist-info}/METADATA +1 -1
- {chalkpy-2.96.3.dist-info → chalkpy-2.96.5.dist-info}/RECORD +27 -23
- {chalkpy-2.96.3.dist-info → chalkpy-2.96.5.dist-info}/WHEEL +1 -1
- {chalkpy-2.96.3.dist-info → chalkpy-2.96.5.dist-info}/entry_points.txt +0 -0
- {chalkpy-2.96.3.dist-info → chalkpy-2.96.5.dist-info}/top_level.txt +0 -0
|
@@ -60,8 +60,16 @@ class TyTuple(_message.Message):
|
|
|
60
60
|
is_variable: bool
|
|
61
61
|
def __init__(self, fixed: _Optional[_Iterable[_Union[Ty, _Mapping]]] = ..., is_variable: bool = ...) -> None: ...
|
|
62
62
|
|
|
63
|
+
class StringTyPair(_message.Message):
|
|
64
|
+
__slots__ = ("key", "ty")
|
|
65
|
+
KEY_FIELD_NUMBER: _ClassVar[int]
|
|
66
|
+
TY_FIELD_NUMBER: _ClassVar[int]
|
|
67
|
+
key: str
|
|
68
|
+
ty: Ty
|
|
69
|
+
def __init__(self, key: _Optional[str] = ..., ty: _Optional[_Union[Ty, _Mapping]] = ...) -> None: ...
|
|
70
|
+
|
|
63
71
|
class TyLogicalStruct(_message.Message):
|
|
64
|
-
__slots__ = ("fields",)
|
|
72
|
+
__slots__ = ("fields", "ordered_fields")
|
|
65
73
|
class FieldsEntry(_message.Message):
|
|
66
74
|
__slots__ = ("key", "value")
|
|
67
75
|
KEY_FIELD_NUMBER: _ClassVar[int]
|
|
@@ -71,11 +79,17 @@ class TyLogicalStruct(_message.Message):
|
|
|
71
79
|
def __init__(self, key: _Optional[str] = ..., value: _Optional[_Union[Ty, _Mapping]] = ...) -> None: ...
|
|
72
80
|
|
|
73
81
|
FIELDS_FIELD_NUMBER: _ClassVar[int]
|
|
82
|
+
ORDERED_FIELDS_FIELD_NUMBER: _ClassVar[int]
|
|
74
83
|
fields: _containers.MessageMap[str, Ty]
|
|
75
|
-
|
|
84
|
+
ordered_fields: _containers.RepeatedCompositeFieldContainer[StringTyPair]
|
|
85
|
+
def __init__(
|
|
86
|
+
self,
|
|
87
|
+
fields: _Optional[_Mapping[str, Ty]] = ...,
|
|
88
|
+
ordered_fields: _Optional[_Iterable[_Union[StringTyPair, _Mapping]]] = ...,
|
|
89
|
+
) -> None: ...
|
|
76
90
|
|
|
77
91
|
class TyFeatureClass(_message.Message):
|
|
78
|
-
__slots__ = ("constructor_namespace", "assigned_fields")
|
|
92
|
+
__slots__ = ("constructor_namespace", "assigned_fields", "ordered_assigned_fields")
|
|
79
93
|
class AssignedFieldsEntry(_message.Message):
|
|
80
94
|
__slots__ = ("key", "value")
|
|
81
95
|
KEY_FIELD_NUMBER: _ClassVar[int]
|
|
@@ -86,10 +100,15 @@ class TyFeatureClass(_message.Message):
|
|
|
86
100
|
|
|
87
101
|
CONSTRUCTOR_NAMESPACE_FIELD_NUMBER: _ClassVar[int]
|
|
88
102
|
ASSIGNED_FIELDS_FIELD_NUMBER: _ClassVar[int]
|
|
103
|
+
ORDERED_ASSIGNED_FIELDS_FIELD_NUMBER: _ClassVar[int]
|
|
89
104
|
constructor_namespace: str
|
|
90
105
|
assigned_fields: _containers.MessageMap[str, Ty]
|
|
106
|
+
ordered_assigned_fields: _containers.RepeatedCompositeFieldContainer[StringTyPair]
|
|
91
107
|
def __init__(
|
|
92
|
-
self,
|
|
108
|
+
self,
|
|
109
|
+
constructor_namespace: _Optional[str] = ...,
|
|
110
|
+
assigned_fields: _Optional[_Mapping[str, Ty]] = ...,
|
|
111
|
+
ordered_assigned_fields: _Optional[_Iterable[_Union[StringTyPair, _Mapping]]] = ...,
|
|
93
112
|
) -> None: ...
|
|
94
113
|
|
|
95
114
|
class TyEnum(_message.Message):
|
|
@@ -21,7 +21,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
|
|
|
21
21
|
|
|
22
22
|
|
|
23
23
|
DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
|
|
24
|
-
b'\n\x1f\x63halk/server/v1/benchmark.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a"chalk/common/v1/online_query.proto\x1a\x1d\x63halk/server/v1/builder.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto"6\n\rInputFeatures\x12%\n\x0einput_features\x18\x01 \x03(\tR\rinputFeatures"\
|
|
24
|
+
b'\n\x1f\x63halk/server/v1/benchmark.proto\x12\x0f\x63halk.server.v1\x1a\x1f\x63halk/auth/v1/permissions.proto\x1a"chalk/common/v1/online_query.proto\x1a\x1d\x63halk/server/v1/builder.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto"6\n\rInputFeatures\x12%\n\x0einput_features\x18\x01 \x03(\tR\rinputFeatures"n\n\x11NamedQueryRequest\x12\x1d\n\nquery_name\x18\x01 \x01(\tR\tqueryName\x12(\n\rquery_version\x18\x02 \x01(\tH\x00R\x0cqueryVersion\x88\x01\x01\x42\x10\n\x0e_query_version"\xe0\x02\n\x1cSimpleOnlineQueryBulkRequest\x12)\n\x0einput_features\x18\x01 \x03(\tB\x02\x18\x01R\rinputFeatures\x12P\n\x13input_features_list\x18\x03 \x01(\x0b\x32\x1e.chalk.server.v1.InputFeaturesH\x00R\x11inputFeaturesList\x12\x1f\n\ninput_file\x18\x04 \x01(\tH\x00R\tinputFile\x12\'\n\x0foutput_features\x18\x02 \x03(\tR\x0eoutputFeatures\x12W\n\x13named_query_request\x18\x05 \x01(\x0b\x32".chalk.server.v1.NamedQueryRequestH\x01R\x11namedQueryRequest\x88\x01\x01\x42\x08\n\x06inputsB\x16\n\x14_named_query_request"\xa9\x01\n\rContainerSpec\x12\x42\n\x07request\x18\x01 \x01(\x0b\x32#.chalk.server.v1.KubeResourceConfigH\x00R\x07request\x88\x01\x01\x12>\n\x05limit\x18\x02 \x01(\x0b\x32#.chalk.server.v1.KubeResourceConfigH\x01R\x05limit\x88\x01\x01\x42\n\n\x08_requestB\x08\n\x06_limit"\xf8\x04\n\x16\x43reateBenchmarkRequest\x12\x1d\n\nwarmup_qps\x18\x01 \x01(\x03R\twarmupQps\x12\x42\n\x0fwarmup_duration\x18\x02 \x01(\x0b\x32\x19.google.protobuf.DurationR\x0ewarmupDuration\x12\x10\n\x03qps\x18\x03 \x01(\x03R\x03qps\x12\x35\n\x08\x64uration\x18\x04 \x01(\x0b\x32\x19.google.protobuf.DurationR\x08\x64uration\x12H\n\nquery_bulk\x18\x05 \x01(\x0b\x32\'.chalk.common.v1.OnlineQueryBulkRequestH\x00R\tqueryBulk\x12[\n\x11simple_query_bulk\x18\x06 \x01(\x0b\x32-.chalk.server.v1.SimpleOnlineQueryBulkRequestH\x00R\x0fsimpleQueryBulk\x12 \n\x0bpercentiles\x18\x07 \x03(\x03R\x0bpercentiles\x12*\n\x0eimage_override\x18\x08 \x01(\tH\x01R\rimageOverride\x88\x01\x01\x12R\n\x15warmup_container_spec\x18\t \x01(\x0b\x32\x1e.chalk.server.v1.ContainerSpecR\x13warmupContainerSpec\x12\x45\n\x0e\x63ontainer_spec\x18\n \x01(\x0b\x32\x1e.chalk.server.v1.ContainerSpecR\rcontainerSpecB\x0f\n\rquery_requestB\x11\n\x0f_image_override"v\n\x17\x43reateBenchmarkResponse\x12\x38\n\x06status\x18\x01 \x01(\x0e\x32 .chalk.server.v1.BenchmarkStatusR\x06status\x12!\n\x0c\x62\x65nchmark_id\x18\x02 \x01(\tR\x0b\x62\x65nchmarkId"\x99\x01\n\x12\x42\x65nchmarkInputFile\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x17\n\x04size\x18\x02 \x01(\x03H\x00R\x04size\x88\x01\x01\x12>\n\nupdated_at\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\tupdatedAt\x88\x01\x01\x42\x07\n\x05_sizeB\r\n\x0b_updated_at"7\n\x1dGetAvailableInputFilesRequest\x12\x16\n\x06prefix\x18\x01 \x01(\tR\x06prefix"f\n\x1eGetAvailableInputFilesResponse\x12\x44\n\x0binput_files\x18\x01 \x03(\x0b\x32#.chalk.server.v1.BenchmarkInputFileR\ninputFiles"e\n\x1dGetInputFileUploadUrlsRequest\x12\x44\n\x0binput_files\x18\x01 \x03(\x0b\x32#.chalk.server.v1.BenchmarkInputFileR\ninputFiles"G\n\x1eGetInputFileUploadUrlsResponse\x12%\n\x0epresigned_urls\x18\x01 \x03(\tR\rpresignedUrls"8\n\x1eGetAvailableResultFilesRequest\x12\x16\n\x06prefix\x18\x01 \x01(\tR\x06prefix"i\n\x1fGetAvailableResultFilesResponse\x12\x46\n\x0cresult_files\x18\x01 \x03(\x0b\x32#.chalk.server.v1.BenchmarkInputFileR\x0bresultFiles"9\n\x18GetResultFileUrlsRequest\x12\x1d\n\nfile_names\x18\x01 \x03(\tR\tfileNames"B\n\x19GetResultFileUrlsResponse\x12%\n\x0epresigned_urls\x18\x01 \x03(\tR\rpresignedUrls*\xe8\x01\n\x0f\x42\x65nchmarkStatus\x12 \n\x1c\x42\x45NCHMARK_STATUS_UNSPECIFIED\x10\x00\x12\x1b\n\x17\x42\x45NCHMARK_STATUS_QUEUED\x10\x01\x12\x1c\n\x18\x42\x45NCHMARK_STATUS_WORKING\x10\x02\x12\x1e\n\x1a\x42\x45NCHMARK_STATUS_COMPLETED\x10\x03\x12\x1b\n\x17\x42\x45NCHMARK_STATUS_FAILED\x10\x04\x12\x1c\n\x18\x42\x45NCHMARK_STATUS_SKIPPED\x10\x05\x12\x1d\n\x19\x42\x45NCHMARK_STATUS_CANCELED\x10\x06\x32\xf2\x04\n\x10\x42\x65nchmarkService\x12i\n\x0f\x43reateBenchmark\x12\'.chalk.server.v1.CreateBenchmarkRequest\x1a(.chalk.server.v1.CreateBenchmarkResponse"\x03\x80}\x03\x12~\n\x16GetAvailableInputFiles\x12..chalk.server.v1.GetAvailableInputFilesRequest\x1a/.chalk.server.v1.GetAvailableInputFilesResponse"\x03\x80}\x03\x12~\n\x16GetInputFileUploadUrls\x12..chalk.server.v1.GetInputFileUploadUrlsRequest\x1a/.chalk.server.v1.GetInputFileUploadUrlsResponse"\x03\x80}\x03\x12\x81\x01\n\x17GetAvailableResultFiles\x12/.chalk.server.v1.GetAvailableResultFilesRequest\x1a\x30.chalk.server.v1.GetAvailableResultFilesResponse"\x03\x80}\x03\x12o\n\x11GetResultFileUrls\x12).chalk.server.v1.GetResultFileUrlsRequest\x1a*.chalk.server.v1.GetResultFileUrlsResponse"\x03\x80}\x03\x42\x97\x01\n\x13\x63om.chalk.server.v1B\x0e\x42\x65nchmarkProtoP\x01Z\x12server/v1;serverv1\xa2\x02\x03\x43SX\xaa\x02\x0f\x43halk.Server.V1\xca\x02\x0f\x43halk\\Server\\V1\xe2\x02\x1b\x43halk\\Server\\V1\\GPBMetadata\xea\x02\x11\x43halk::Server::V1b\x06proto3'
|
|
25
25
|
)
|
|
26
26
|
|
|
27
27
|
_globals = globals()
|
|
@@ -44,36 +44,38 @@ if _descriptor._USE_C_DESCRIPTORS == False:
|
|
|
44
44
|
_globals["_BENCHMARKSERVICE"].methods_by_name["GetAvailableResultFiles"]._serialized_options = b"\200}\003"
|
|
45
45
|
_globals["_BENCHMARKSERVICE"].methods_by_name["GetResultFileUrls"]._options = None
|
|
46
46
|
_globals["_BENCHMARKSERVICE"].methods_by_name["GetResultFileUrls"]._serialized_options = b"\200}\003"
|
|
47
|
-
_globals["_BENCHMARKSTATUS"]._serialized_start =
|
|
48
|
-
_globals["_BENCHMARKSTATUS"]._serialized_end =
|
|
47
|
+
_globals["_BENCHMARKSTATUS"]._serialized_start = 2453
|
|
48
|
+
_globals["_BENCHMARKSTATUS"]._serialized_end = 2685
|
|
49
49
|
_globals["_INPUTFEATURES"]._serialized_start = 217
|
|
50
50
|
_globals["_INPUTFEATURES"]._serialized_end = 271
|
|
51
|
-
_globals["
|
|
52
|
-
_globals["
|
|
53
|
-
_globals["
|
|
54
|
-
_globals["
|
|
55
|
-
_globals["
|
|
56
|
-
_globals["
|
|
57
|
-
_globals["
|
|
58
|
-
_globals["
|
|
59
|
-
_globals["
|
|
60
|
-
_globals["
|
|
61
|
-
_globals["
|
|
62
|
-
_globals["
|
|
63
|
-
_globals["
|
|
64
|
-
_globals["
|
|
65
|
-
_globals["
|
|
66
|
-
_globals["
|
|
67
|
-
_globals["
|
|
68
|
-
_globals["
|
|
69
|
-
_globals["
|
|
70
|
-
_globals["
|
|
71
|
-
_globals["
|
|
72
|
-
_globals["
|
|
73
|
-
_globals["
|
|
74
|
-
_globals["
|
|
75
|
-
_globals["
|
|
76
|
-
_globals["
|
|
77
|
-
_globals["
|
|
78
|
-
_globals["
|
|
51
|
+
_globals["_NAMEDQUERYREQUEST"]._serialized_start = 273
|
|
52
|
+
_globals["_NAMEDQUERYREQUEST"]._serialized_end = 383
|
|
53
|
+
_globals["_SIMPLEONLINEQUERYBULKREQUEST"]._serialized_start = 386
|
|
54
|
+
_globals["_SIMPLEONLINEQUERYBULKREQUEST"]._serialized_end = 738
|
|
55
|
+
_globals["_CONTAINERSPEC"]._serialized_start = 741
|
|
56
|
+
_globals["_CONTAINERSPEC"]._serialized_end = 910
|
|
57
|
+
_globals["_CREATEBENCHMARKREQUEST"]._serialized_start = 913
|
|
58
|
+
_globals["_CREATEBENCHMARKREQUEST"]._serialized_end = 1545
|
|
59
|
+
_globals["_CREATEBENCHMARKRESPONSE"]._serialized_start = 1547
|
|
60
|
+
_globals["_CREATEBENCHMARKRESPONSE"]._serialized_end = 1665
|
|
61
|
+
_globals["_BENCHMARKINPUTFILE"]._serialized_start = 1668
|
|
62
|
+
_globals["_BENCHMARKINPUTFILE"]._serialized_end = 1821
|
|
63
|
+
_globals["_GETAVAILABLEINPUTFILESREQUEST"]._serialized_start = 1823
|
|
64
|
+
_globals["_GETAVAILABLEINPUTFILESREQUEST"]._serialized_end = 1878
|
|
65
|
+
_globals["_GETAVAILABLEINPUTFILESRESPONSE"]._serialized_start = 1880
|
|
66
|
+
_globals["_GETAVAILABLEINPUTFILESRESPONSE"]._serialized_end = 1982
|
|
67
|
+
_globals["_GETINPUTFILEUPLOADURLSREQUEST"]._serialized_start = 1984
|
|
68
|
+
_globals["_GETINPUTFILEUPLOADURLSREQUEST"]._serialized_end = 2085
|
|
69
|
+
_globals["_GETINPUTFILEUPLOADURLSRESPONSE"]._serialized_start = 2087
|
|
70
|
+
_globals["_GETINPUTFILEUPLOADURLSRESPONSE"]._serialized_end = 2158
|
|
71
|
+
_globals["_GETAVAILABLERESULTFILESREQUEST"]._serialized_start = 2160
|
|
72
|
+
_globals["_GETAVAILABLERESULTFILESREQUEST"]._serialized_end = 2216
|
|
73
|
+
_globals["_GETAVAILABLERESULTFILESRESPONSE"]._serialized_start = 2218
|
|
74
|
+
_globals["_GETAVAILABLERESULTFILESRESPONSE"]._serialized_end = 2323
|
|
75
|
+
_globals["_GETRESULTFILEURLSREQUEST"]._serialized_start = 2325
|
|
76
|
+
_globals["_GETRESULTFILEURLSREQUEST"]._serialized_end = 2382
|
|
77
|
+
_globals["_GETRESULTFILEURLSRESPONSE"]._serialized_start = 2384
|
|
78
|
+
_globals["_GETRESULTFILEURLSRESPONSE"]._serialized_end = 2450
|
|
79
|
+
_globals["_BENCHMARKSERVICE"]._serialized_start = 2688
|
|
80
|
+
_globals["_BENCHMARKSERVICE"]._serialized_end = 3314
|
|
79
81
|
# @@protoc_insertion_point(module_scope)
|
|
@@ -41,22 +41,33 @@ class InputFeatures(_message.Message):
|
|
|
41
41
|
input_features: _containers.RepeatedScalarFieldContainer[str]
|
|
42
42
|
def __init__(self, input_features: _Optional[_Iterable[str]] = ...) -> None: ...
|
|
43
43
|
|
|
44
|
+
class NamedQueryRequest(_message.Message):
|
|
45
|
+
__slots__ = ("query_name", "query_version")
|
|
46
|
+
QUERY_NAME_FIELD_NUMBER: _ClassVar[int]
|
|
47
|
+
QUERY_VERSION_FIELD_NUMBER: _ClassVar[int]
|
|
48
|
+
query_name: str
|
|
49
|
+
query_version: str
|
|
50
|
+
def __init__(self, query_name: _Optional[str] = ..., query_version: _Optional[str] = ...) -> None: ...
|
|
51
|
+
|
|
44
52
|
class SimpleOnlineQueryBulkRequest(_message.Message):
|
|
45
|
-
__slots__ = ("input_features", "input_features_list", "input_file", "output_features")
|
|
53
|
+
__slots__ = ("input_features", "input_features_list", "input_file", "output_features", "named_query_request")
|
|
46
54
|
INPUT_FEATURES_FIELD_NUMBER: _ClassVar[int]
|
|
47
55
|
INPUT_FEATURES_LIST_FIELD_NUMBER: _ClassVar[int]
|
|
48
56
|
INPUT_FILE_FIELD_NUMBER: _ClassVar[int]
|
|
49
57
|
OUTPUT_FEATURES_FIELD_NUMBER: _ClassVar[int]
|
|
58
|
+
NAMED_QUERY_REQUEST_FIELD_NUMBER: _ClassVar[int]
|
|
50
59
|
input_features: _containers.RepeatedScalarFieldContainer[str]
|
|
51
60
|
input_features_list: InputFeatures
|
|
52
61
|
input_file: str
|
|
53
62
|
output_features: _containers.RepeatedScalarFieldContainer[str]
|
|
63
|
+
named_query_request: NamedQueryRequest
|
|
54
64
|
def __init__(
|
|
55
65
|
self,
|
|
56
66
|
input_features: _Optional[_Iterable[str]] = ...,
|
|
57
67
|
input_features_list: _Optional[_Union[InputFeatures, _Mapping]] = ...,
|
|
58
68
|
input_file: _Optional[str] = ...,
|
|
59
69
|
output_features: _Optional[_Iterable[str]] = ...,
|
|
70
|
+
named_query_request: _Optional[_Union[NamedQueryRequest, _Mapping]] = ...,
|
|
60
71
|
) -> None: ...
|
|
61
72
|
|
|
62
73
|
class ContainerSpec(_message.Message):
|