chalkpy 2.89.22__py3-none-any.whl → 2.95.3__py3-none-any.whl

chalk/_gen/chalk/common/v1/online_query_pb2.py

@@ -22,7 +22,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n"chalk/common/v1/online_query.proto\x12\x0f\x63halk.common.v1\x1a!chalk/common/v1/chalk_error.proto\x1a$chalk/expression/v1/expression.proto\x1a\x1a\x63halk/graph/v1/graph.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xc9\x04\n\x12OnlineQueryRequest\x12G\n\x06inputs\x18\x01 \x03(\x0b\x32/.chalk.common.v1.OnlineQueryRequest.InputsEntryR\x06inputs\x12\x35\n\x07outputs\x18\x02 \x03(\x0b\x32\x1b.chalk.common.v1.OutputExprR\x07outputs\x12\x31\n\x03now\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x03now\x88\x01\x01\x12P\n\tstaleness\x18\x04 \x03(\x0b\x32\x32.chalk.common.v1.OnlineQueryRequest.StalenessEntryR\tstaleness\x12=\n\x07\x63ontext\x18\x05 \x01(\x0b\x32#.chalk.common.v1.OnlineQueryContextR\x07\x63ontext\x12V\n\x10response_options\x18\x06 \x01(\x0b\x32+.chalk.common.v1.OnlineQueryResponseOptionsR\x0fresponseOptions\x1aQ\n\x0bInputsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x1a<\n\x0eStalenessEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x06\n\x04_now"\x8e\x04\n\x16OnlineQueryBulkRequest\x12%\n\x0einputs_feather\x18\x01 \x01(\x0cR\rinputsFeather\x12\x35\n\x07outputs\x18\x02 \x03(\x0b\x32\x1b.chalk.common.v1.OutputExprR\x07outputs\x12,\n\x03now\x18\x03 \x03(\x0b\x32\x1a.google.protobuf.TimestampR\x03now\x12T\n\tstaleness\x18\x04 \x03(\x0b\x32\x36.chalk.common.v1.OnlineQueryBulkRequest.StalenessEntryR\tstaleness\x12=\n\x07\x63ontext\x18\x05 \x01(\x0b\x32#.chalk.common.v1.OnlineQueryContextR\x07\x63ontext\x12V\n\x10response_options\x18\x06 \x01(\x0b\x32+.chalk.common.v1.OnlineQueryResponseOptionsR\x0fresponseOptions\x12=\n\tbody_type\x18\x07 \x01(\x0e\x32 .chalk.common.v1.FeatherBodyTypeR\x08\x62odyType\x1a<\n\x0eStalenessEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01"\xb9\x01\n\x12GenericSingleQuery\x12L\n\x0esingle_request\x18\x01 \x01(\x0b\x32#.chalk.common.v1.OnlineQueryRequestH\x00R\rsingleRequest\x12L\n\x0c\x62ulk_request\x18\x02 \x01(\x0b\x32\'.chalk.common.v1.OnlineQueryBulkRequestH\x00R\x0b\x62ulkRequestB\x07\n\x05query"X\n\x17OnlineQueryMultiRequest\x12=\n\x07queries\x18\x01 \x03(\x0b\x32#.chalk.common.v1.GenericSingleQueryR\x07queries"\x99\x01\n\x11\x46\x65\x61tureExpression\x12,\n\x12output_column_name\x18\x01 \x01(\tR\x10outputColumnName\x12\x1c\n\tnamespace\x18\x03 \x01(\tR\tnamespace\x12\x38\n\x04\x65xpr\x18\x02 \x01(\x0b\x32$.chalk.expression.v1.LogicalExprNodeR\x04\x65xpr"\x8c\x01\n\nOutputExpr\x12!\n\x0b\x66\x65\x61ture_fqn\x18\x01 \x01(\tH\x00R\nfeatureFqn\x12S\n\x12\x66\x65\x61ture_expression\x18\x02 \x01(\x0b\x32".chalk.common.v1.FeatureExpressionH\x00R\x11\x66\x65\x61tureExpressionB\x06\n\x04\x65xpr"\xc2\x07\n\x12OnlineQueryContext\x12$\n\x0b\x65nvironment\x18\x01 \x01(\tB\x02\x18\x01R\x0b\x65nvironment\x12\x12\n\x04tags\x18\x02 \x03(\tR\x04tags\x12\x34\n\x16required_resolver_tags\x18\x03 \x03(\tR\x14requiredResolverTags\x12,\n\rdeployment_id\x18\x04 \x01(\tB\x02\x18\x01H\x00R\x0c\x64\x65ploymentId\x88\x01\x01\x12$\n\tbranch_id\x18\x05 \x01(\tB\x02\x18\x01H\x01R\x08\x62ranchId\x88\x01\x01\x12*\n\x0e\x63orrelation_id\x18\x06 \x01(\tH\x02R\rcorrelationId\x88\x01\x01\x12"\n\nquery_name\x18\x07 \x01(\tH\x03R\tqueryName\x88\x01\x01\x12\x31\n\x12query_name_version\x18\x08 \x01(\tH\x04R\x10queryNameVersion\x88\x01\x01\x12J\n\x07options\x18\t \x03(\x0b\x32\x30.chalk.common.v1.OnlineQueryContext.OptionsEntryR\x07options\x12]\n\x1dvalue_metrics_tag_by_features\x18\n \x03(\x0b\x32\x1b.chalk.common.v1.OutputExprR\x19valueMetricsTagByFeatures\x12Z\n\rquery_context\x18\x0b \x03(\x0b\x32\x35.chalk.common.v1.OnlineQueryContext.QueryContextEntryR\x0cqueryContext\x12\x46\n\roverlay_graph\x18\x0c \x01(\x0b\x32\x1c.chalk.graph.v1.OverlayGraphH\x05R\x0coverlayGraph\x88\x01\x01\x1aR\n\x0cOptionsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x1aW\n\x11QueryContextEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\x10\n\x0e_deployment_idB\x0c\n\n_branch_idB\x11\n\x0f_correlation_idB\r\n\x0b_query_nameB\x15\n\x13_query_name_versionB\x10\n\x0e_overlay_graph"\xe2\x02\n\x1aOnlineQueryResponseOptions\x12!\n\x0cinclude_meta\x18\x01 \x01(\x08R\x0bincludeMeta\x12\x39\n\x07\x65xplain\x18\x02 \x01(\x0b\x32\x1f.chalk.common.v1.ExplainOptionsR\x07\x65xplain\x12R\n\x10\x65ncoding_options\x18\x03 \x01(\x0b\x32\'.chalk.common.v1.FeatureEncodingOptionsR\x0f\x65ncodingOptions\x12U\n\x08metadata\x18\x04 \x03(\x0b\x32\x39.chalk.common.v1.OnlineQueryResponseOptions.MetadataEntryR\x08metadata\x1a;\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01"\x10\n\x0e\x45xplainOptions"S\n\x16\x46\x65\x61tureEncodingOptions\x12\x39\n\x19\x65ncode_structs_as_objects\x18\x01 \x01(\x08R\x16\x65ncodeStructsAsObjects"\xcd\x01\n\x13OnlineQueryResponse\x12\x36\n\x04\x64\x61ta\x18\x01 \x01(\x0b\x32".chalk.common.v1.OnlineQueryResultR\x04\x64\x61ta\x12\x33\n\x06\x65rrors\x18\x02 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors\x12I\n\rresponse_meta\x18\x03 \x01(\x0b\x32$.chalk.common.v1.OnlineQueryMetadataR\x0cresponseMeta"\xd6\x02\n\x17OnlineQueryBulkResponse\x12!\n\x0cscalars_data\x18\x01 \x01(\x0cR\x0bscalarsData\x12Y\n\x0bgroups_data\x18\x02 \x03(\x0b\x32\x38.chalk.common.v1.OnlineQueryBulkResponse.GroupsDataEntryR\ngroupsData\x12\x33\n\x06\x65rrors\x18\x03 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors\x12I\n\rresponse_meta\x18\x04 \x01(\x0b\x32$.chalk.common.v1.OnlineQueryMetadataR\x0cresponseMeta\x1a=\n\x0fGroupsDataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\x0cR\x05value:\x02\x38\x01"\xc2\x01\n\x15GenericSingleResponse\x12O\n\x0fsingle_response\x18\x01 \x01(\x0b\x32$.chalk.common.v1.OnlineQueryResponseH\x00R\x0esingleResponse\x12O\n\rbulk_response\x18\x02 \x01(\x0b\x32(.chalk.common.v1.OnlineQueryBulkResponseH\x00R\x0c\x62ulkResponseB\x07\n\x05query"\x95\x01\n\x18OnlineQueryMultiResponse\x12\x44\n\tresponses\x18\x01 \x03(\x0b\x32&.chalk.common.v1.GenericSingleResponseR\tresponses\x12\x33\n\x06\x65rrors\x18\x02 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"M\n\x11OnlineQueryResult\x12\x38\n\x07results\x18\x01 \x03(\x0b\x32\x1e.chalk.common.v1.FeatureResultR\x07results"\xb9\x02\n\rFeatureResult\x12\x14\n\x05\x66ield\x18\x01 \x01(\tR\x05\x66ield\x12*\n\x04pkey\x18\x06 \x01(\x0b\x32\x16.google.protobuf.ValueR\x04pkey\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value\x12\x36\n\x05\x65rror\x18\x03 \x01(\x0b\x32\x1b.chalk.common.v1.ChalkErrorH\x00R\x05\x65rror\x88\x01\x01\x12/\n\x02ts\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\x02ts\x88\x01\x01\x12\x35\n\x04meta\x18\x05 \x01(\x0b\x32\x1c.chalk.common.v1.FeatureMetaH\x02R\x04meta\x88\x01\x01\x42\x08\n\x06_errorB\x05\n\x03_tsB\x07\n\x05_meta"\x9b\x01\n\x0b\x46\x65\x61tureMeta\x12.\n\x13\x63hosen_resolver_fqn\x18\x01 \x01(\tR\x11\x63hosenResolverFqn\x12\x1b\n\tcache_hit\x18\x02 \x01(\x08R\x08\x63\x61\x63heHit\x12%\n\x0eprimitive_type\x18\x03 \x01(\tR\rprimitiveType\x12\x18\n\x07version\x18\x04 \x01(\x03R\x07version"\xfa\x05\n\x13OnlineQueryMetadata\x12H\n\x12\x65xecution_duration\x18\x01 \x01(\x0b\x32\x19.google.protobuf.DurationR\x11\x65xecutionDuration\x12#\n\rdeployment_id\x18\x02 \x01(\tR\x0c\x64\x65ploymentId\x12%\n\x0e\x65nvironment_id\x18\x03 \x01(\tR\renvironmentId\x12)\n\x10\x65nvironment_name\x18\x04 \x01(\tR\x0f\x65nvironmentName\x12\x19\n\x08query_id\x18\x05 \x01(\tR\x07queryId\x12\x43\n\x0fquery_timestamp\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x0equeryTimestamp\x12\x1d\n\nquery_hash\x18\x07 \x01(\tR\tqueryHash\x12H\n\x0e\x65xplain_output\x18\x08 \x01(\x0b\x32!.chalk.common.v1.QueryExplainInfoR\rexplainOutput\x12N\n\x08metadata\x18\t \x03(\x0b\x32\x32.chalk.common.v1.OnlineQueryMetadata.MetadataEntryR\x08metadata\x12m\n\x13\x61\x64\x64itional_metadata\x18\n \x03(\x0b\x32<.chalk.common.v1.OnlineQueryMetadata.AdditionalMetadataEntryR\x12\x61\x64\x64itionalMetadata\x1a;\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x1a]\n\x17\x41\x64\x64itionalMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01"H\n\x10QueryExplainInfo\x12$\n\x0bplan_string\x18\x01 \x01(\tH\x00R\nplanString\x88\x01\x01\x42\x0e\n\x0c_plan_string"\x81\x01\n\x19UploadFeaturesBulkRequest\x12%\n\x0einputs_feather\x18\x01 \x01(\x0cR\rinputsFeather\x12=\n\tbody_type\x18\x07 \x01(\x0e\x32 .chalk.common.v1.FeatherBodyTypeR\x08\x62odyType"Q\n\x1aUploadFeaturesBulkResponse\x12\x33\n\x06\x65rrors\x18\x01 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors*w\n\x0f\x46\x65\x61therBodyType\x12!\n\x1d\x46\x45\x41THER_BODY_TYPE_UNSPECIFIED\x10\x00\x12\x1b\n\x17\x46\x45\x41THER_BODY_TYPE_TABLE\x10\x01\x12$\n FEATHER_BODY_TYPE_RECORD_BATCHES\x10\x02\x42\x85\x01\n\x13\x63om.chalk.common.v1B\x10OnlineQueryProtoP\x01\xa2\x02\x03\x43\x43X\xaa\x02\x0f\x43halk.Common.V1\xca\x02\x0f\x43halk\\Common\\V1\xe2\x02\x1b\x43halk\\Common\\V1\\GPBMetadata\xea\x02\x11\x43halk::Common::V1b\x06proto3'
+    b'\n"chalk/common/v1/online_query.proto\x12\x0f\x63halk.common.v1\x1a!chalk/common/v1/chalk_error.proto\x1a$chalk/expression/v1/expression.proto\x1a\x1a\x63halk/graph/v1/graph.proto\x1a\x1egoogle/protobuf/duration.proto\x1a\x1cgoogle/protobuf/struct.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xc9\x04\n\x12OnlineQueryRequest\x12G\n\x06inputs\x18\x01 \x03(\x0b\x32/.chalk.common.v1.OnlineQueryRequest.InputsEntryR\x06inputs\x12\x35\n\x07outputs\x18\x02 \x03(\x0b\x32\x1b.chalk.common.v1.OutputExprR\x07outputs\x12\x31\n\x03now\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x00R\x03now\x88\x01\x01\x12P\n\tstaleness\x18\x04 \x03(\x0b\x32\x32.chalk.common.v1.OnlineQueryRequest.StalenessEntryR\tstaleness\x12=\n\x07\x63ontext\x18\x05 \x01(\x0b\x32#.chalk.common.v1.OnlineQueryContextR\x07\x63ontext\x12V\n\x10response_options\x18\x06 \x01(\x0b\x32+.chalk.common.v1.OnlineQueryResponseOptionsR\x0fresponseOptions\x1aQ\n\x0bInputsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x1a<\n\x0eStalenessEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x06\n\x04_now"\xbb\x04\n\x16OnlineQueryBulkRequest\x12\'\n\x0einputs_feather\x18\x01 \x01(\x0cH\x00R\rinputsFeather\x12\x1f\n\ninputs_sql\x18\x08 \x01(\tH\x00R\tinputsSql\x12\x35\n\x07outputs\x18\x02 \x03(\x0b\x32\x1b.chalk.common.v1.OutputExprR\x07outputs\x12,\n\x03now\x18\x03 \x03(\x0b\x32\x1a.google.protobuf.TimestampR\x03now\x12T\n\tstaleness\x18\x04 \x03(\x0b\x32\x36.chalk.common.v1.OnlineQueryBulkRequest.StalenessEntryR\tstaleness\x12=\n\x07\x63ontext\x18\x05 \x01(\x0b\x32#.chalk.common.v1.OnlineQueryContextR\x07\x63ontext\x12V\n\x10response_options\x18\x06 \x01(\x0b\x32+.chalk.common.v1.OnlineQueryResponseOptionsR\x0fresponseOptions\x12=\n\tbody_type\x18\x07 \x01(\x0e\x32 .chalk.common.v1.FeatherBodyTypeR\x08\x62odyType\x1a<\n\x0eStalenessEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x08\n\x06inputs"\xb9\x01\n\x12GenericSingleQuery\x12L\n\x0esingle_request\x18\x01 \x01(\x0b\x32#.chalk.common.v1.OnlineQueryRequestH\x00R\rsingleRequest\x12L\n\x0c\x62ulk_request\x18\x02 \x01(\x0b\x32\'.chalk.common.v1.OnlineQueryBulkRequestH\x00R\x0b\x62ulkRequestB\x07\n\x05query"X\n\x17OnlineQueryMultiRequest\x12=\n\x07queries\x18\x01 \x03(\x0b\x32#.chalk.common.v1.GenericSingleQueryR\x07queries"\x99\x01\n\x11\x46\x65\x61tureExpression\x12,\n\x12output_column_name\x18\x01 \x01(\tR\x10outputColumnName\x12\x1c\n\tnamespace\x18\x03 \x01(\tR\tnamespace\x12\x38\n\x04\x65xpr\x18\x02 \x01(\x0b\x32$.chalk.expression.v1.LogicalExprNodeR\x04\x65xpr"\x8c\x01\n\nOutputExpr\x12!\n\x0b\x66\x65\x61ture_fqn\x18\x01 \x01(\tH\x00R\nfeatureFqn\x12S\n\x12\x66\x65\x61ture_expression\x18\x02 \x01(\x0b\x32".chalk.common.v1.FeatureExpressionH\x00R\x11\x66\x65\x61tureExpressionB\x06\n\x04\x65xpr"\xc2\x07\n\x12OnlineQueryContext\x12$\n\x0b\x65nvironment\x18\x01 \x01(\tB\x02\x18\x01R\x0b\x65nvironment\x12\x12\n\x04tags\x18\x02 \x03(\tR\x04tags\x12\x34\n\x16required_resolver_tags\x18\x03 \x03(\tR\x14requiredResolverTags\x12,\n\rdeployment_id\x18\x04 \x01(\tB\x02\x18\x01H\x00R\x0c\x64\x65ploymentId\x88\x01\x01\x12$\n\tbranch_id\x18\x05 \x01(\tB\x02\x18\x01H\x01R\x08\x62ranchId\x88\x01\x01\x12*\n\x0e\x63orrelation_id\x18\x06 \x01(\tH\x02R\rcorrelationId\x88\x01\x01\x12"\n\nquery_name\x18\x07 \x01(\tH\x03R\tqueryName\x88\x01\x01\x12\x31\n\x12query_name_version\x18\x08 \x01(\tH\x04R\x10queryNameVersion\x88\x01\x01\x12J\n\x07options\x18\t \x03(\x0b\x32\x30.chalk.common.v1.OnlineQueryContext.OptionsEntryR\x07options\x12]\n\x1dvalue_metrics_tag_by_features\x18\n \x03(\x0b\x32\x1b.chalk.common.v1.OutputExprR\x19valueMetricsTagByFeatures\x12Z\n\rquery_context\x18\x0b \x03(\x0b\x32\x35.chalk.common.v1.OnlineQueryContext.QueryContextEntryR\x0cqueryContext\x12\x46\n\roverlay_graph\x18\x0c \x01(\x0b\x32\x1c.chalk.graph.v1.OverlayGraphH\x05R\x0coverlayGraph\x88\x01\x01\x1aR\n\x0cOptionsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x1aW\n\x11QueryContextEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01\x42\x10\n\x0e_deployment_idB\x0c\n\n_branch_idB\x11\n\x0f_correlation_idB\r\n\x0b_query_nameB\x15\n\x13_query_name_versionB\x10\n\x0e_overlay_graph"\xe2\x02\n\x1aOnlineQueryResponseOptions\x12!\n\x0cinclude_meta\x18\x01 \x01(\x08R\x0bincludeMeta\x12\x39\n\x07\x65xplain\x18\x02 \x01(\x0b\x32\x1f.chalk.common.v1.ExplainOptionsR\x07\x65xplain\x12R\n\x10\x65ncoding_options\x18\x03 \x01(\x0b\x32\'.chalk.common.v1.FeatureEncodingOptionsR\x0f\x65ncodingOptions\x12U\n\x08metadata\x18\x04 \x03(\x0b\x32\x39.chalk.common.v1.OnlineQueryResponseOptions.MetadataEntryR\x08metadata\x1a;\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01"\x10\n\x0e\x45xplainOptions"S\n\x16\x46\x65\x61tureEncodingOptions\x12\x39\n\x19\x65ncode_structs_as_objects\x18\x01 \x01(\x08R\x16\x65ncodeStructsAsObjects"\xcd\x01\n\x13OnlineQueryResponse\x12\x36\n\x04\x64\x61ta\x18\x01 \x01(\x0b\x32".chalk.common.v1.OnlineQueryResultR\x04\x64\x61ta\x12\x33\n\x06\x65rrors\x18\x02 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors\x12I\n\rresponse_meta\x18\x03 \x01(\x0b\x32$.chalk.common.v1.OnlineQueryMetadataR\x0cresponseMeta"\xd6\x02\n\x17OnlineQueryBulkResponse\x12!\n\x0cscalars_data\x18\x01 \x01(\x0cR\x0bscalarsData\x12Y\n\x0bgroups_data\x18\x02 \x03(\x0b\x32\x38.chalk.common.v1.OnlineQueryBulkResponse.GroupsDataEntryR\ngroupsData\x12\x33\n\x06\x65rrors\x18\x03 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors\x12I\n\rresponse_meta\x18\x04 \x01(\x0b\x32$.chalk.common.v1.OnlineQueryMetadataR\x0cresponseMeta\x1a=\n\x0fGroupsDataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\x0cR\x05value:\x02\x38\x01"\xc2\x01\n\x15GenericSingleResponse\x12O\n\x0fsingle_response\x18\x01 \x01(\x0b\x32$.chalk.common.v1.OnlineQueryResponseH\x00R\x0esingleResponse\x12O\n\rbulk_response\x18\x02 \x01(\x0b\x32(.chalk.common.v1.OnlineQueryBulkResponseH\x00R\x0c\x62ulkResponseB\x07\n\x05query"\x95\x01\n\x18OnlineQueryMultiResponse\x12\x44\n\tresponses\x18\x01 \x03(\x0b\x32&.chalk.common.v1.GenericSingleResponseR\tresponses\x12\x33\n\x06\x65rrors\x18\x02 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors"M\n\x11OnlineQueryResult\x12\x38\n\x07results\x18\x01 \x03(\x0b\x32\x1e.chalk.common.v1.FeatureResultR\x07results"\xb9\x02\n\rFeatureResult\x12\x14\n\x05\x66ield\x18\x01 \x01(\tR\x05\x66ield\x12*\n\x04pkey\x18\x06 \x01(\x0b\x32\x16.google.protobuf.ValueR\x04pkey\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value\x12\x36\n\x05\x65rror\x18\x03 \x01(\x0b\x32\x1b.chalk.common.v1.ChalkErrorH\x00R\x05\x65rror\x88\x01\x01\x12/\n\x02ts\x18\x04 \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x01R\x02ts\x88\x01\x01\x12\x35\n\x04meta\x18\x05 \x01(\x0b\x32\x1c.chalk.common.v1.FeatureMetaH\x02R\x04meta\x88\x01\x01\x42\x08\n\x06_errorB\x05\n\x03_tsB\x07\n\x05_meta"\x9b\x01\n\x0b\x46\x65\x61tureMeta\x12.\n\x13\x63hosen_resolver_fqn\x18\x01 \x01(\tR\x11\x63hosenResolverFqn\x12\x1b\n\tcache_hit\x18\x02 \x01(\x08R\x08\x63\x61\x63heHit\x12%\n\x0eprimitive_type\x18\x03 \x01(\tR\rprimitiveType\x12\x18\n\x07version\x18\x04 \x01(\x03R\x07version"\xfa\x05\n\x13OnlineQueryMetadata\x12H\n\x12\x65xecution_duration\x18\x01 \x01(\x0b\x32\x19.google.protobuf.DurationR\x11\x65xecutionDuration\x12#\n\rdeployment_id\x18\x02 \x01(\tR\x0c\x64\x65ploymentId\x12%\n\x0e\x65nvironment_id\x18\x03 \x01(\tR\renvironmentId\x12)\n\x10\x65nvironment_name\x18\x04 \x01(\tR\x0f\x65nvironmentName\x12\x19\n\x08query_id\x18\x05 \x01(\tR\x07queryId\x12\x43\n\x0fquery_timestamp\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\x0equeryTimestamp\x12\x1d\n\nquery_hash\x18\x07 \x01(\tR\tqueryHash\x12H\n\x0e\x65xplain_output\x18\x08 \x01(\x0b\x32!.chalk.common.v1.QueryExplainInfoR\rexplainOutput\x12N\n\x08metadata\x18\t \x03(\x0b\x32\x32.chalk.common.v1.OnlineQueryMetadata.MetadataEntryR\x08metadata\x12m\n\x13\x61\x64\x64itional_metadata\x18\n \x03(\x0b\x32<.chalk.common.v1.OnlineQueryMetadata.AdditionalMetadataEntryR\x12\x61\x64\x64itionalMetadata\x1a;\n\rMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x1a]\n\x17\x41\x64\x64itionalMetadataEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12,\n\x05value\x18\x02 \x01(\x0b\x32\x16.google.protobuf.ValueR\x05value:\x02\x38\x01"H\n\x10QueryExplainInfo\x12$\n\x0bplan_string\x18\x01 \x01(\tH\x00R\nplanString\x88\x01\x01\x42\x0e\n\x0c_plan_string"\x81\x01\n\x19UploadFeaturesBulkRequest\x12%\n\x0einputs_feather\x18\x01 \x01(\x0cR\rinputsFeather\x12=\n\tbody_type\x18\x07 \x01(\x0e\x32 .chalk.common.v1.FeatherBodyTypeR\x08\x62odyType"Q\n\x1aUploadFeaturesBulkResponse\x12\x33\n\x06\x65rrors\x18\x01 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors*w\n\x0f\x46\x65\x61therBodyType\x12!\n\x1d\x46\x45\x41THER_BODY_TYPE_UNSPECIFIED\x10\x00\x12\x1b\n\x17\x46\x45\x41THER_BODY_TYPE_TABLE\x10\x01\x12$\n FEATHER_BODY_TYPE_RECORD_BATCHES\x10\x02\x42\x85\x01\n\x13\x63om.chalk.common.v1B\x10OnlineQueryProtoP\x01\xa2\x02\x03\x43\x43X\xaa\x02\x0f\x43halk.Common.V1\xca\x02\x0f\x43halk\\Common\\V1\xe2\x02\x1b\x43halk\\Common\\V1\\GPBMetadata\xea\x02\x11\x43halk::Common::V1b\x06proto3'
 )
 
 _globals = globals()
@@ -57,8 +57,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_ONLINEQUERYMETADATA_METADATAENTRY"]._serialized_options = b"8\001"
     _globals["_ONLINEQUERYMETADATA_ADDITIONALMETADATAENTRY"]._options = None
     _globals["_ONLINEQUERYMETADATA_ADDITIONALMETADATAENTRY"]._serialized_options = b"8\001"
-    _globals["_FEATHERBODYTYPE"]._serialized_start = 5879
-    _globals["_FEATHERBODYTYPE"]._serialized_end = 5998
+    _globals["_FEATHERBODYTYPE"]._serialized_start = 5924
+    _globals["_FEATHERBODYTYPE"]._serialized_end = 6043
     _globals["_ONLINEQUERYREQUEST"]._serialized_start = 252
     _globals["_ONLINEQUERYREQUEST"]._serialized_end = 837
     _globals["_ONLINEQUERYREQUEST_INPUTSENTRY"]._serialized_start = 686
@@ -66,57 +66,57 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_ONLINEQUERYREQUEST_STALENESSENTRY"]._serialized_start = 769
     _globals["_ONLINEQUERYREQUEST_STALENESSENTRY"]._serialized_end = 829
     _globals["_ONLINEQUERYBULKREQUEST"]._serialized_start = 840
-    _globals["_ONLINEQUERYBULKREQUEST"]._serialized_end = 1366
+    _globals["_ONLINEQUERYBULKREQUEST"]._serialized_end = 1411
     _globals["_ONLINEQUERYBULKREQUEST_STALENESSENTRY"]._serialized_start = 769
     _globals["_ONLINEQUERYBULKREQUEST_STALENESSENTRY"]._serialized_end = 829
-    _globals["_GENERICSINGLEQUERY"]._serialized_start = 1369
-    _globals["_GENERICSINGLEQUERY"]._serialized_end = 1554
-    _globals["_ONLINEQUERYMULTIREQUEST"]._serialized_start = 1556
-    _globals["_ONLINEQUERYMULTIREQUEST"]._serialized_end = 1644
-    _globals["_FEATUREEXPRESSION"]._serialized_start = 1647
-    _globals["_FEATUREEXPRESSION"]._serialized_end = 1800
-    _globals["_OUTPUTEXPR"]._serialized_start = 1803
-    _globals["_OUTPUTEXPR"]._serialized_end = 1943
-    _globals["_ONLINEQUERYCONTEXT"]._serialized_start = 1946
-    _globals["_ONLINEQUERYCONTEXT"]._serialized_end = 2908
-    _globals["_ONLINEQUERYCONTEXT_OPTIONSENTRY"]._serialized_start = 2630
-    _globals["_ONLINEQUERYCONTEXT_OPTIONSENTRY"]._serialized_end = 2712
-    _globals["_ONLINEQUERYCONTEXT_QUERYCONTEXTENTRY"]._serialized_start = 2714
-    _globals["_ONLINEQUERYCONTEXT_QUERYCONTEXTENTRY"]._serialized_end = 2801
-    _globals["_ONLINEQUERYRESPONSEOPTIONS"]._serialized_start = 2911
-    _globals["_ONLINEQUERYRESPONSEOPTIONS"]._serialized_end = 3265
-    _globals["_ONLINEQUERYRESPONSEOPTIONS_METADATAENTRY"]._serialized_start = 3206
-    _globals["_ONLINEQUERYRESPONSEOPTIONS_METADATAENTRY"]._serialized_end = 3265
-    _globals["_EXPLAINOPTIONS"]._serialized_start = 3267
-    _globals["_EXPLAINOPTIONS"]._serialized_end = 3283
-    _globals["_FEATUREENCODINGOPTIONS"]._serialized_start = 3285
-    _globals["_FEATUREENCODINGOPTIONS"]._serialized_end = 3368
-    _globals["_ONLINEQUERYRESPONSE"]._serialized_start = 3371
-    _globals["_ONLINEQUERYRESPONSE"]._serialized_end = 3576
-    _globals["_ONLINEQUERYBULKRESPONSE"]._serialized_start = 3579
-    _globals["_ONLINEQUERYBULKRESPONSE"]._serialized_end = 3921
-    _globals["_ONLINEQUERYBULKRESPONSE_GROUPSDATAENTRY"]._serialized_start = 3860
-    _globals["_ONLINEQUERYBULKRESPONSE_GROUPSDATAENTRY"]._serialized_end = 3921
-    _globals["_GENERICSINGLERESPONSE"]._serialized_start = 3924
-    _globals["_GENERICSINGLERESPONSE"]._serialized_end = 4118
-    _globals["_ONLINEQUERYMULTIRESPONSE"]._serialized_start = 4121
-    _globals["_ONLINEQUERYMULTIRESPONSE"]._serialized_end = 4270
-    _globals["_ONLINEQUERYRESULT"]._serialized_start = 4272
-    _globals["_ONLINEQUERYRESULT"]._serialized_end = 4349
-    _globals["_FEATURERESULT"]._serialized_start = 4352
-    _globals["_FEATURERESULT"]._serialized_end = 4665
-    _globals["_FEATUREMETA"]._serialized_start = 4668
-    _globals["_FEATUREMETA"]._serialized_end = 4823
-    _globals["_ONLINEQUERYMETADATA"]._serialized_start = 4826
-    _globals["_ONLINEQUERYMETADATA"]._serialized_end = 5588
-    _globals["_ONLINEQUERYMETADATA_METADATAENTRY"]._serialized_start = 3206
-    _globals["_ONLINEQUERYMETADATA_METADATAENTRY"]._serialized_end = 3265
-    _globals["_ONLINEQUERYMETADATA_ADDITIONALMETADATAENTRY"]._serialized_start = 5495
-    _globals["_ONLINEQUERYMETADATA_ADDITIONALMETADATAENTRY"]._serialized_end = 5588
-    _globals["_QUERYEXPLAININFO"]._serialized_start = 5590
-    _globals["_QUERYEXPLAININFO"]._serialized_end = 5662
-    _globals["_UPLOADFEATURESBULKREQUEST"]._serialized_start = 5665
-    _globals["_UPLOADFEATURESBULKREQUEST"]._serialized_end = 5794
-    _globals["_UPLOADFEATURESBULKRESPONSE"]._serialized_start = 5796
-    _globals["_UPLOADFEATURESBULKRESPONSE"]._serialized_end = 5877
+    _globals["_GENERICSINGLEQUERY"]._serialized_start = 1414
+    _globals["_GENERICSINGLEQUERY"]._serialized_end = 1599
+    _globals["_ONLINEQUERYMULTIREQUEST"]._serialized_start = 1601
+    _globals["_ONLINEQUERYMULTIREQUEST"]._serialized_end = 1689
+    _globals["_FEATUREEXPRESSION"]._serialized_start = 1692
+    _globals["_FEATUREEXPRESSION"]._serialized_end = 1845
+    _globals["_OUTPUTEXPR"]._serialized_start = 1848
+    _globals["_OUTPUTEXPR"]._serialized_end = 1988
+    _globals["_ONLINEQUERYCONTEXT"]._serialized_start = 1991
+    _globals["_ONLINEQUERYCONTEXT"]._serialized_end = 2953
+    _globals["_ONLINEQUERYCONTEXT_OPTIONSENTRY"]._serialized_start = 2675
+    _globals["_ONLINEQUERYCONTEXT_OPTIONSENTRY"]._serialized_end = 2757
+    _globals["_ONLINEQUERYCONTEXT_QUERYCONTEXTENTRY"]._serialized_start = 2759
+    _globals["_ONLINEQUERYCONTEXT_QUERYCONTEXTENTRY"]._serialized_end = 2846
+    _globals["_ONLINEQUERYRESPONSEOPTIONS"]._serialized_start = 2956
+    _globals["_ONLINEQUERYRESPONSEOPTIONS"]._serialized_end = 3310
+    _globals["_ONLINEQUERYRESPONSEOPTIONS_METADATAENTRY"]._serialized_start = 3251
+    _globals["_ONLINEQUERYRESPONSEOPTIONS_METADATAENTRY"]._serialized_end = 3310
+    _globals["_EXPLAINOPTIONS"]._serialized_start = 3312
+    _globals["_EXPLAINOPTIONS"]._serialized_end = 3328
+    _globals["_FEATUREENCODINGOPTIONS"]._serialized_start = 3330
+    _globals["_FEATUREENCODINGOPTIONS"]._serialized_end = 3413
+    _globals["_ONLINEQUERYRESPONSE"]._serialized_start = 3416
+    _globals["_ONLINEQUERYRESPONSE"]._serialized_end = 3621
+    _globals["_ONLINEQUERYBULKRESPONSE"]._serialized_start = 3624
+    _globals["_ONLINEQUERYBULKRESPONSE"]._serialized_end = 3966
+    _globals["_ONLINEQUERYBULKRESPONSE_GROUPSDATAENTRY"]._serialized_start = 3905
+    _globals["_ONLINEQUERYBULKRESPONSE_GROUPSDATAENTRY"]._serialized_end = 3966
+    _globals["_GENERICSINGLERESPONSE"]._serialized_start = 3969
+    _globals["_GENERICSINGLERESPONSE"]._serialized_end = 4163
+    _globals["_ONLINEQUERYMULTIRESPONSE"]._serialized_start = 4166
+    _globals["_ONLINEQUERYMULTIRESPONSE"]._serialized_end = 4315
+    _globals["_ONLINEQUERYRESULT"]._serialized_start = 4317
+    _globals["_ONLINEQUERYRESULT"]._serialized_end = 4394
+    _globals["_FEATURERESULT"]._serialized_start = 4397
+    _globals["_FEATURERESULT"]._serialized_end = 4710
+    _globals["_FEATUREMETA"]._serialized_start = 4713
+    _globals["_FEATUREMETA"]._serialized_end = 4868
+    _globals["_ONLINEQUERYMETADATA"]._serialized_start = 4871
+    _globals["_ONLINEQUERYMETADATA"]._serialized_end = 5633
+    _globals["_ONLINEQUERYMETADATA_METADATAENTRY"]._serialized_start = 3251
+    _globals["_ONLINEQUERYMETADATA_METADATAENTRY"]._serialized_end = 3310
+    _globals["_ONLINEQUERYMETADATA_ADDITIONALMETADATAENTRY"]._serialized_start = 5540
+    _globals["_ONLINEQUERYMETADATA_ADDITIONALMETADATAENTRY"]._serialized_end = 5633
+    _globals["_QUERYEXPLAININFO"]._serialized_start = 5635
+    _globals["_QUERYEXPLAININFO"]._serialized_end = 5707
+    _globals["_UPLOADFEATURESBULKREQUEST"]._serialized_start = 5710
+    _globals["_UPLOADFEATURESBULKREQUEST"]._serialized_end = 5839
+    _globals["_UPLOADFEATURESBULKRESPONSE"]._serialized_start = 5841
+    _globals["_UPLOADFEATURESBULKRESPONSE"]._serialized_end = 5922
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/common/v1/online_query_pb2.pyi

@@ -71,7 +71,16 @@ class OnlineQueryRequest(_message.Message):
     ) -> None: ...
 
 class OnlineQueryBulkRequest(_message.Message):
-    __slots__ = ("inputs_feather", "outputs", "now", "staleness", "context", "response_options", "body_type")
+    __slots__ = (
+        "inputs_feather",
+        "inputs_sql",
+        "outputs",
+        "now",
+        "staleness",
+        "context",
+        "response_options",
+        "body_type",
+    )
     class StalenessEntry(_message.Message):
         __slots__ = ("key", "value")
         KEY_FIELD_NUMBER: _ClassVar[int]
@@ -81,6 +90,7 @@ class OnlineQueryBulkRequest(_message.Message):
         def __init__(self, key: _Optional[str] = ..., value: _Optional[str] = ...) -> None: ...
 
     INPUTS_FEATHER_FIELD_NUMBER: _ClassVar[int]
+    INPUTS_SQL_FIELD_NUMBER: _ClassVar[int]
     OUTPUTS_FIELD_NUMBER: _ClassVar[int]
     NOW_FIELD_NUMBER: _ClassVar[int]
     STALENESS_FIELD_NUMBER: _ClassVar[int]
@@ -88,6 +98,7 @@ class OnlineQueryBulkRequest(_message.Message):
     RESPONSE_OPTIONS_FIELD_NUMBER: _ClassVar[int]
     BODY_TYPE_FIELD_NUMBER: _ClassVar[int]
     inputs_feather: bytes
+    inputs_sql: str
     outputs: _containers.RepeatedCompositeFieldContainer[OutputExpr]
     now: _containers.RepeatedCompositeFieldContainer[_timestamp_pb2.Timestamp]
     staleness: _containers.ScalarMap[str, str]
@@ -97,6 +108,7 @@ class OnlineQueryBulkRequest(_message.Message):
     def __init__(
         self,
         inputs_feather: _Optional[bytes] = ...,
+        inputs_sql: _Optional[str] = ...,
         outputs: _Optional[_Iterable[_Union[OutputExpr, _Mapping]]] = ...,
         now: _Optional[_Iterable[_Union[_timestamp_pb2.Timestamp, _Mapping]]] = ...,
         staleness: _Optional[_Mapping[str, str]] = ...,

chalk/_gen/chalk/common/v1/script_task_pb2.py

@@ -17,7 +17,7 @@ from chalk._gen.chalk.common.v1 import resources_pb2 as chalk_dot_common_dot_v1_
 
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n!chalk/common/v1/script_task.proto\x12\x0f\x63halk.common.v1\x1a\x1f\x63halk/common/v1/resources.proto"\xc2\x06\n\x11ScriptTaskRequest\x12\x36\n\x17\x66unction_reference_type\x18\x01 \x01(\tR\x15\x66unctionReferenceType\x12-\n\x12\x66unction_reference\x18\x02 \x01(\tR\x11\x66unctionReference\x12*\n\x0e\x61rguments_json\x18\x0b \x01(\tH\x00R\rargumentsJson\x88\x01\x01\x12\x33\n\x04kind\x18\x0c \x01(\x0e\x32\x1f.chalk.common.v1.ScriptTaskKindR\x04kind\x12"\n\nsource_key\x18\x03 \x01(\tH\x01R\tsourceKey\x88\x01\x01\x12\x1b\n\x06\x62ranch\x18\x04 \x01(\tH\x02R\x06\x62ranch\x88\x01\x01\x12*\n\x0eresource_group\x18\x05 \x01(\tH\x03R\rresourceGroup\x88\x01\x01\x12W\n\x11resource_requests\x18\x06 \x01(\x0b\x32%.chalk.common.v1.ResourceRequirementsH\x04R\x10resourceRequests\x88\x01\x01\x12Y\n\renv_overrides\x18\x07 \x03(\x0b\x32\x34.chalk.common.v1.ScriptTaskRequest.EnvOverridesEntryR\x0c\x65nvOverrides\x12)\n\x10\x65nable_profiling\x18\x08 \x01(\x08R\x0f\x65nableProfiling\x12$\n\x0bmax_retries\x18\t \x01(\x05H\x05R\nmaxRetries\x88\x01\x01\x12\x34\n\x13\x63ompletion_deadline\x18\n \x01(\tH\x06R\x12\x63ompletionDeadline\x88\x01\x01\x1a?\n\x11\x45nvOverridesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x11\n\x0f_arguments_jsonB\r\n\x0b_source_keyB\t\n\x07_branchB\x11\n\x0f_resource_groupB\x14\n\x12_resource_requestsB\x0e\n\x0c_max_retriesB\x16\n\x14_completion_deadline"\x88\x01\n\x10ScriptTaskFilter\x12=\n\x08statuses\x18\x01 \x03(\x0e\x32!.chalk.common.v1.ScriptTaskStatusR\x08statuses\x12\x35\n\x05kinds\x18\x02 \x03(\x0e\x32\x1f.chalk.common.v1.ScriptTaskKindR\x05kinds*\xd7\x01\n\x10ScriptTaskStatus\x12"\n\x1eSCRIPT_TASK_STATUS_UNSPECIFIED\x10\x00\x12\x1d\n\x19SCRIPT_TASK_STATUS_QUEUED\x10\x01\x12\x1e\n\x1aSCRIPT_TASK_STATUS_WORKING\x10\x02\x12 \n\x1cSCRIPT_TASK_STATUS_COMPLETED\x10\x03\x12\x1d\n\x19SCRIPT_TASK_STATUS_FAILED\x10\x04\x12\x1f\n\x1bSCRIPT_TASK_STATUS_CANCELED\x10\x05*y\n\x0eScriptTaskKind\x12 \n\x1cSCRIPT_TASK_KIND_UNSPECIFIED\x10\x00\x12"\n\x1eSCRIPT_TASK_KIND_PYTHON_SCRIPT\x10\x01\x12!\n\x1dSCRIPT_TASK_KIND_TRAINING_RUN\x10\x02\x42\x84\x01\n\x13\x63om.chalk.common.v1B\x0fScriptTaskProtoP\x01\xa2\x02\x03\x43\x43X\xaa\x02\x0f\x43halk.Common.V1\xca\x02\x0f\x43halk\\Common\\V1\xe2\x02\x1b\x43halk\\Common\\V1\\GPBMetadata\xea\x02\x11\x43halk::Common::V1b\x06proto3'
+    b'\n!chalk/common/v1/script_task.proto\x12\x0f\x63halk.common.v1\x1a\x1f\x63halk/common/v1/resources.proto":\n\x0fTrainingRunArgs\x12\'\n\x0f\x65xperiment_name\x18\x01 \x01(\tR\x0e\x65xperimentName"\x96\x07\n\x11ScriptTaskRequest\x12\x36\n\x17\x66unction_reference_type\x18\x01 \x01(\tR\x15\x66unctionReferenceType\x12-\n\x12\x66unction_reference\x18\x02 \x01(\tR\x11\x66unctionReference\x12*\n\x0e\x61rguments_json\x18\x0b \x01(\tH\x01R\rargumentsJson\x88\x01\x01\x12\x33\n\x04kind\x18\x0c \x01(\x0e\x32\x1f.chalk.common.v1.ScriptTaskKindR\x04kind\x12"\n\nsource_key\x18\x03 \x01(\tH\x02R\tsourceKey\x88\x01\x01\x12\x1b\n\x06\x62ranch\x18\x04 \x01(\tH\x03R\x06\x62ranch\x88\x01\x01\x12*\n\x0eresource_group\x18\x05 \x01(\tH\x04R\rresourceGroup\x88\x01\x01\x12W\n\x11resource_requests\x18\x06 \x01(\x0b\x32%.chalk.common.v1.ResourceRequirementsH\x05R\x10resourceRequests\x88\x01\x01\x12Y\n\renv_overrides\x18\x07 \x03(\x0b\x32\x34.chalk.common.v1.ScriptTaskRequest.EnvOverridesEntryR\x0c\x65nvOverrides\x12)\n\x10\x65nable_profiling\x18\x08 \x01(\x08R\x0f\x65nableProfiling\x12$\n\x0bmax_retries\x18\t \x01(\x05H\x06R\nmaxRetries\x88\x01\x01\x12\x34\n\x13\x63ompletion_deadline\x18\n \x01(\tH\x07R\x12\x63ompletionDeadline\x88\x01\x01\x12\x45\n\x0ctraining_run\x18\r \x01(\x0b\x32 .chalk.common.v1.TrainingRunArgsH\x00R\x0btrainingRun\x1a?\n\x11\x45nvOverridesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x0b\n\tkind_argsB\x11\n\x0f_arguments_jsonB\r\n\x0b_source_keyB\t\n\x07_branchB\x11\n\x0f_resource_groupB\x14\n\x12_resource_requestsB\x0e\n\x0c_max_retriesB\x16\n\x14_completion_deadline"\xe8\x01\n\x10ScriptTaskFilter\x12=\n\x08statuses\x18\x01 \x03(\x0e\x32!.chalk.common.v1.ScriptTaskStatusR\x08statuses\x12\x35\n\x05kinds\x18\x02 \x03(\x0e\x32\x1f.chalk.common.v1.ScriptTaskKindR\x05kinds\x12\x1c\n\x07task_id\x18\x03 \x01(\tH\x00R\x06taskId\x88\x01\x01\x12$\n\x0b\x62ranch_name\x18\x04 \x01(\tH\x01R\nbranchName\x88\x01\x01\x42\n\n\x08_task_idB\x0e\n\x0c_branch_name*\xd7\x01\n\x10ScriptTaskStatus\x12"\n\x1eSCRIPT_TASK_STATUS_UNSPECIFIED\x10\x00\x12\x1d\n\x19SCRIPT_TASK_STATUS_QUEUED\x10\x01\x12\x1e\n\x1aSCRIPT_TASK_STATUS_WORKING\x10\x02\x12 \n\x1cSCRIPT_TASK_STATUS_COMPLETED\x10\x03\x12\x1d\n\x19SCRIPT_TASK_STATUS_FAILED\x10\x04\x12\x1f\n\x1bSCRIPT_TASK_STATUS_CANCELED\x10\x05*\x98\x01\n\x0eScriptTaskKind\x12 \n\x1cSCRIPT_TASK_KIND_UNSPECIFIED\x10\x00\x12"\n\x1eSCRIPT_TASK_KIND_PYTHON_SCRIPT\x10\x01\x12!\n\x1dSCRIPT_TASK_KIND_TRAINING_RUN\x10\x02\x12\x1d\n\x19SCRIPT_TASK_KIND_METAPLAN\x10\x03\x42\x84\x01\n\x13\x63om.chalk.common.v1B\x0fScriptTaskProtoP\x01\xa2\x02\x03\x43\x43X\xaa\x02\x0f\x43halk.Common.V1\xca\x02\x0f\x43halk\\Common\\V1\xe2\x02\x1b\x43halk\\Common\\V1\\GPBMetadata\xea\x02\x11\x43halk::Common::V1b\x06proto3'
 )
 
 _globals = globals()
@@ -30,14 +30,16 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     ]._serialized_options = b"\n\023com.chalk.common.v1B\017ScriptTaskProtoP\001\242\002\003CCX\252\002\017Chalk.Common.V1\312\002\017Chalk\\Common\\V1\342\002\033Chalk\\Common\\V1\\GPBMetadata\352\002\021Chalk::Common::V1"
     _globals["_SCRIPTTASKREQUEST_ENVOVERRIDESENTRY"]._options = None
     _globals["_SCRIPTTASKREQUEST_ENVOVERRIDESENTRY"]._serialized_options = b"8\001"
-    _globals["_SCRIPTTASKSTATUS"]._serialized_start = 1064
-    _globals["_SCRIPTTASKSTATUS"]._serialized_end = 1279
-    _globals["_SCRIPTTASKKIND"]._serialized_start = 1281
-    _globals["_SCRIPTTASKKIND"]._serialized_end = 1402
-    _globals["_SCRIPTTASKREQUEST"]._serialized_start = 88
-    _globals["_SCRIPTTASKREQUEST"]._serialized_end = 922
-    _globals["_SCRIPTTASKREQUEST_ENVOVERRIDESENTRY"]._serialized_start = 733
-    _globals["_SCRIPTTASKREQUEST_ENVOVERRIDESENTRY"]._serialized_end = 796
-    _globals["_SCRIPTTASKFILTER"]._serialized_start = 925
-    _globals["_SCRIPTTASKFILTER"]._serialized_end = 1061
+    _globals["_SCRIPTTASKSTATUS"]._serialized_start = 1304
+    _globals["_SCRIPTTASKSTATUS"]._serialized_end = 1519
+    _globals["_SCRIPTTASKKIND"]._serialized_start = 1522
+    _globals["_SCRIPTTASKKIND"]._serialized_end = 1674
+    _globals["_TRAININGRUNARGS"]._serialized_start = 87
+    _globals["_TRAININGRUNARGS"]._serialized_end = 145
+    _globals["_SCRIPTTASKREQUEST"]._serialized_start = 148
+    _globals["_SCRIPTTASKREQUEST"]._serialized_end = 1066
+    _globals["_SCRIPTTASKREQUEST_ENVOVERRIDESENTRY"]._serialized_start = 864
+    _globals["_SCRIPTTASKREQUEST_ENVOVERRIDESENTRY"]._serialized_end = 927
+    _globals["_SCRIPTTASKFILTER"]._serialized_start = 1069
+    _globals["_SCRIPTTASKFILTER"]._serialized_end = 1301
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/common/v1/script_task_pb2.pyi

@@ -27,6 +27,7 @@ class ScriptTaskKind(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     SCRIPT_TASK_KIND_UNSPECIFIED: _ClassVar[ScriptTaskKind]
     SCRIPT_TASK_KIND_PYTHON_SCRIPT: _ClassVar[ScriptTaskKind]
     SCRIPT_TASK_KIND_TRAINING_RUN: _ClassVar[ScriptTaskKind]
+    SCRIPT_TASK_KIND_METAPLAN: _ClassVar[ScriptTaskKind]
 
 SCRIPT_TASK_STATUS_UNSPECIFIED: ScriptTaskStatus
 SCRIPT_TASK_STATUS_QUEUED: ScriptTaskStatus
@@ -37,6 +38,13 @@ SCRIPT_TASK_STATUS_CANCELED: ScriptTaskStatus
 SCRIPT_TASK_KIND_UNSPECIFIED: ScriptTaskKind
 SCRIPT_TASK_KIND_PYTHON_SCRIPT: ScriptTaskKind
 SCRIPT_TASK_KIND_TRAINING_RUN: ScriptTaskKind
+SCRIPT_TASK_KIND_METAPLAN: ScriptTaskKind
+
+class TrainingRunArgs(_message.Message):
+    __slots__ = ("experiment_name",)
+    EXPERIMENT_NAME_FIELD_NUMBER: _ClassVar[int]
+    experiment_name: str
+    def __init__(self, experiment_name: _Optional[str] = ...) -> None: ...
 
 class ScriptTaskRequest(_message.Message):
     __slots__ = (
@@ -52,6 +60,7 @@ class ScriptTaskRequest(_message.Message):
         "enable_profiling",
         "max_retries",
         "completion_deadline",
+        "training_run",
     )
     class EnvOverridesEntry(_message.Message):
         __slots__ = ("key", "value")
@@ -73,6 +82,7 @@ class ScriptTaskRequest(_message.Message):
     ENABLE_PROFILING_FIELD_NUMBER: _ClassVar[int]
     MAX_RETRIES_FIELD_NUMBER: _ClassVar[int]
     COMPLETION_DEADLINE_FIELD_NUMBER: _ClassVar[int]
+    TRAINING_RUN_FIELD_NUMBER: _ClassVar[int]
     function_reference_type: str
     function_reference: str
     arguments_json: str
@@ -85,6 +95,7 @@ class ScriptTaskRequest(_message.Message):
     enable_profiling: bool
     max_retries: int
     completion_deadline: str
+    training_run: TrainingRunArgs
     def __init__(
         self,
         function_reference_type: _Optional[str] = ...,
@@ -99,16 +110,23 @@ class ScriptTaskRequest(_message.Message):
         enable_profiling: bool = ...,
         max_retries: _Optional[int] = ...,
         completion_deadline: _Optional[str] = ...,
+        training_run: _Optional[_Union[TrainingRunArgs, _Mapping]] = ...,
     ) -> None: ...
 
 class ScriptTaskFilter(_message.Message):
-    __slots__ = ("statuses", "kinds")
+    __slots__ = ("statuses", "kinds", "task_id", "branch_name")
     STATUSES_FIELD_NUMBER: _ClassVar[int]
     KINDS_FIELD_NUMBER: _ClassVar[int]
+    TASK_ID_FIELD_NUMBER: _ClassVar[int]
+    BRANCH_NAME_FIELD_NUMBER: _ClassVar[int]
     statuses: _containers.RepeatedScalarFieldContainer[ScriptTaskStatus]
     kinds: _containers.RepeatedScalarFieldContainer[ScriptTaskKind]
+    task_id: str
+    branch_name: str
     def __init__(
         self,
         statuses: _Optional[_Iterable[_Union[ScriptTaskStatus, str]]] = ...,
         kinds: _Optional[_Iterable[_Union[ScriptTaskKind, str]]] = ...,
+        task_id: _Optional[str] = ...,
+        branch_name: _Optional[str] = ...,
     ) -> None: ...

chalk/_gen/chalk/dataframe/v1/dataframe_pb2.py

@@ -0,0 +1,48 @@
+# -*- coding: utf-8 -*-
+# Generated by the protocol buffer compiler.  DO NOT EDIT!
+# source: chalk/dataframe/v1/dataframe.proto
+# Protobuf Python Version: 4.25.3
+"""Generated protocol buffer code."""
+
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import descriptor_pool as _descriptor_pool
+from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
+# @@protoc_insertion_point(imports)
+
+_sym_db = _symbol_database.Default()
+
+
+from chalk._gen.chalk.arrow.v1 import arrow_pb2 as chalk_dot_arrow_dot_v1_dot_arrow__pb2
+from chalk._gen.chalk.expression.v1 import expression_pb2 as chalk_dot_expression_dot_v1_dot_expression__pb2
+
+
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
+    b'\n"chalk/dataframe/v1/dataframe.proto\x12\x12\x63halk.dataframe.v1\x1a\x1a\x63halk/arrow/v1/arrow.proto\x1a$chalk/expression/v1/expression.proto"\xb4\x05\n\x10\x44\x61taFrameOperand\x12#\n\x0cvalue_string\x18\x01 \x01(\tH\x00R\x0bvalueString\x12\x1d\n\tvalue_int\x18\x02 \x01(\x03H\x00R\x08valueInt\x12\x1f\n\nvalue_bool\x18\x03 \x01(\x08H\x00R\tvalueBool\x12;\n\nvalue_none\x18\x04 \x01(\x0b\x32\x1a.chalk.dataframe.v1.PyNoneH\x00R\tvalueNone\x12;\n\nvalue_list\x18\x05 \x01(\x0b\x32\x1a.chalk.dataframe.v1.PyListH\x00R\tvalueList\x12;\n\nvalue_dict\x18\x06 \x01(\x0b\x32\x1a.chalk.dataframe.v1.PyDictH\x00R\tvalueDict\x12X\n\x15value_dataframe_index\x18\x07 \x01(\x0b\x32".chalk.dataframe.v1.DataFrameIndexH\x00R\x13valueDataframeIndex\x12;\n\x0c\x61rrow_schema\x18\x08 \x01(\x0b\x32\x16.chalk.arrow.v1.SchemaH\x00R\x0b\x61rrowSchema\x12\x44\n\x0b\x61rrow_table\x18\t \x01(\x0b\x32!.chalk.arrow.v1.TableParquetBytesH\x00R\narrowTable\x12O\n\x0funderscore_expr\x18\n \x01(\x0b\x32$.chalk.expression.v1.LogicalExprNodeH\x00R\x0eunderscoreExpr\x12K\n\rlibchalk_expr\x18\x0b \x01(\x0b\x32$.chalk.expression.v1.LogicalExprNodeH\x00R\x0clibchalkExprB\t\n\x07operand">\n\x0e\x44\x61taFrameIndex\x12,\n\x12\x64\x61taframe_op_index\x18\x01 \x01(\x05R\x10\x64\x61taframeOpIndex"\x08\n\x06PyNone"M\n\x06PyList\x12\x43\n\nlist_items\x18\x01 \x03(\x0b\x32$.chalk.dataframe.v1.DataFrameOperandR\tlistItems"\x97\x01\n\x0bPyDictEntry\x12\x41\n\tentry_key\x18\x01 \x01(\x0b\x32$.chalk.dataframe.v1.DataFrameOperandR\x08\x65ntryKey\x12\x45\n\x0b\x65ntry_value\x18\x02 \x01(\x0b\x32$.chalk.dataframe.v1.DataFrameOperandR\nentryValue"L\n\x06PyDict\x12\x42\n\x0c\x64ict_entries\x18\x01 \x03(\x0b\x32\x1f.chalk.dataframe.v1.PyDictEntryR\x0b\x64ictEntries"\xfc\x01\n\x14\x44\x61taFrameConstructor\x12J\n\x0cself_operand\x18\x01 \x01(\x0b\x32".chalk.dataframe.v1.DataFrameIndexH\x00R\x0bselfOperand\x88\x01\x01\x12#\n\rfunction_name\x18\x02 \x01(\tR\x0c\x66unctionName\x12.\n\x04\x61rgs\x18\x03 \x01(\x0b\x32\x1a.chalk.dataframe.v1.PyListR\x04\x61rgs\x12\x32\n\x06kwargs\x18\x04 \x01(\x0b\x32\x1a.chalk.dataframe.v1.PyDictR\x06kwargsB\x0f\n\r_self_operand"]\n\rDataFramePlan\x12L\n\x0c\x63onstructors\x18\x01 \x03(\x0b\x32(.chalk.dataframe.v1.DataFrameConstructorR\x0c\x63onstructorsB\x92\x01\n\x16\x63om.chalk.dataframe.v1B\x0e\x44\x61taframeProtoP\x01\xa2\x02\x03\x43\x44X\xaa\x02\x12\x43halk.Dataframe.V1\xca\x02\x12\x43halk\\Dataframe\\V1\xe2\x02\x1e\x43halk\\Dataframe\\V1\\GPBMetadata\xea\x02\x14\x43halk::Dataframe::V1b\x06proto3'
+)
+
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, "chalk.dataframe.v1.dataframe_pb2", _globals)
+if _descriptor._USE_C_DESCRIPTORS == False:
+    _globals["DESCRIPTOR"]._options = None
+    _globals[
+        "DESCRIPTOR"
+    ]._serialized_options = b"\n\026com.chalk.dataframe.v1B\016DataframeProtoP\001\242\002\003CDX\252\002\022Chalk.Dataframe.V1\312\002\022Chalk\\Dataframe\\V1\342\002\036Chalk\\Dataframe\\V1\\GPBMetadata\352\002\024Chalk::Dataframe::V1"
+    _globals["_DATAFRAMEOPERAND"]._serialized_start = 125
+    _globals["_DATAFRAMEOPERAND"]._serialized_end = 817
+    _globals["_DATAFRAMEINDEX"]._serialized_start = 819
+    _globals["_DATAFRAMEINDEX"]._serialized_end = 881
+    _globals["_PYNONE"]._serialized_start = 883
+    _globals["_PYNONE"]._serialized_end = 891
+    _globals["_PYLIST"]._serialized_start = 893
+    _globals["_PYLIST"]._serialized_end = 970
+    _globals["_PYDICTENTRY"]._serialized_start = 973
+    _globals["_PYDICTENTRY"]._serialized_end = 1124
+    _globals["_PYDICT"]._serialized_start = 1126
+    _globals["_PYDICT"]._serialized_end = 1202
+    _globals["_DATAFRAMECONSTRUCTOR"]._serialized_start = 1205
+    _globals["_DATAFRAMECONSTRUCTOR"]._serialized_end = 1457
+    _globals["_DATAFRAMEPLAN"]._serialized_start = 1459
+    _globals["_DATAFRAMEPLAN"]._serialized_end = 1552
+# @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/dataframe/v1/dataframe_pb2.pyi

@@ -0,0 +1,123 @@
+from chalk._gen.chalk.arrow.v1 import arrow_pb2 as _arrow_pb2
+from chalk._gen.chalk.expression.v1 import expression_pb2 as _expression_pb2
+from google.protobuf.internal import containers as _containers
+from google.protobuf import descriptor as _descriptor
+from google.protobuf import message as _message
+from typing import (
+    ClassVar as _ClassVar,
+    Iterable as _Iterable,
+    Mapping as _Mapping,
+    Optional as _Optional,
+    Union as _Union,
+)
+
+DESCRIPTOR: _descriptor.FileDescriptor
+
+class DataFrameOperand(_message.Message):
+    __slots__ = (
+        "value_string",
+        "value_int",
+        "value_bool",
+        "value_none",
+        "value_list",
+        "value_dict",
+        "value_dataframe_index",
+        "arrow_schema",
+        "arrow_table",
+        "underscore_expr",
+        "libchalk_expr",
+    )
+    VALUE_STRING_FIELD_NUMBER: _ClassVar[int]
+    VALUE_INT_FIELD_NUMBER: _ClassVar[int]
+    VALUE_BOOL_FIELD_NUMBER: _ClassVar[int]
+    VALUE_NONE_FIELD_NUMBER: _ClassVar[int]
+    VALUE_LIST_FIELD_NUMBER: _ClassVar[int]
+    VALUE_DICT_FIELD_NUMBER: _ClassVar[int]
+    VALUE_DATAFRAME_INDEX_FIELD_NUMBER: _ClassVar[int]
+    ARROW_SCHEMA_FIELD_NUMBER: _ClassVar[int]
+    ARROW_TABLE_FIELD_NUMBER: _ClassVar[int]
+    UNDERSCORE_EXPR_FIELD_NUMBER: _ClassVar[int]
+    LIBCHALK_EXPR_FIELD_NUMBER: _ClassVar[int]
+    value_string: str
+    value_int: int
+    value_bool: bool
+    value_none: PyNone
+    value_list: PyList
+    value_dict: PyDict
+    value_dataframe_index: DataFrameIndex
+    arrow_schema: _arrow_pb2.Schema
+    arrow_table: _arrow_pb2.TableParquetBytes
+    underscore_expr: _expression_pb2.LogicalExprNode
+    libchalk_expr: _expression_pb2.LogicalExprNode
+    def __init__(
+        self,
+        value_string: _Optional[str] = ...,
+        value_int: _Optional[int] = ...,
+        value_bool: bool = ...,
+        value_none: _Optional[_Union[PyNone, _Mapping]] = ...,
+        value_list: _Optional[_Union[PyList, _Mapping]] = ...,
+        value_dict: _Optional[_Union[PyDict, _Mapping]] = ...,
+        value_dataframe_index: _Optional[_Union[DataFrameIndex, _Mapping]] = ...,
+        arrow_schema: _Optional[_Union[_arrow_pb2.Schema, _Mapping]] = ...,
+        arrow_table: _Optional[_Union[_arrow_pb2.TableParquetBytes, _Mapping]] = ...,
+        underscore_expr: _Optional[_Union[_expression_pb2.LogicalExprNode, _Mapping]] = ...,
+        libchalk_expr: _Optional[_Union[_expression_pb2.LogicalExprNode, _Mapping]] = ...,
+    ) -> None: ...
+
+class DataFrameIndex(_message.Message):
+    __slots__ = ("dataframe_op_index",)
+    DATAFRAME_OP_INDEX_FIELD_NUMBER: _ClassVar[int]
+    dataframe_op_index: int
+    def __init__(self, dataframe_op_index: _Optional[int] = ...) -> None: ...
+
+class PyNone(_message.Message):
+    __slots__ = ()
+    def __init__(self) -> None: ...
+
+class PyList(_message.Message):
+    __slots__ = ("list_items",)
+    LIST_ITEMS_FIELD_NUMBER: _ClassVar[int]
+    list_items: _containers.RepeatedCompositeFieldContainer[DataFrameOperand]
+    def __init__(self, list_items: _Optional[_Iterable[_Union[DataFrameOperand, _Mapping]]] = ...) -> None: ...
+
+class PyDictEntry(_message.Message):
+    __slots__ = ("entry_key", "entry_value")
+    ENTRY_KEY_FIELD_NUMBER: _ClassVar[int]
+    ENTRY_VALUE_FIELD_NUMBER: _ClassVar[int]
+    entry_key: DataFrameOperand
+    entry_value: DataFrameOperand
+    def __init__(
+        self,
+        entry_key: _Optional[_Union[DataFrameOperand, _Mapping]] = ...,
+        entry_value: _Optional[_Union[DataFrameOperand, _Mapping]] = ...,
+    ) -> None: ...
+
+class PyDict(_message.Message):
+    __slots__ = ("dict_entries",)
+    DICT_ENTRIES_FIELD_NUMBER: _ClassVar[int]
+    dict_entries: _containers.RepeatedCompositeFieldContainer[PyDictEntry]
+    def __init__(self, dict_entries: _Optional[_Iterable[_Union[PyDictEntry, _Mapping]]] = ...) -> None: ...
+
+class DataFrameConstructor(_message.Message):
+    __slots__ = ("self_operand", "function_name", "args", "kwargs")
+    SELF_OPERAND_FIELD_NUMBER: _ClassVar[int]
+    FUNCTION_NAME_FIELD_NUMBER: _ClassVar[int]
+    ARGS_FIELD_NUMBER: _ClassVar[int]
+    KWARGS_FIELD_NUMBER: _ClassVar[int]
+    self_operand: DataFrameIndex
+    function_name: str
+    args: PyList
+    kwargs: PyDict
+    def __init__(
+        self,
+        self_operand: _Optional[_Union[DataFrameIndex, _Mapping]] = ...,
+        function_name: _Optional[str] = ...,
+        args: _Optional[_Union[PyList, _Mapping]] = ...,
+        kwargs: _Optional[_Union[PyDict, _Mapping]] = ...,
+    ) -> None: ...
+
+class DataFramePlan(_message.Message):
+    __slots__ = ("constructors",)
+    CONSTRUCTORS_FIELD_NUMBER: _ClassVar[int]
+    constructors: _containers.RepeatedCompositeFieldContainer[DataFrameConstructor]
+    def __init__(self, constructors: _Optional[_Iterable[_Union[DataFrameConstructor, _Mapping]]] = ...) -> None: ...

chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.py

@@ -0,0 +1,4 @@
+# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
+"""Client and server classes corresponding to protobuf-defined services."""
+
+import grpc

chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.pyi

@@ -0,0 +1,4 @@
+"""
+@generated by mypy-protobuf.  Do not edit manually!
+isort:skip_file
+"""