PyPI - chalkpy - Versions diffs - 2.90.1__py3-none-any.whl → 2.95.3__py3-none-any.whl - Mend

chalkpy 2.90.1py3-none-any.whl → 2.95.3py3-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.

Files changed (202) hide show

chalk/__init__.py +2 -1
chalk/_gen/chalk/arrow/v1/arrow_pb2.py +7 -5
chalk/_gen/chalk/arrow/v1/arrow_pb2.pyi +6 -0
chalk/_gen/chalk/artifacts/v1/chart_pb2.py +16 -16
chalk/_gen/chalk/artifacts/v1/chart_pb2.pyi +4 -0
chalk/_gen/chalk/artifacts/v1/cron_query_pb2.py +8 -7
chalk/_gen/chalk/artifacts/v1/cron_query_pb2.pyi +5 -0
chalk/_gen/chalk/common/v1/offline_query_pb2.py +17 -15
chalk/_gen/chalk/common/v1/offline_query_pb2.pyi +25 -0
chalk/_gen/chalk/common/v1/script_task_pb2.py +3 -3
chalk/_gen/chalk/common/v1/script_task_pb2.pyi +2 -0
chalk/_gen/chalk/dataframe/__init__.py +0 -0
chalk/_gen/chalk/dataframe/v1/__init__.py +0 -0
chalk/_gen/chalk/dataframe/v1/dataframe_pb2.py +48 -0
chalk/_gen/chalk/dataframe/v1/dataframe_pb2.pyi +123 -0
chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.py +4 -0
chalk/_gen/chalk/dataframe/v1/dataframe_pb2_grpc.pyi +4 -0
chalk/_gen/chalk/graph/v1/graph_pb2.py +150 -149
chalk/_gen/chalk/graph/v1/graph_pb2.pyi +25 -0
chalk/_gen/chalk/graph/v1/sources_pb2.py +94 -84
chalk/_gen/chalk/graph/v1/sources_pb2.pyi +56 -0
chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2.py +79 -0
chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2.pyi +377 -0
chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2_grpc.py +4 -0
chalk/_gen/chalk/kubernetes/v1/horizontalpodautoscaler_pb2_grpc.pyi +4 -0
chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.py +43 -7
chalk/_gen/chalk/kubernetes/v1/scaledobject_pb2.pyi +252 -2
chalk/_gen/chalk/protosql/v1/sql_service_pb2.py +54 -27
chalk/_gen/chalk/protosql/v1/sql_service_pb2.pyi +131 -3
chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.py +45 -0
chalk/_gen/chalk/protosql/v1/sql_service_pb2_grpc.pyi +14 -0
chalk/_gen/chalk/python/v1/types_pb2.py +14 -14
chalk/_gen/chalk/python/v1/types_pb2.pyi +8 -0
chalk/_gen/chalk/server/v1/benchmark_pb2.py +76 -0
chalk/_gen/chalk/server/v1/benchmark_pb2.pyi +156 -0
chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.py +258 -0
chalk/_gen/chalk/server/v1/benchmark_pb2_grpc.pyi +84 -0
chalk/_gen/chalk/server/v1/billing_pb2.py +40 -38
chalk/_gen/chalk/server/v1/billing_pb2.pyi +17 -1
chalk/_gen/chalk/server/v1/branches_pb2.py +45 -0
chalk/_gen/chalk/server/v1/branches_pb2.pyi +80 -0
chalk/_gen/chalk/server/v1/branches_pb2_grpc.pyi +36 -0
chalk/_gen/chalk/server/v1/builder_pb2.py +358 -288
chalk/_gen/chalk/server/v1/builder_pb2.pyi +360 -10
chalk/_gen/chalk/server/v1/builder_pb2_grpc.py +225 -0
chalk/_gen/chalk/server/v1/builder_pb2_grpc.pyi +60 -0
chalk/_gen/chalk/server/v1/chart_pb2.py +10 -10
chalk/_gen/chalk/server/v1/chart_pb2.pyi +18 -2
chalk/_gen/chalk/server/v1/clickhouse_pb2.py +42 -0
chalk/_gen/chalk/server/v1/clickhouse_pb2.pyi +17 -0
chalk/_gen/chalk/server/v1/clickhouse_pb2_grpc.py +78 -0
chalk/_gen/chalk/server/v1/clickhouse_pb2_grpc.pyi +38 -0
chalk/_gen/chalk/server/v1/cloud_components_pb2.py +141 -119
chalk/_gen/chalk/server/v1/cloud_components_pb2.pyi +106 -4
chalk/_gen/chalk/server/v1/cloud_components_pb2_grpc.py +45 -0
chalk/_gen/chalk/server/v1/cloud_components_pb2_grpc.pyi +12 -0
chalk/_gen/chalk/server/v1/cloud_credentials_pb2.py +11 -3
chalk/_gen/chalk/server/v1/cloud_credentials_pb2.pyi +20 -0
chalk/_gen/chalk/server/v1/cloud_credentials_pb2_grpc.py +45 -0
chalk/_gen/chalk/server/v1/cloud_credentials_pb2_grpc.pyi +12 -0
chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2.py +52 -38
chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2.pyi +62 -1
chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2_grpc.py +90 -0
chalk/_gen/chalk/server/v1/dataplanejobqueue_pb2_grpc.pyi +24 -0
chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.py +90 -0
chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2.pyi +264 -0
chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.py +170 -0
chalk/_gen/chalk/server/v1/dataplaneworkflows_pb2_grpc.pyi +62 -0
chalk/_gen/chalk/server/v1/deploy_pb2.py +9 -3
chalk/_gen/chalk/server/v1/deploy_pb2.pyi +12 -0
chalk/_gen/chalk/server/v1/deploy_pb2_grpc.py +45 -0
chalk/_gen/chalk/server/v1/deploy_pb2_grpc.pyi +12 -0
chalk/_gen/chalk/server/v1/deployment_pb2.py +6 -6
chalk/_gen/chalk/server/v1/deployment_pb2.pyi +20 -0
chalk/_gen/chalk/server/v1/environment_pb2.py +14 -12
chalk/_gen/chalk/server/v1/environment_pb2.pyi +19 -0
chalk/_gen/chalk/server/v1/eventbus_pb2.py +4 -2
chalk/_gen/chalk/server/v1/files_pb2.py +65 -0
chalk/_gen/chalk/server/v1/files_pb2.pyi +167 -0
chalk/_gen/chalk/server/v1/files_pb2_grpc.py +4 -0
chalk/_gen/chalk/server/v1/files_pb2_grpc.pyi +4 -0
chalk/_gen/chalk/server/v1/graph_pb2.py +38 -26
chalk/_gen/chalk/server/v1/graph_pb2.pyi +58 -0
chalk/_gen/chalk/server/v1/graph_pb2_grpc.py +47 -0
chalk/_gen/chalk/server/v1/graph_pb2_grpc.pyi +18 -0
chalk/_gen/chalk/server/v1/incident_pb2.py +23 -21
chalk/_gen/chalk/server/v1/incident_pb2.pyi +15 -1
chalk/_gen/chalk/server/v1/indexing_job_pb2.py +44 -0
chalk/_gen/chalk/server/v1/indexing_job_pb2.pyi +38 -0
chalk/_gen/chalk/server/v1/indexing_job_pb2_grpc.py +78 -0
chalk/_gen/chalk/server/v1/indexing_job_pb2_grpc.pyi +38 -0
chalk/_gen/chalk/server/v1/integrations_pb2.py +11 -9
chalk/_gen/chalk/server/v1/integrations_pb2.pyi +34 -2
chalk/_gen/chalk/server/v1/kube_pb2.py +29 -19
chalk/_gen/chalk/server/v1/kube_pb2.pyi +28 -0
chalk/_gen/chalk/server/v1/kube_pb2_grpc.py +45 -0
chalk/_gen/chalk/server/v1/kube_pb2_grpc.pyi +12 -0
chalk/_gen/chalk/server/v1/log_pb2.py +21 -3
chalk/_gen/chalk/server/v1/log_pb2.pyi +68 -0
chalk/_gen/chalk/server/v1/log_pb2_grpc.py +90 -0
chalk/_gen/chalk/server/v1/log_pb2_grpc.pyi +24 -0
chalk/_gen/chalk/server/v1/model_registry_pb2.py +10 -10
chalk/_gen/chalk/server/v1/model_registry_pb2.pyi +4 -1
chalk/_gen/chalk/server/v1/plandebug_pb2.py +53 -0
chalk/_gen/chalk/server/v1/plandebug_pb2.pyi +86 -0
chalk/_gen/chalk/server/v1/plandebug_pb2_grpc.py +168 -0
chalk/_gen/chalk/server/v1/plandebug_pb2_grpc.pyi +60 -0
chalk/_gen/chalk/server/v1/queries_pb2.py +66 -66
chalk/_gen/chalk/server/v1/queries_pb2.pyi +32 -2
chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py +12 -12
chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi +16 -3
chalk/_gen/chalk/server/v1/scheduler_pb2.py +24 -12
chalk/_gen/chalk/server/v1/scheduler_pb2.pyi +61 -1
chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.py +90 -0
chalk/_gen/chalk/server/v1/scheduler_pb2_grpc.pyi +24 -0
chalk/_gen/chalk/server/v1/script_tasks_pb2.py +15 -3
chalk/_gen/chalk/server/v1/script_tasks_pb2.pyi +22 -0
chalk/_gen/chalk/server/v1/script_tasks_pb2_grpc.py +90 -0
chalk/_gen/chalk/server/v1/script_tasks_pb2_grpc.pyi +24 -0
chalk/_gen/chalk/server/v1/sql_interface_pb2.py +75 -0
chalk/_gen/chalk/server/v1/sql_interface_pb2.pyi +142 -0
chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.py +349 -0
chalk/_gen/chalk/server/v1/sql_interface_pb2_grpc.pyi +114 -0
chalk/_gen/chalk/server/v1/sql_queries_pb2.py +48 -0
chalk/_gen/chalk/server/v1/sql_queries_pb2.pyi +150 -0
chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.py +123 -0
chalk/_gen/chalk/server/v1/sql_queries_pb2_grpc.pyi +52 -0
chalk/_gen/chalk/server/v1/team_pb2.py +154 -141
chalk/_gen/chalk/server/v1/team_pb2.pyi +30 -2
chalk/_gen/chalk/server/v1/team_pb2_grpc.py +45 -0
chalk/_gen/chalk/server/v1/team_pb2_grpc.pyi +12 -0
chalk/_gen/chalk/server/v1/topic_pb2.py +5 -3
chalk/_gen/chalk/server/v1/topic_pb2.pyi +10 -1
chalk/_gen/chalk/server/v1/trace_pb2.py +44 -40
chalk/_gen/chalk/server/v1/trace_pb2.pyi +20 -0
chalk/_gen/chalk/streaming/v1/debug_service_pb2.py +62 -0
chalk/_gen/chalk/streaming/v1/debug_service_pb2.pyi +75 -0
chalk/_gen/chalk/streaming/v1/debug_service_pb2_grpc.py +221 -0
chalk/_gen/chalk/streaming/v1/debug_service_pb2_grpc.pyi +88 -0
chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.py +16 -10
chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2.pyi +52 -1
chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.py +48 -0
chalk/_gen/chalk/streaming/v1/simple_streaming_service_pb2_grpc.pyi +20 -0
chalk/_gen/chalk/utils/v1/field_change_pb2.py +32 -0
chalk/_gen/chalk/utils/v1/field_change_pb2.pyi +42 -0
chalk/_gen/chalk/utils/v1/field_change_pb2_grpc.py +4 -0
chalk/_gen/chalk/utils/v1/field_change_pb2_grpc.pyi +4 -0
chalk/_lsp/error_builder.py +11 -0
chalk/_version.py +1 -1
chalk/client/client.py +128 -43
chalk/client/client_async.py +149 -0
chalk/client/client_async_impl.py +22 -0
chalk/client/client_grpc.py +539 -104
chalk/client/client_impl.py +449 -122
chalk/client/dataset.py +7 -1
chalk/client/models.py +98 -0
chalk/client/serialization/model_serialization.py +92 -9
chalk/df/LazyFramePlaceholder.py +1154 -0
chalk/features/_class_property.py +7 -0
chalk/features/_embedding/embedding.py +1 -0
chalk/features/_encoding/converter.py +83 -2
chalk/features/feature_field.py +40 -30
chalk/features/feature_set_decorator.py +1 -0
chalk/features/feature_wrapper.py +42 -3
chalk/features/hooks.py +81 -10
chalk/features/inference.py +33 -31
chalk/features/resolver.py +224 -24
chalk/functions/__init__.py +65 -3
chalk/gitignore/gitignore_parser.py +5 -1
chalk/importer.py +142 -68
chalk/ml/__init__.py +2 -0
chalk/ml/model_hooks.py +194 -26
chalk/ml/model_reference.py +56 -8
chalk/ml/model_version.py +24 -15
chalk/ml/utils.py +20 -17
chalk/operators/_utils.py +10 -3
chalk/parsed/_proto/export.py +22 -0
chalk/parsed/duplicate_input_gql.py +3 -0
chalk/parsed/json_conversions.py +20 -14
chalk/parsed/to_proto.py +16 -4
chalk/parsed/user_types_to_json.py +31 -10
chalk/parsed/validation_from_registries.py +182 -0
chalk/queries/named_query.py +16 -6
chalk/queries/scheduled_query.py +9 -1
chalk/serialization/parsed_annotation.py +24 -11
chalk/sql/__init__.py +18 -0
chalk/sql/_internal/integrations/databricks.py +55 -17
chalk/sql/_internal/integrations/mssql.py +127 -62
chalk/sql/_internal/integrations/redshift.py +4 -0
chalk/sql/_internal/sql_file_resolver.py +53 -9
chalk/sql/_internal/sql_source.py +35 -2
chalk/streams/_kafka_source.py +5 -1
chalk/streams/_windows.py +15 -2
chalk/utils/_otel_version.py +13 -0
chalk/utils/async_helpers.py +2 -2
chalk/utils/missing_dependency.py +5 -4
chalk/utils/tracing.py +185 -95
{chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/METADATA +4 -6
{chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/RECORD +202 -146
{chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/WHEEL +0 -0
{chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/entry_points.txt +0 -0
{chalkpy-2.90.1.dist-info → chalkpy-2.95.3.dist-info}/top_level.txt +0 -0

chalk/__init__.py CHANGED Viewed

@@ -26,7 +26,7 @@ from chalk.features._document import Document
 from chalk.features._last import Last
 from chalk.features.filter import freeze_time
 from chalk.features.pseudofeatures import Distance, Now
-from chalk.features.resolver import OfflineResolver, OnlineResolver, Resolver, offline, online
+from chalk.features.resolver import OfflineResolver, OnlineResolver, Resolver, make_model_resolver, offline, online
 from chalk.features.tag import BranchId, EnvironmentId
 from chalk.features.underscore import _, __, underscore
 from chalk.importer import get_resolver
@@ -98,6 +98,7 @@ __all__ = (
     "has_many",
     "has_one",
     "is_primary",
+    "make_model_resolver",
     "make_sql_file_resolver",
     "offline",
     "online",

chalk/_gen/chalk/arrow/v1/arrow_pb2.py CHANGED Viewed

@@ -14,7 +14,7 @@ _sym_db = _symbol_database.Default()
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n\x1a\x63halk/arrow/v1/arrow.proto\x12\x0e\x63halk.arrow.v1"\xb8\x01\n\x06Schema\x12/\n\x07\x63olumns\x18\x01 \x03(\x0b\x32\x15.chalk.arrow.v1.FieldR\x07\x63olumns\x12@\n\x08metadata\x18\x02 \x03(\x0b\x32$.chalk.arrow.v1.Schema.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"\xa2\x02\n\x05\x46ield\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x38\n\narrow_type\x18\x02 \x01(\x0b\x32\x19.chalk.arrow.v1.ArrowTypeR\tarrowType\x12\x1a\n\x08nullable\x18\x03 \x01(\x08R\x08nullable\x12\x31\n\x08\x63hildren\x18\x04 \x03(\x0b\x32\x15.chalk.arrow.v1.FieldR\x08\x63hildren\x12?\n\x08metadata\x18\x05 \x03(\x0b\x32#.chalk.arrow.v1.Field.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")\n\x0f\x46ixedSizeBinary\x12\x16\n\x06length\x18\x01 \x01(\x05R\x06length"^\n\tTimestamp\x12\x35\n\ttime_unit\x18\x01 \x01(\x0e\x32\x18.chalk.arrow.v1.TimeUnitR\x08timeUnit\x12\x1a\n\x08timezone\x18\x02 \x01(\tR\x08timezone"=\n\x07\x44\x65\x63imal\x12\x1c\n\tprecision\x18\x01 \x01(\x05R\tprecision\x12\x14\n\x05scale\x18\x02 \x01(\x05R\x05scale"<\n\x04List\x12\x34\n\nfield_type\x18\x01 \x01(\x0b\x32\x15.chalk.arrow.v1.FieldR\tfieldType"b\n\rFixedSizeList\x12\x34\n\nfield_type\x18\x01 \x01(\x0b\x32\x15.chalk.arrow.v1.FieldR\tfieldType\x12\x1b\n\tlist_size\x18\x02 \x01(\x05R\x08listSize"G\n\x06Struct\x12=\n\x0fsub_field_types\x18\x01 \x03(\x0b\x32\x15.chalk.arrow.v1.FieldR\rsubFieldTypes"\x90\x01\n\x03Map\x12\x32\n\tkey_field\x18\x01 \x01(\x0b\x32\x15.chalk.arrow.v1.FieldR\x08keyField\x12\x34\n\nitem_field\x18\x02 \x01(\x0b\x32\x15.chalk.arrow.v1.FieldR\titemField\x12\x1f\n\x0bkeys_sorted\x18\x03 \x01(\x08R\nkeysSorted"`\n\x0fScalarListValue\x12\x1d\n\narrow_data\x18\x01 \x01(\x0cR\tarrowData\x12.\n\x06schema\x18\x02 \x01(\x0b\x32\x16.chalk.arrow.v1.SchemaR\x06schema"\x8a\x01\n\x11ScalarTime32Value\x12\x30\n\x13time32_second_value\x18\x01 \x01(\x05H\x00R\x11time32SecondValue\x12:\n\x18time32_millisecond_value\x18\x02 \x01(\x05H\x00R\x16time32MillisecondValueB\x07\n\x05value"\x92\x01\n\x11ScalarTime64Value\x12:\n\x18time64_microsecond_value\x18\x01 \x01(\x03H\x00R\x16time64MicrosecondValue\x12\x38\n\x17time64_nanosecond_value\x18\x02 \x01(\x03H\x00R\x15time64NanosecondValueB\x07\n\x05value"\x8f\x02\n\x14ScalarTimestampValue\x12\x36\n\x16time_microsecond_value\x18\x01 \x01(\x03H\x00R\x14timeMicrosecondValue\x12\x34\n\x15time_nanosecond_value\x18\x02 \x01(\x03H\x00R\x13timeNanosecondValue\x12,\n\x11time_second_value\x18\x03 \x01(\x03H\x00R\x0ftimeSecondValue\x12\x36\n\x16time_millisecond_value\x18\x04 \x01(\x03H\x00R\x14timeMillisecondValue\x12\x1a\n\x08timezone\x18\x05 \x01(\tR\x08timezoneB\x07\n\x05value"|\n\x0bStructValue\x12>\n\x0c\x66ield_values\x18\x02 \x03(\x0b\x32\x1b.chalk.arrow.v1.ScalarValueR\x0b\x66ieldValues\x12-\n\x06\x66ields\x18\x03 \x03(\x0b\x32\x15.chalk.arrow.v1.FieldR\x06\x66ields"G\n\x15ScalarFixedSizeBinary\x12\x16\n\x06values\x18\x01 \x01(\x0cR\x06values\x12\x16\n\x06length\x18\x02 \x01(\x05R\x06length"X\n\x0c\x44\x65\x63imalValue\x12\x14\n\x05value\x18\x01 \x01(\x0cR\x05value\x12\x1c\n\tprecision\x18\x02 \x01(\x05R\tprecision\x12\x14\n\x05scale\x18\x03 \x01(\x05R\x05scale"]\n\tExtension\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12<\n\x0cstorage_type\x18\x02 \x01(\x0b\x32\x19.chalk.arrow.v1.ArrowTypeR\x0bstorageType"\x94\x01\n\x0e\x45xtensionValue\x12@\n\x0e\x65xtension_type\x18\x01 \x01(\x0b\x32\x19.chalk.arrow.v1.ExtensionR\rextensionType\x12@\n\rstorage_value\x18\x02 \x01(\x0b\x32\x1b.chalk.arrow.v1.ScalarValueR\x0cstorageValue"\xef\x0e\n\x0bScalarValue\x12:\n\nnull_value\x18\x01 \x01(\x0b\x32\x19.chalk.arrow.v1.ArrowTypeH\x00R\tnullValue\x12\x1f\n\nbool_value\x18\x02 \x01(\x08H\x00R\tboolValue\x12%\n\rfloat64_value\x18\x03 \x01(\x01H\x00R\x0c\x66loat64Value\x12!\n\x0bint64_value\x18\x04 \x01(\x03H\x00R\nint64Value\x12*\n\x10large_utf8_value\x18\x05 \x01(\tH\x00R\x0elargeUtf8Value\x12O\n\x0ftimestamp_value\x18\x06 \x01(\x0b\x32$.chalk.arrow.v1.ScalarTimestampValueH\x00R\x0etimestampValue\x12$\n\rdate_64_value\x18\x07 \x01(\x03H\x00R\x0b\x64\x61te64Value\x12@\n\x0cstruct_value\x18\x08 \x01(\x0b\x32\x1b.chalk.arrow.v1.StructValueH\x00R\x0bstructValue\x12K\n\x10large_list_value\x18\t \x01(\x0b\x32\x1f.chalk.arrow.v1.ScalarListValueH\x00R\x0elargeListValue\x12\x46\n\x0ctime64_value\x18\n \x01(\x0b\x32!.chalk.arrow.v1.ScalarTime64ValueH\x00R\x0btime64Value\x12\x34\n\x15\x64uration_second_value\x18\x10 \x01(\x03H\x00R\x13\x64urationSecondValue\x12>\n\x1a\x64uration_millisecond_value\x18\x11 \x01(\x03H\x00R\x18\x64urationMillisecondValue\x12>\n\x1a\x64uration_microsecond_value\x18\x12 \x01(\x03H\x00R\x18\x64urationMicrosecondValue\x12<\n\x19\x64uration_nanosecond_value\x18\x13 \x01(\x03H\x00R\x17\x64urationNanosecondValue\x12\x1f\n\nutf8_value\x18\x14 \x01(\tH\x00R\tutf8Value\x12\x1f\n\nint8_value\x18\x15 \x01(\x05H\x00R\tint8Value\x12!\n\x0bint16_value\x18\x16 \x01(\x05H\x00R\nint16Value\x12!\n\x0bint32_value\x18\x17 \x01(\x05H\x00R\nint32Value\x12!\n\x0buint8_value\x18\x18 \x01(\rH\x00R\nuint8Value\x12#\n\x0cuint16_value\x18\x19 \x01(\rH\x00R\x0buint16Value\x12#\n\x0cuint32_value\x18\x1a \x01(\rH\x00R\x0buint32Value\x12#\n\x0cuint64_value\x18\x1b \x01(\x04H\x00R\x0buint64Value\x12%\n\rfloat16_value\x18\x1c \x01(\x02H\x00R\x0c\x66loat16Value\x12%\n\rfloat32_value\x18\x1d \x01(\x02H\x00R\x0c\x66loat32Value\x12$\n\rdate_32_value\x18\x1e \x01(\x05H\x00R\x0b\x64\x61te32Value\x12\x46\n\x0ctime32_value\x18\x1f \x01(\x0b\x32!.chalk.arrow.v1.ScalarTime32ValueH\x00R\x0btime32Value\x12@\n\nlist_value\x18  \x01(\x0b\x32\x1f.chalk.arrow.v1.ScalarListValueH\x00R\tlistValue\x12T\n\x15\x66ixed_size_list_value\x18! \x01(\x0b\x32\x1f.chalk.arrow.v1.ScalarListValueH\x00R\x12\x66ixedSizeListValue\x12>\n\tmap_value\x18" \x01(\x0b\x32\x1f.chalk.arrow.v1.ScalarListValueH\x00R\x08mapValue\x12#\n\x0c\x62inary_value\x18# \x01(\x0cH\x00R\x0b\x62inaryValue\x12.\n\x12large_binary_value\x18$ \x01(\x0cH\x00R\x10largeBinaryValue\x12^\n\x17\x66ixed_size_binary_value\x18% \x01(\x0b\x32%.chalk.arrow.v1.ScalarFixedSizeBinaryH\x00R\x14\x66ixedSizeBinaryValue\x12I\n\x10\x64\x65\x63imal128_value\x18& \x01(\x0b\x32\x1c.chalk.arrow.v1.DecimalValueH\x00R\x0f\x64\x65\x63imal128Value\x12I\n\x10\x64\x65\x63imal256_value\x18\' \x01(\x0b\x32\x1c.chalk.arrow.v1.DecimalValueH\x00R\x0f\x64\x65\x63imal256Value\x12I\n\x0f\x65xtension_value\x18( \x01(\x0b\x32\x1e.chalk.arrow.v1.ExtensionValueH\x00R\x0e\x65xtensionValueB\x07\n\x05value"\x89\x0e\n\tArrowType\x12\x32\n\x04none\x18\x01 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x04none\x12\x32\n\x04\x62ool\x18\x02 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x04\x62ool\x12\x38\n\x07\x66loat64\x18\x03 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x07\x66loat64\x12\x34\n\x05int64\x18\x04 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x05int64\x12=\n\nlarge_utf8\x18\x05 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\tlargeUtf8\x12\x39\n\ttimestamp\x18\x06 \x01(\x0b\x32\x19.chalk.arrow.v1.TimestampH\x00R\ttimestamp\x12\x36\n\x06\x64\x61te64\x18\x07 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x06\x64\x61te64\x12\x30\n\x06struct\x18\x08 \x01(\x0b\x32\x16.chalk.arrow.v1.StructH\x00R\x06struct\x12\x35\n\nlarge_list\x18\t \x01(\x0b\x32\x14.chalk.arrow.v1.ListH\x00R\tlargeList\x12\x32\n\x06time64\x18\n \x01(\x0e\x32\x18.chalk.arrow.v1.TimeUnitH\x00R\x06time64\x12\x36\n\x08\x64uration\x18\x10 \x01(\x0e\x32\x18.chalk.arrow.v1.TimeUnitH\x00R\x08\x64uration\x12\x32\n\x04utf8\x18\x11 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x04utf8\x12\x32\n\x04int8\x18\x12 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x04int8\x12\x34\n\x05int16\x18\x13 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x05int16\x12\x34\n\x05int32\x18\x14 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x05int32\x12\x34\n\x05uint8\x18\x15 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x05uint8\x12\x36\n\x06uint16\x18\x16 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x06uint16\x12\x36\n\x06uint32\x18\x17 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x06uint32\x12\x36\n\x06uint64\x18\x18 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x06uint64\x12\x38\n\x07\x66loat16\x18\x19 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x07\x66loat16\x12\x38\n\x07\x66loat32\x18\x1a \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x07\x66loat32\x12\x36\n\x06\x64\x61te32\x18\x1b \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x06\x64\x61te32\x12\x32\n\x06time32\x18\x1c \x01(\x0e\x32\x18.chalk.arrow.v1.TimeUnitH\x00R\x06time32\x12*\n\x04list\x18\x1d \x01(\x0b\x32\x14.chalk.arrow.v1.ListH\x00R\x04list\x12G\n\x0f\x66ixed_size_list\x18\x1e \x01(\x0b\x32\x1d.chalk.arrow.v1.FixedSizeListH\x00R\rfixedSizeList\x12\x36\n\x06\x62inary\x18\x1f \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x06\x62inary\x12\x41\n\x0clarge_binary\x18  \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x0blargeBinary\x12,\n\x11\x66ixed_size_binary\x18! \x01(\x05H\x00R\x0f\x66ixedSizeBinary\x12:\n\x0b\x64\x65\x63imal_128\x18" \x01(\x0b\x32\x17.chalk.arrow.v1.DecimalH\x00R\ndecimal128\x12:\n\x0b\x64\x65\x63imal_256\x18# \x01(\x0b\x32\x17.chalk.arrow.v1.DecimalH\x00R\ndecimal256\x12\'\n\x03map\x18$ \x01(\x0b\x32\x13.chalk.arrow.v1.MapH\x00R\x03map\x12\x39\n\textension\x18% \x01(\x0b\x32\x19.chalk.arrow.v1.ExtensionH\x00R\textensionB\x11\n\x0f\x61rrow_type_enum"\x0e\n\x0c\x45mptyMessage*\x8b\x01\n\x08TimeUnit\x12\x19\n\x15TIME_UNIT_UNSPECIFIED\x10\x00\x12\x14\n\x10TIME_UNIT_SECOND\x10\x01\x12\x19\n\x15TIME_UNIT_MILLISECOND\x10\x02\x12\x19\n\x15TIME_UNIT_MICROSECOND\x10\x03\x12\x18\n\x14TIME_UNIT_NANOSECOND\x10\x04*\x9a\x01\n\x12\x46\x65\x61therCompression\x12#\n\x1f\x46\x45\x41THER_COMPRESSION_UNSPECIFIED\x10\x00\x12$\n FEATHER_COMPRESSION_UNCOMPRESSED\x10\x01\x12\x1b\n\x17\x46\x45\x41THER_COMPRESSION_LZ4\x10\x02\x12\x1c\n\x18\x46\x45\x41THER_COMPRESSION_ZSTD\x10\x03\x42z\n\x12\x63om.chalk.arrow.v1B\nArrowProtoP\x01\xa2\x02\x03\x43\x41X\xaa\x02\x0e\x43halk.Arrow.V1\xca\x02\x0e\x43halk\\Arrow\\V1\xe2\x02\x1a\x43halk\\Arrow\\V1\\GPBMetadata\xea\x02\x10\x43halk::Arrow::V1b\x06proto3'
+    b'\n\x1a\x63halk/arrow/v1/arrow.proto\x12\x0e\x63halk.arrow.v1"\xb8\x01\n\x06Schema\x12/\n\x07\x63olumns\x18\x01 \x03(\x0b\x32\x15.chalk.arrow.v1.FieldR\x07\x63olumns\x12@\n\x08metadata\x18\x02 \x03(\x0b\x32$.chalk.arrow.v1.Schema.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"\xa2\x02\n\x05\x46ield\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x38\n\narrow_type\x18\x02 \x01(\x0b\x32\x19.chalk.arrow.v1.ArrowTypeR\tarrowType\x12\x1a\n\x08nullable\x18\x03 \x01(\x08R\x08nullable\x12\x31\n\x08\x63hildren\x18\x04 \x03(\x0b\x32\x15.chalk.arrow.v1.FieldR\x08\x63hildren\x12?\n\x08metadata\x18\x05 \x03(\x0b\x32#.chalk.arrow.v1.Field.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")\n\x0f\x46ixedSizeBinary\x12\x16\n\x06length\x18\x01 \x01(\x05R\x06length"^\n\tTimestamp\x12\x35\n\ttime_unit\x18\x01 \x01(\x0e\x32\x18.chalk.arrow.v1.TimeUnitR\x08timeUnit\x12\x1a\n\x08timezone\x18\x02 \x01(\tR\x08timezone"=\n\x07\x44\x65\x63imal\x12\x1c\n\tprecision\x18\x01 \x01(\x05R\tprecision\x12\x14\n\x05scale\x18\x02 \x01(\x05R\x05scale"<\n\x04List\x12\x34\n\nfield_type\x18\x01 \x01(\x0b\x32\x15.chalk.arrow.v1.FieldR\tfieldType"b\n\rFixedSizeList\x12\x34\n\nfield_type\x18\x01 \x01(\x0b\x32\x15.chalk.arrow.v1.FieldR\tfieldType\x12\x1b\n\tlist_size\x18\x02 \x01(\x05R\x08listSize"G\n\x06Struct\x12=\n\x0fsub_field_types\x18\x01 \x03(\x0b\x32\x15.chalk.arrow.v1.FieldR\rsubFieldTypes"\x90\x01\n\x03Map\x12\x32\n\tkey_field\x18\x01 \x01(\x0b\x32\x15.chalk.arrow.v1.FieldR\x08keyField\x12\x34\n\nitem_field\x18\x02 \x01(\x0b\x32\x15.chalk.arrow.v1.FieldR\titemField\x12\x1f\n\x0bkeys_sorted\x18\x03 \x01(\x08R\nkeysSorted"`\n\x0fScalarListValue\x12\x1d\n\narrow_data\x18\x01 \x01(\x0cR\tarrowData\x12.\n\x06schema\x18\x02 \x01(\x0b\x32\x16.chalk.arrow.v1.SchemaR\x06schema"\x8a\x01\n\x11ScalarTime32Value\x12\x30\n\x13time32_second_value\x18\x01 \x01(\x05H\x00R\x11time32SecondValue\x12:\n\x18time32_millisecond_value\x18\x02 \x01(\x05H\x00R\x16time32MillisecondValueB\x07\n\x05value"\x92\x01\n\x11ScalarTime64Value\x12:\n\x18time64_microsecond_value\x18\x01 \x01(\x03H\x00R\x16time64MicrosecondValue\x12\x38\n\x17time64_nanosecond_value\x18\x02 \x01(\x03H\x00R\x15time64NanosecondValueB\x07\n\x05value"\x8f\x02\n\x14ScalarTimestampValue\x12\x36\n\x16time_microsecond_value\x18\x01 \x01(\x03H\x00R\x14timeMicrosecondValue\x12\x34\n\x15time_nanosecond_value\x18\x02 \x01(\x03H\x00R\x13timeNanosecondValue\x12,\n\x11time_second_value\x18\x03 \x01(\x03H\x00R\x0ftimeSecondValue\x12\x36\n\x16time_millisecond_value\x18\x04 \x01(\x03H\x00R\x14timeMillisecondValue\x12\x1a\n\x08timezone\x18\x05 \x01(\tR\x08timezoneB\x07\n\x05value"|\n\x0bStructValue\x12>\n\x0c\x66ield_values\x18\x02 \x03(\x0b\x32\x1b.chalk.arrow.v1.ScalarValueR\x0b\x66ieldValues\x12-\n\x06\x66ields\x18\x03 \x03(\x0b\x32\x15.chalk.arrow.v1.FieldR\x06\x66ields"G\n\x15ScalarFixedSizeBinary\x12\x16\n\x06values\x18\x01 \x01(\x0cR\x06values\x12\x16\n\x06length\x18\x02 \x01(\x05R\x06length"X\n\x0c\x44\x65\x63imalValue\x12\x14\n\x05value\x18\x01 \x01(\x0cR\x05value\x12\x1c\n\tprecision\x18\x02 \x01(\x05R\tprecision\x12\x14\n\x05scale\x18\x03 \x01(\x05R\x05scale"]\n\tExtension\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12<\n\x0cstorage_type\x18\x02 \x01(\x0b\x32\x19.chalk.arrow.v1.ArrowTypeR\x0bstorageType"\x94\x01\n\x0e\x45xtensionValue\x12@\n\x0e\x65xtension_type\x18\x01 \x01(\x0b\x32\x19.chalk.arrow.v1.ExtensionR\rextensionType\x12@\n\rstorage_value\x18\x02 \x01(\x0b\x32\x1b.chalk.arrow.v1.ScalarValueR\x0cstorageValue"\xef\x0e\n\x0bScalarValue\x12:\n\nnull_value\x18\x01 \x01(\x0b\x32\x19.chalk.arrow.v1.ArrowTypeH\x00R\tnullValue\x12\x1f\n\nbool_value\x18\x02 \x01(\x08H\x00R\tboolValue\x12%\n\rfloat64_value\x18\x03 \x01(\x01H\x00R\x0c\x66loat64Value\x12!\n\x0bint64_value\x18\x04 \x01(\x03H\x00R\nint64Value\x12*\n\x10large_utf8_value\x18\x05 \x01(\tH\x00R\x0elargeUtf8Value\x12O\n\x0ftimestamp_value\x18\x06 \x01(\x0b\x32$.chalk.arrow.v1.ScalarTimestampValueH\x00R\x0etimestampValue\x12$\n\rdate_64_value\x18\x07 \x01(\x03H\x00R\x0b\x64\x61te64Value\x12@\n\x0cstruct_value\x18\x08 \x01(\x0b\x32\x1b.chalk.arrow.v1.StructValueH\x00R\x0bstructValue\x12K\n\x10large_list_value\x18\t \x01(\x0b\x32\x1f.chalk.arrow.v1.ScalarListValueH\x00R\x0elargeListValue\x12\x46\n\x0ctime64_value\x18\n \x01(\x0b\x32!.chalk.arrow.v1.ScalarTime64ValueH\x00R\x0btime64Value\x12\x34\n\x15\x64uration_second_value\x18\x10 \x01(\x03H\x00R\x13\x64urationSecondValue\x12>\n\x1a\x64uration_millisecond_value\x18\x11 \x01(\x03H\x00R\x18\x64urationMillisecondValue\x12>\n\x1a\x64uration_microsecond_value\x18\x12 \x01(\x03H\x00R\x18\x64urationMicrosecondValue\x12<\n\x19\x64uration_nanosecond_value\x18\x13 \x01(\x03H\x00R\x17\x64urationNanosecondValue\x12\x1f\n\nutf8_value\x18\x14 \x01(\tH\x00R\tutf8Value\x12\x1f\n\nint8_value\x18\x15 \x01(\x05H\x00R\tint8Value\x12!\n\x0bint16_value\x18\x16 \x01(\x05H\x00R\nint16Value\x12!\n\x0bint32_value\x18\x17 \x01(\x05H\x00R\nint32Value\x12!\n\x0buint8_value\x18\x18 \x01(\rH\x00R\nuint8Value\x12#\n\x0cuint16_value\x18\x19 \x01(\rH\x00R\x0buint16Value\x12#\n\x0cuint32_value\x18\x1a \x01(\rH\x00R\x0buint32Value\x12#\n\x0cuint64_value\x18\x1b \x01(\x04H\x00R\x0buint64Value\x12%\n\rfloat16_value\x18\x1c \x01(\x02H\x00R\x0c\x66loat16Value\x12%\n\rfloat32_value\x18\x1d \x01(\x02H\x00R\x0c\x66loat32Value\x12$\n\rdate_32_value\x18\x1e \x01(\x05H\x00R\x0b\x64\x61te32Value\x12\x46\n\x0ctime32_value\x18\x1f \x01(\x0b\x32!.chalk.arrow.v1.ScalarTime32ValueH\x00R\x0btime32Value\x12@\n\nlist_value\x18  \x01(\x0b\x32\x1f.chalk.arrow.v1.ScalarListValueH\x00R\tlistValue\x12T\n\x15\x66ixed_size_list_value\x18! \x01(\x0b\x32\x1f.chalk.arrow.v1.ScalarListValueH\x00R\x12\x66ixedSizeListValue\x12>\n\tmap_value\x18" \x01(\x0b\x32\x1f.chalk.arrow.v1.ScalarListValueH\x00R\x08mapValue\x12#\n\x0c\x62inary_value\x18# \x01(\x0cH\x00R\x0b\x62inaryValue\x12.\n\x12large_binary_value\x18$ \x01(\x0cH\x00R\x10largeBinaryValue\x12^\n\x17\x66ixed_size_binary_value\x18% \x01(\x0b\x32%.chalk.arrow.v1.ScalarFixedSizeBinaryH\x00R\x14\x66ixedSizeBinaryValue\x12I\n\x10\x64\x65\x63imal128_value\x18& \x01(\x0b\x32\x1c.chalk.arrow.v1.DecimalValueH\x00R\x0f\x64\x65\x63imal128Value\x12I\n\x10\x64\x65\x63imal256_value\x18\' \x01(\x0b\x32\x1c.chalk.arrow.v1.DecimalValueH\x00R\x0f\x64\x65\x63imal256Value\x12I\n\x0f\x65xtension_value\x18( \x01(\x0b\x32\x1e.chalk.arrow.v1.ExtensionValueH\x00R\x0e\x65xtensionValueB\x07\n\x05value"\x89\x0e\n\tArrowType\x12\x32\n\x04none\x18\x01 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x04none\x12\x32\n\x04\x62ool\x18\x02 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x04\x62ool\x12\x38\n\x07\x66loat64\x18\x03 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x07\x66loat64\x12\x34\n\x05int64\x18\x04 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x05int64\x12=\n\nlarge_utf8\x18\x05 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\tlargeUtf8\x12\x39\n\ttimestamp\x18\x06 \x01(\x0b\x32\x19.chalk.arrow.v1.TimestampH\x00R\ttimestamp\x12\x36\n\x06\x64\x61te64\x18\x07 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x06\x64\x61te64\x12\x30\n\x06struct\x18\x08 \x01(\x0b\x32\x16.chalk.arrow.v1.StructH\x00R\x06struct\x12\x35\n\nlarge_list\x18\t \x01(\x0b\x32\x14.chalk.arrow.v1.ListH\x00R\tlargeList\x12\x32\n\x06time64\x18\n \x01(\x0e\x32\x18.chalk.arrow.v1.TimeUnitH\x00R\x06time64\x12\x36\n\x08\x64uration\x18\x10 \x01(\x0e\x32\x18.chalk.arrow.v1.TimeUnitH\x00R\x08\x64uration\x12\x32\n\x04utf8\x18\x11 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x04utf8\x12\x32\n\x04int8\x18\x12 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x04int8\x12\x34\n\x05int16\x18\x13 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x05int16\x12\x34\n\x05int32\x18\x14 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x05int32\x12\x34\n\x05uint8\x18\x15 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x05uint8\x12\x36\n\x06uint16\x18\x16 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x06uint16\x12\x36\n\x06uint32\x18\x17 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x06uint32\x12\x36\n\x06uint64\x18\x18 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x06uint64\x12\x38\n\x07\x66loat16\x18\x19 \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x07\x66loat16\x12\x38\n\x07\x66loat32\x18\x1a \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x07\x66loat32\x12\x36\n\x06\x64\x61te32\x18\x1b \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x06\x64\x61te32\x12\x32\n\x06time32\x18\x1c \x01(\x0e\x32\x18.chalk.arrow.v1.TimeUnitH\x00R\x06time32\x12*\n\x04list\x18\x1d \x01(\x0b\x32\x14.chalk.arrow.v1.ListH\x00R\x04list\x12G\n\x0f\x66ixed_size_list\x18\x1e \x01(\x0b\x32\x1d.chalk.arrow.v1.FixedSizeListH\x00R\rfixedSizeList\x12\x36\n\x06\x62inary\x18\x1f \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x06\x62inary\x12\x41\n\x0clarge_binary\x18  \x01(\x0b\x32\x1c.chalk.arrow.v1.EmptyMessageH\x00R\x0blargeBinary\x12,\n\x11\x66ixed_size_binary\x18! \x01(\x05H\x00R\x0f\x66ixedSizeBinary\x12:\n\x0b\x64\x65\x63imal_128\x18" \x01(\x0b\x32\x17.chalk.arrow.v1.DecimalH\x00R\ndecimal128\x12:\n\x0b\x64\x65\x63imal_256\x18# \x01(\x0b\x32\x17.chalk.arrow.v1.DecimalH\x00R\ndecimal256\x12\'\n\x03map\x18$ \x01(\x0b\x32\x13.chalk.arrow.v1.MapH\x00R\x03map\x12\x39\n\textension\x18% \x01(\x0b\x32\x19.chalk.arrow.v1.ExtensionH\x00R\textensionB\x11\n\x0f\x61rrow_type_enum"\x0e\n\x0c\x45mptyMessage"G\n\x11TableParquetBytes\x12\x32\n\x15\x65ncoded_parquet_bytes\x18\x01 \x01(\x0cR\x13\x65ncodedParquetBytes*\x8b\x01\n\x08TimeUnit\x12\x19\n\x15TIME_UNIT_UNSPECIFIED\x10\x00\x12\x14\n\x10TIME_UNIT_SECOND\x10\x01\x12\x19\n\x15TIME_UNIT_MILLISECOND\x10\x02\x12\x19\n\x15TIME_UNIT_MICROSECOND\x10\x03\x12\x18\n\x14TIME_UNIT_NANOSECOND\x10\x04*\x9a\x01\n\x12\x46\x65\x61therCompression\x12#\n\x1f\x46\x45\x41THER_COMPRESSION_UNSPECIFIED\x10\x00\x12$\n FEATHER_COMPRESSION_UNCOMPRESSED\x10\x01\x12\x1b\n\x17\x46\x45\x41THER_COMPRESSION_LZ4\x10\x02\x12\x1c\n\x18\x46\x45\x41THER_COMPRESSION_ZSTD\x10\x03\x42z\n\x12\x63om.chalk.arrow.v1B\nArrowProtoP\x01\xa2\x02\x03\x43\x41X\xaa\x02\x0e\x43halk.Arrow.V1\xca\x02\x0e\x43halk\\Arrow\\V1\xe2\x02\x1a\x43halk\\Arrow\\V1\\GPBMetadata\xea\x02\x10\x43halk::Arrow::V1b\x06proto3'
 )
 _globals = globals()
@@ -29,10 +29,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_SCHEMA_METADATAENTRY"]._serialized_options = b"8\001"
     _globals["_FIELD_METADATAENTRY"]._options = None
     _globals["_FIELD_METADATAENTRY"]._serialized_options = b"8\001"
-    _globals["_TIMEUNIT"]._serialized_start = 6034
-    _globals["_TIMEUNIT"]._serialized_end = 6173
-    _globals["_FEATHERCOMPRESSION"]._serialized_start = 6176
-    _globals["_FEATHERCOMPRESSION"]._serialized_end = 6330
+    _globals["_TIMEUNIT"]._serialized_start = 6107
+    _globals["_TIMEUNIT"]._serialized_end = 6246
+    _globals["_FEATHERCOMPRESSION"]._serialized_start = 6249
+    _globals["_FEATHERCOMPRESSION"]._serialized_end = 6403
     _globals["_SCHEMA"]._serialized_start = 47
     _globals["_SCHEMA"]._serialized_end = 231
     _globals["_SCHEMA_METADATAENTRY"]._serialized_start = 172
@@ -79,4 +79,6 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_ARROWTYPE"]._serialized_end = 6015
     _globals["_EMPTYMESSAGE"]._serialized_start = 6017
     _globals["_EMPTYMESSAGE"]._serialized_end = 6031
+    _globals["_TABLEPARQUETBYTES"]._serialized_start = 6033
+    _globals["_TABLEPARQUETBYTES"]._serialized_end = 6104
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/arrow/v1/arrow_pb2.pyi CHANGED Viewed

@@ -541,3 +541,9 @@ class ArrowType(_message.Message):
 class EmptyMessage(_message.Message):
     __slots__ = ()
     def __init__(self) -> None: ...
+class TableParquetBytes(_message.Message):
+    __slots__ = ("encoded_parquet_bytes",)
+    ENCODED_PARQUET_BYTES_FIELD_NUMBER: _ClassVar[int]
+    encoded_parquet_bytes: bytes
+    def __init__(self, encoded_parquet_bytes: _Optional[bytes] = ...) -> None: ...

chalk/_gen/chalk/artifacts/v1/chart_pb2.py CHANGED Viewed

@@ -17,7 +17,7 @@ from google.protobuf import duration_pb2 as google_dot_protobuf_dot_duration__pb
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n\x1e\x63halk/artifacts/v1/chart.proto\x12\x12\x63halk.artifacts.v1\x1a\x1egoogle/protobuf/duration.proto"\x86\x03\n\x0c\x41lertTrigger\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x41\n\x08severity\x18\x02 \x01(\x0e\x32%.chalk.artifacts.v1.AlertSeverityKindR\x08severity\x12P\n\x12threshold_position\x18\x03 \x01(\x0e\x32!.chalk.artifacts.v1.ThresholdKindR\x11thresholdPosition\x12\'\n\x0fthreshold_value\x18\x04 \x01(\x02R\x0ethresholdValue\x12$\n\x0bseries_name\x18\x05 \x01(\tH\x00R\nseriesName\x88\x01\x01\x12&\n\x0c\x63hannel_name\x18\x06 \x01(\tH\x01R\x0b\x63hannelName\x88\x01\x01\x12%\n\x0b\x64\x65scription\x18\x07 \x01(\tH\x02R\x0b\x64\x65scription\x88\x01\x01\x42\x0e\n\x0c_series_nameB\x0f\n\r_channel_nameB\x0e\n\x0c_description"K\n\x15\x44\x61tasetFeatureOperand\x12\x18\n\x07\x64\x61taset\x18\x01 \x01(\tR\x07\x64\x61taset\x12\x18\n\x07\x66\x65\x61ture\x18\x02 \x01(\tR\x07\x66\x65\x61ture"\xdb\x02\n\rMetricFormula\x12\x39\n\x04kind\x18\x01 \x01(\x0e\x32%.chalk.artifacts.v1.MetricFormulaKindR\x04kind\x12\x39\n\x16single_series_operands\x18\x02 \x01(\x03H\x00R\x14singleSeriesOperands\x88\x01\x01\x12\x32\n\x15multi_series_operands\x18\x03 \x03(\x03R\x13multiSeriesOperands\x12\x63\n\x18\x64\x61taset_feature_operands\x18\x04 \x01(\x0b\x32).chalk.artifacts.v1.DatasetFeatureOperandR\x16\x64\x61tasetFeatureOperands\x12\x17\n\x04name\x18\x05 \x01(\tH\x01R\x04name\x88\x01\x01\x42\x19\n\x17_single_series_operandsB\x07\n\x05_name"\x9c\x01\n\x0cMetricFilter\x12\x32\n\x04kind\x18\x01 \x01(\x0e\x32\x1e.chalk.artifacts.v1.FilterKindR\x04kind\x12\x42\n\ncomparator\x18\x02 \x01(\x0e\x32".chalk.artifacts.v1.ComparatorKindR\ncomparator\x12\x14\n\x05value\x18\x03 \x03(\tR\x05value"\x85\x03\n\x12MetricConfigSeries\x12\x36\n\x06metric\x18\x01 \x01(\x0e\x32\x1e.chalk.artifacts.v1.MetricKindR\x06metric\x12:\n\x07\x66ilters\x18\x02 \x03(\x0b\x32 .chalk.artifacts.v1.MetricFilterR\x07\x66ilters\x12\x17\n\x04name\x18\x03 \x01(\tH\x00R\x04name\x88\x01\x01\x12O\n\x0fwindow_function\x18\x04 \x01(\x0e\x32&.chalk.artifacts.v1.WindowFunctionKindR\x0ewindowFunction\x12:\n\x08group_by\x18\x05 \x03(\x0e\x32\x1f.chalk.artifacts.v1.GroupByKindR\x07groupBy\x12=\n\ntime_shift\x18\x06 \x01(\x0b\x32\x19.google.protobuf.DurationH\x01R\ttimeShift\x88\x01\x01\x42\x07\n\x05_nameB\r\n\x0b_time_shift"\xbb\x02\n\x0cMetricConfig\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12#\n\rwindow_period\x18\x02 \x01(\tR\x0cwindowPeriod\x12>\n\x06series\x18\x03 \x03(\x0b\x32&.chalk.artifacts.v1.MetricConfigSeriesR\x06series\x12=\n\x08\x66ormulas\x18\x04 \x03(\x0b\x32!.chalk.artifacts.v1.MetricFormulaR\x08\x66ormulas\x12:\n\x07trigger\x18\x05 \x01(\x0b\x32 .chalk.artifacts.v1.AlertTriggerR\x07trigger\x12\'\n\x0fgraph_generated\x18\x06 \x01(\x08R\x0egraphGenerated\x12\x0e\n\x02id\x18\x07 \x01(\tR\x02id"\xee\x01\n\x05\x43hart\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x38\n\x06\x63onfig\x18\x02 \x01(\x0b\x32 .chalk.artifacts.v1.MetricConfigR\x06\x63onfig\x12\x42\n\x0b\x65ntity_kind\x18\x03 \x01(\x0e\x32!.chalk.artifacts.v1.ChartLinkKindR\nentityKind\x12 \n\tentity_id\x18\x04 \x01(\tH\x00R\x08\x65ntityId\x88\x01\x01\x12\'\n\x0fgraph_generated\x18\x05 \x01(\x08R\x0egraphGeneratedB\x0c\n\n_entity_id*\xe7\x10\n\nMetricKind\x12\x1b\n\x17METRIC_KIND_UNSPECIFIED\x10\x00\x12%\n!METRIC_KIND_FEATURE_REQUEST_COUNT\x10\x01\x12\x1f\n\x1bMETRIC_KIND_FEATURE_LATENCY\x10\x02\x12!\n\x1dMETRIC_KIND_FEATURE_STALENESS\x10\x03\x12\x1d\n\x19METRIC_KIND_FEATURE_VALUE\x10\x04\x12\x1d\n\x19METRIC_KIND_FEATURE_WRITE\x10\x05\x12"\n\x1eMETRIC_KIND_FEATURE_NULL_RATIO\x10\x06\x12&\n"METRIC_KIND_RESOLVER_REQUEST_COUNT\x10\x07\x12 \n\x1cMETRIC_KIND_RESOLVER_LATENCY\x10\x08\x12&\n"METRIC_KIND_RESOLVER_SUCCESS_RATIO\x10\t\x12\x1b\n\x17METRIC_KIND_QUERY_COUNT\x10\n\x12\x1d\n\x19METRIC_KIND_QUERY_LATENCY\x10\x0b\x12#\n\x1fMETRIC_KIND_QUERY_SUCCESS_RATIO\x10\x0c\x12!\n\x1dMETRIC_KIND_BILLING_INFERENCE\x10\r\x12\x1c\n\x18METRIC_KIND_BILLING_CRON\x10\x0e\x12!\n\x1dMETRIC_KIND_BILLING_MIGRATION\x10\x0f\x12\x1a\n\x16METRIC_KIND_CRON_COUNT\x10\x10\x12\x1c\n\x18METRIC_KIND_CRON_LATENCY\x10\x11\x12)\n%METRIC_KIND_STREAM_MESSAGES_PROCESSED\x10\x12\x12&\n"METRIC_KIND_STREAM_MESSAGE_LATENCY\x10\x13\x12(\n$METRIC_KIND_STREAM_WINDOWS_PROCESSED\x10\x14\x12%\n!METRIC_KIND_STREAM_WINDOW_LATENCY\x10\x15\x12&\n"METRIC_KIND_ONLINE_STORE_KEY_COUNT\x10\x16\x12.\n*METRIC_KIND_ONLINE_STORE_EXPIRED_KEY_COUNT\x10\x17\x12\x30\n,METRIC_KIND_ONLINE_STORE_REQUESTS_PER_SECOND\x10\x18\x12\'\n#METRIC_KIND_CPU_UTILIZATION_PERCENT\x10\x19\x12\x1d\n\x19METRIC_KIND_REPLICA_COUNT\x10\x1a\x12(\n$METRIC_KIND_ONLINE_STORE_USED_MEMORY\x10\x1b\x12)\n%METRIC_KIND_ONLINE_STORE_TOTAL_MEMORY\x10\x1c\x12"\n\x1eMETRIC_KIND_MEMORY_USAGE_BYTES\x10\x1d\x12,\n(METRIC_KIND_TOTAL_MEMORY_AVAILABLE_BYTES\x10\x1e\x12"\n\x1eMETRIC_KIND_NETWORK_READ_BYTES\x10\x1f\x12#\n\x1fMETRIC_KIND_NETWORK_WRITE_BYTES\x10 \x12\x1f\n\x1bMETRIC_KIND_DISK_READ_BYTES\x10!\x12 \n\x1cMETRIC_KIND_DISK_WRITE_BYTES\x10"\x12\x1a\n\x16METRIC_KIND_STREAM_LAG\x10#\x12\x15\n\x11METRIC_KIND_USAGE\x10$\x12&\n"METRIC_KIND_FEATURE_COMPUTED_COUNT\x10%\x12\'\n#METRIC_KIND_FEATURE_LOOKED_UP_COUNT\x10&\x12*\n&METRIC_KIND_FEATURE_INTERMEDIATE_COUNT\x10\'\x12+\n\'METRIC_KIND_FEATURE_COMPUTED_NULL_RATIO\x10(\x12,\n(METRIC_KIND_FEATURE_LOOKED_UP_NULL_RATIO\x10)\x12/\n+METRIC_KIND_FEATURE_INTERMEDIATE_NULL_RATIO\x10*\x12#\n\x1fMETRIC_KIND_STREAM_INGEST_DELAY\x10+\x12&\n"METRIC_KIND_CONTAINER_MEMORY_BYTES\x10,\x12!\n\x1dMETRIC_KIND_HOST_MEMORY_BYTES\x10-\x12)\n%METRIC_KIND_CONTAINER_CPU_UTILIZATION\x10.\x12\x1f\n\x1bMETRIC_KIND_DISK_USED_BYTES\x10/\x12$\n METRIC_KIND_DISK_AVAILABLE_BYTES\x10\x30\x12\'\n#METRIC_KIND_RESOLVER_INVOKER_NET_TX\x10\x31\x12\'\n#METRIC_KIND_RESOLVER_INVOKER_NET_RX\x10\x32\x12-\n)METRIC_KIND_RESOLVER_INVOKER_ROWS_WRITTEN\x10\x33\x12(\n$METRIC_KIND_TOPIC_MESSAGES_PROCESSED\x10\x34\x12\x31\n-METRIC_KIND_SUBSCRIPTION_NUM_UNACKED_MESSAGES\x10\x35\x12\x37\n3METRIC_KIND_SUBSCRIPTION_OLDEST_UNACKED_MESSAGE_AGE\x10\x36\x12 \n\x1cMETRIC_KIND_TOPIC_OFFSET_LAG\x10\x37*\x87\x05\n\nFilterKind\x12\x1b\n\x17\x46ILTER_KIND_UNSPECIFIED\x10\x00\x12\x1e\n\x1a\x46ILTER_KIND_FEATURE_STATUS\x10\x01\x12\x1c\n\x18\x46ILTER_KIND_FEATURE_NAME\x10\x02\x12\x1b\n\x17\x46ILTER_KIND_FEATURE_TAG\x10\x03\x12\x1f\n\x1b\x46ILTER_KIND_RESOLVER_STATUS\x10\x04\x12\x1d\n\x19\x46ILTER_KIND_RESOLVER_NAME\x10\x05\x12\x1c\n\x18\x46ILTER_KIND_RESOLVER_TAG\x10\x06\x12\x1b\n\x17\x46ILTER_KIND_CRON_STATUS\x10\x07\x12 \n\x1c\x46ILTER_KIND_MIGRATION_STATUS\x10\x08\x12\x1e\n\x1a\x46ILTER_KIND_ONLINE_OFFLINE\x10\t\x12\x19\n\x15\x46ILTER_KIND_CACHE_HIT\x10\n\x12\x1c\n\x18\x46ILTER_KIND_OPERATION_ID\x10\x0b\x12\x1a\n\x16\x46ILTER_KIND_QUERY_NAME\x10\x0c\x12\x1c\n\x18\x46ILTER_KIND_QUERY_STATUS\x10\r\x12\x17\n\x13\x46ILTER_KIND_IS_NULL\x10\x0e\x12\x1a\n\x16\x46ILTER_KIND_USAGE_KIND\x10\x0f\x12\x1e\n\x1a\x46ILTER_KIND_RESOURCE_GROUP\x10\x10\x12\x18\n\x14\x46ILTER_KIND_POD_NAME\x10\x11\x12#\n\x1f\x46ILTER_KIND_COMPUTATION_CONTEXT\x10\x12\x12\x1a\n\x16\x46ILTER_KIND_TOPIC_NAME\x10\x13\x12!\n\x1d\x46ILTER_KIND_SUBSCRIPTION_NAME\x10\x14*~\n\x0e\x43omparatorKind\x12\x1f\n\x1b\x43OMPARATOR_KIND_UNSPECIFIED\x10\x00\x12\x16\n\x12\x43OMPARATOR_KIND_EQ\x10\x01\x12\x17\n\x13\x43OMPARATOR_KIND_NEQ\x10\x02\x12\x1a\n\x16\x43OMPARATOR_KIND_ONE_OF\x10\x03*\xec\x03\n\x12WindowFunctionKind\x12$\n WINDOW_FUNCTION_KIND_UNSPECIFIED\x10\x00\x12\x1e\n\x1aWINDOW_FUNCTION_KIND_COUNT\x10\x01\x12\x1d\n\x19WINDOW_FUNCTION_KIND_MEAN\x10\x02\x12\x1c\n\x18WINDOW_FUNCTION_KIND_SUM\x10\x03\x12\x1c\n\x18WINDOW_FUNCTION_KIND_MIN\x10\x04\x12\x1c\n\x18WINDOW_FUNCTION_KIND_MAX\x10\x05\x12&\n"WINDOW_FUNCTION_KIND_PERCENTILE_99\x10\x06\x12&\n"WINDOW_FUNCTION_KIND_PERCENTILE_95\x10\x07\x12&\n"WINDOW_FUNCTION_KIND_PERCENTILE_75\x10\x08\x12&\n"WINDOW_FUNCTION_KIND_PERCENTILE_50\x10\t\x12&\n"WINDOW_FUNCTION_KIND_PERCENTILE_25\x10\n\x12%\n!WINDOW_FUNCTION_KIND_PERCENTILE_5\x10\x0b\x12(\n$WINDOW_FUNCTION_KIND_ALL_PERCENTILES\x10\x0c*\xaa\x04\n\x0bGroupByKind\x12\x1d\n\x19GROUP_BY_KIND_UNSPECIFIED\x10\x00\x12 \n\x1cGROUP_BY_KIND_FEATURE_STATUS\x10\x01\x12\x1e\n\x1aGROUP_BY_KIND_FEATURE_NAME\x10\x02\x12\x19\n\x15GROUP_BY_KIND_IS_NULL\x10\x03\x12!\n\x1dGROUP_BY_KIND_RESOLVER_STATUS\x10\x04\x12\x1f\n\x1bGROUP_BY_KIND_RESOLVER_NAME\x10\x05\x12\x1e\n\x1aGROUP_BY_KIND_QUERY_STATUS\x10\x06\x12\x1c\n\x18GROUP_BY_KIND_QUERY_NAME\x10\x07\x12 \n\x1cGROUP_BY_KIND_ONLINE_OFFLINE\x10\x08\x12\x1b\n\x17GROUP_BY_KIND_CACHE_HIT\x10\t\x12\x1c\n\x18GROUP_BY_KIND_USAGE_KIND\x10\n\x12 \n\x1cGROUP_BY_KIND_RESOURCE_GROUP\x10\x0b\x12\x1f\n\x1bGROUP_BY_KIND_DEPLOYMENT_ID\x10\x0c\x12\x1e\n\x1aGROUP_BY_KIND_OPERATION_ID\x10\r\x12\x1a\n\x16GROUP_BY_KIND_POD_NAME\x10\x0e\x12\x1c\n\x18GROUP_BY_KIND_TOPIC_NAME\x10\x0f\x12#\n\x1fGROUP_BY_KIND_SUBSCRIPTION_NAME\x10\x10*\xe4\x02\n\x11MetricFormulaKind\x12#\n\x1fMETRIC_FORMULA_KIND_UNSPECIFIED\x10\x00\x12\x1b\n\x17METRIC_FORMULA_KIND_SUM\x10\x01\x12#\n\x1fMETRIC_FORMULA_KIND_TOTAL_RATIO\x10\x02\x12\x1d\n\x19METRIC_FORMULA_KIND_RATIO\x10\x03\x12\x1f\n\x1bMETRIC_FORMULA_KIND_PRODUCT\x10\x04\x12\x1b\n\x17METRIC_FORMULA_KIND_ABS\x10\x05\x12\x1f\n\x1bMETRIC_FORMULA_KIND_KS_STAT\x10\x06\x12\x1f\n\x1bMETRIC_FORMULA_KIND_KS_TEST\x10\x07\x12$\n METRIC_FORMULA_KIND_KS_THRESHOLD\x10\x08\x12#\n\x1fMETRIC_FORMULA_KIND_TIME_OFFSET\x10\t*\xda\x01\n\x11\x41lertSeverityKind\x12#\n\x1f\x41LERT_SEVERITY_KIND_UNSPECIFIED\x10\x00\x12 \n\x1c\x41LERT_SEVERITY_KIND_CRITICAL\x10\x01\x12\x1d\n\x19\x41LERT_SEVERITY_KIND_ERROR\x10\x02\x12\x1f\n\x1b\x41LERT_SEVERITY_KIND_WARNING\x10\x03\x12\x1c\n\x18\x41LERT_SEVERITY_KIND_INFO\x10\x04\x12 \n\x1c\x41LERT_SEVERITY_KIND_RESOLVED\x10\x05*\xdc\x01\n\rThresholdKind\x12\x1e\n\x1aTHRESHOLD_KIND_UNSPECIFIED\x10\x00\x12\x18\n\x14THRESHOLD_KIND_ABOVE\x10\x01\x12\x18\n\x14THRESHOLD_KIND_BELOW\x10\x02\x12 \n\x1cTHRESHOLD_KIND_GREATER_EQUAL\x10\x03\x12\x1d\n\x19THRESHOLD_KIND_LESS_EQUAL\x10\x04\x12\x18\n\x14THRESHOLD_KIND_EQUAL\x10\x05\x12\x1c\n\x18THRESHOLD_KIND_NOT_EQUAL\x10\x06*\xc7\x01\n\rChartLinkKind\x12\x1f\n\x1b\x43HART_LINK_KIND_UNSPECIFIED\x10\x00\x12\x1c\n\x18\x43HART_LINK_KIND_RESOLVER\x10\x01\x12\x1b\n\x17\x43HART_LINK_KIND_FEATURE\x10\x02\x12\x19\n\x15\x43HART_LINK_KIND_QUERY\x10\x03\x12\x1a\n\x16\x43HART_LINK_KIND_MANUAL\x10\x04\x12#\n\x1f\x43HART_LINK_KIND_SCHEDULED_QUERY\x10\x05\x42\x8e\x01\n\x16\x63om.chalk.artifacts.v1B\nChartProtoP\x01\xa2\x02\x03\x43\x41X\xaa\x02\x12\x43halk.Artifacts.V1\xca\x02\x12\x43halk\\Artifacts\\V1\xe2\x02\x1e\x43halk\\Artifacts\\V1\\GPBMetadata\xea\x02\x14\x43halk::Artifacts::V1b\x06proto3'
+    b'\n\x1e\x63halk/artifacts/v1/chart.proto\x12\x12\x63halk.artifacts.v1\x1a\x1egoogle/protobuf/duration.proto"\x86\x03\n\x0c\x41lertTrigger\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x41\n\x08severity\x18\x02 \x01(\x0e\x32%.chalk.artifacts.v1.AlertSeverityKindR\x08severity\x12P\n\x12threshold_position\x18\x03 \x01(\x0e\x32!.chalk.artifacts.v1.ThresholdKindR\x11thresholdPosition\x12\'\n\x0fthreshold_value\x18\x04 \x01(\x02R\x0ethresholdValue\x12$\n\x0bseries_name\x18\x05 \x01(\tH\x00R\nseriesName\x88\x01\x01\x12&\n\x0c\x63hannel_name\x18\x06 \x01(\tH\x01R\x0b\x63hannelName\x88\x01\x01\x12%\n\x0b\x64\x65scription\x18\x07 \x01(\tH\x02R\x0b\x64\x65scription\x88\x01\x01\x42\x0e\n\x0c_series_nameB\x0f\n\r_channel_nameB\x0e\n\x0c_description"K\n\x15\x44\x61tasetFeatureOperand\x12\x18\n\x07\x64\x61taset\x18\x01 \x01(\tR\x07\x64\x61taset\x12\x18\n\x07\x66\x65\x61ture\x18\x02 \x01(\tR\x07\x66\x65\x61ture"\xdb\x02\n\rMetricFormula\x12\x39\n\x04kind\x18\x01 \x01(\x0e\x32%.chalk.artifacts.v1.MetricFormulaKindR\x04kind\x12\x39\n\x16single_series_operands\x18\x02 \x01(\x03H\x00R\x14singleSeriesOperands\x88\x01\x01\x12\x32\n\x15multi_series_operands\x18\x03 \x03(\x03R\x13multiSeriesOperands\x12\x63\n\x18\x64\x61taset_feature_operands\x18\x04 \x01(\x0b\x32).chalk.artifacts.v1.DatasetFeatureOperandR\x16\x64\x61tasetFeatureOperands\x12\x17\n\x04name\x18\x05 \x01(\tH\x01R\x04name\x88\x01\x01\x42\x19\n\x17_single_series_operandsB\x07\n\x05_name"\x9c\x01\n\x0cMetricFilter\x12\x32\n\x04kind\x18\x01 \x01(\x0e\x32\x1e.chalk.artifacts.v1.FilterKindR\x04kind\x12\x42\n\ncomparator\x18\x02 \x01(\x0e\x32".chalk.artifacts.v1.ComparatorKindR\ncomparator\x12\x14\n\x05value\x18\x03 \x03(\tR\x05value"\x85\x03\n\x12MetricConfigSeries\x12\x36\n\x06metric\x18\x01 \x01(\x0e\x32\x1e.chalk.artifacts.v1.MetricKindR\x06metric\x12:\n\x07\x66ilters\x18\x02 \x03(\x0b\x32 .chalk.artifacts.v1.MetricFilterR\x07\x66ilters\x12\x17\n\x04name\x18\x03 \x01(\tH\x00R\x04name\x88\x01\x01\x12O\n\x0fwindow_function\x18\x04 \x01(\x0e\x32&.chalk.artifacts.v1.WindowFunctionKindR\x0ewindowFunction\x12:\n\x08group_by\x18\x05 \x03(\x0e\x32\x1f.chalk.artifacts.v1.GroupByKindR\x07groupBy\x12=\n\ntime_shift\x18\x06 \x01(\x0b\x32\x19.google.protobuf.DurationH\x01R\ttimeShift\x88\x01\x01\x42\x07\n\x05_nameB\r\n\x0b_time_shift"\xbb\x02\n\x0cMetricConfig\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12#\n\rwindow_period\x18\x02 \x01(\tR\x0cwindowPeriod\x12>\n\x06series\x18\x03 \x03(\x0b\x32&.chalk.artifacts.v1.MetricConfigSeriesR\x06series\x12=\n\x08\x66ormulas\x18\x04 \x03(\x0b\x32!.chalk.artifacts.v1.MetricFormulaR\x08\x66ormulas\x12:\n\x07trigger\x18\x05 \x01(\x0b\x32 .chalk.artifacts.v1.AlertTriggerR\x07trigger\x12\'\n\x0fgraph_generated\x18\x06 \x01(\x08R\x0egraphGenerated\x12\x0e\n\x02id\x18\x07 \x01(\tR\x02id"\xee\x01\n\x05\x43hart\x12\x0e\n\x02id\x18\x01 \x01(\tR\x02id\x12\x38\n\x06\x63onfig\x18\x02 \x01(\x0b\x32 .chalk.artifacts.v1.MetricConfigR\x06\x63onfig\x12\x42\n\x0b\x65ntity_kind\x18\x03 \x01(\x0e\x32!.chalk.artifacts.v1.ChartLinkKindR\nentityKind\x12 \n\tentity_id\x18\x04 \x01(\tH\x00R\x08\x65ntityId\x88\x01\x01\x12\'\n\x0fgraph_generated\x18\x05 \x01(\x08R\x0egraphGeneratedB\x0c\n\n_entity_id*\xe7\x10\n\nMetricKind\x12\x1b\n\x17METRIC_KIND_UNSPECIFIED\x10\x00\x12%\n!METRIC_KIND_FEATURE_REQUEST_COUNT\x10\x01\x12\x1f\n\x1bMETRIC_KIND_FEATURE_LATENCY\x10\x02\x12!\n\x1dMETRIC_KIND_FEATURE_STALENESS\x10\x03\x12\x1d\n\x19METRIC_KIND_FEATURE_VALUE\x10\x04\x12\x1d\n\x19METRIC_KIND_FEATURE_WRITE\x10\x05\x12"\n\x1eMETRIC_KIND_FEATURE_NULL_RATIO\x10\x06\x12&\n"METRIC_KIND_RESOLVER_REQUEST_COUNT\x10\x07\x12 \n\x1cMETRIC_KIND_RESOLVER_LATENCY\x10\x08\x12&\n"METRIC_KIND_RESOLVER_SUCCESS_RATIO\x10\t\x12\x1b\n\x17METRIC_KIND_QUERY_COUNT\x10\n\x12\x1d\n\x19METRIC_KIND_QUERY_LATENCY\x10\x0b\x12#\n\x1fMETRIC_KIND_QUERY_SUCCESS_RATIO\x10\x0c\x12!\n\x1dMETRIC_KIND_BILLING_INFERENCE\x10\r\x12\x1c\n\x18METRIC_KIND_BILLING_CRON\x10\x0e\x12!\n\x1dMETRIC_KIND_BILLING_MIGRATION\x10\x0f\x12\x1a\n\x16METRIC_KIND_CRON_COUNT\x10\x10\x12\x1c\n\x18METRIC_KIND_CRON_LATENCY\x10\x11\x12)\n%METRIC_KIND_STREAM_MESSAGES_PROCESSED\x10\x12\x12&\n"METRIC_KIND_STREAM_MESSAGE_LATENCY\x10\x13\x12(\n$METRIC_KIND_STREAM_WINDOWS_PROCESSED\x10\x14\x12%\n!METRIC_KIND_STREAM_WINDOW_LATENCY\x10\x15\x12&\n"METRIC_KIND_ONLINE_STORE_KEY_COUNT\x10\x16\x12.\n*METRIC_KIND_ONLINE_STORE_EXPIRED_KEY_COUNT\x10\x17\x12\x30\n,METRIC_KIND_ONLINE_STORE_REQUESTS_PER_SECOND\x10\x18\x12\'\n#METRIC_KIND_CPU_UTILIZATION_PERCENT\x10\x19\x12\x1d\n\x19METRIC_KIND_REPLICA_COUNT\x10\x1a\x12(\n$METRIC_KIND_ONLINE_STORE_USED_MEMORY\x10\x1b\x12)\n%METRIC_KIND_ONLINE_STORE_TOTAL_MEMORY\x10\x1c\x12"\n\x1eMETRIC_KIND_MEMORY_USAGE_BYTES\x10\x1d\x12,\n(METRIC_KIND_TOTAL_MEMORY_AVAILABLE_BYTES\x10\x1e\x12"\n\x1eMETRIC_KIND_NETWORK_READ_BYTES\x10\x1f\x12#\n\x1fMETRIC_KIND_NETWORK_WRITE_BYTES\x10 \x12\x1f\n\x1bMETRIC_KIND_DISK_READ_BYTES\x10!\x12 \n\x1cMETRIC_KIND_DISK_WRITE_BYTES\x10"\x12\x1a\n\x16METRIC_KIND_STREAM_LAG\x10#\x12\x15\n\x11METRIC_KIND_USAGE\x10$\x12&\n"METRIC_KIND_FEATURE_COMPUTED_COUNT\x10%\x12\'\n#METRIC_KIND_FEATURE_LOOKED_UP_COUNT\x10&\x12*\n&METRIC_KIND_FEATURE_INTERMEDIATE_COUNT\x10\'\x12+\n\'METRIC_KIND_FEATURE_COMPUTED_NULL_RATIO\x10(\x12,\n(METRIC_KIND_FEATURE_LOOKED_UP_NULL_RATIO\x10)\x12/\n+METRIC_KIND_FEATURE_INTERMEDIATE_NULL_RATIO\x10*\x12#\n\x1fMETRIC_KIND_STREAM_INGEST_DELAY\x10+\x12&\n"METRIC_KIND_CONTAINER_MEMORY_BYTES\x10,\x12!\n\x1dMETRIC_KIND_HOST_MEMORY_BYTES\x10-\x12)\n%METRIC_KIND_CONTAINER_CPU_UTILIZATION\x10.\x12\x1f\n\x1bMETRIC_KIND_DISK_USED_BYTES\x10/\x12$\n METRIC_KIND_DISK_AVAILABLE_BYTES\x10\x30\x12\'\n#METRIC_KIND_RESOLVER_INVOKER_NET_TX\x10\x31\x12\'\n#METRIC_KIND_RESOLVER_INVOKER_NET_RX\x10\x32\x12-\n)METRIC_KIND_RESOLVER_INVOKER_ROWS_WRITTEN\x10\x33\x12(\n$METRIC_KIND_TOPIC_MESSAGES_PROCESSED\x10\x34\x12\x31\n-METRIC_KIND_SUBSCRIPTION_NUM_UNACKED_MESSAGES\x10\x35\x12\x37\n3METRIC_KIND_SUBSCRIPTION_OLDEST_UNACKED_MESSAGE_AGE\x10\x36\x12 \n\x1cMETRIC_KIND_TOPIC_OFFSET_LAG\x10\x37*\xa7\x05\n\nFilterKind\x12\x1b\n\x17\x46ILTER_KIND_UNSPECIFIED\x10\x00\x12\x1e\n\x1a\x46ILTER_KIND_FEATURE_STATUS\x10\x01\x12\x1c\n\x18\x46ILTER_KIND_FEATURE_NAME\x10\x02\x12\x1b\n\x17\x46ILTER_KIND_FEATURE_TAG\x10\x03\x12\x1f\n\x1b\x46ILTER_KIND_RESOLVER_STATUS\x10\x04\x12\x1d\n\x19\x46ILTER_KIND_RESOLVER_NAME\x10\x05\x12\x1c\n\x18\x46ILTER_KIND_RESOLVER_TAG\x10\x06\x12\x1b\n\x17\x46ILTER_KIND_CRON_STATUS\x10\x07\x12 \n\x1c\x46ILTER_KIND_MIGRATION_STATUS\x10\x08\x12\x1e\n\x1a\x46ILTER_KIND_ONLINE_OFFLINE\x10\t\x12\x19\n\x15\x46ILTER_KIND_CACHE_HIT\x10\n\x12\x1c\n\x18\x46ILTER_KIND_OPERATION_ID\x10\x0b\x12\x1a\n\x16\x46ILTER_KIND_QUERY_NAME\x10\x0c\x12\x1c\n\x18\x46ILTER_KIND_QUERY_STATUS\x10\r\x12\x17\n\x13\x46ILTER_KIND_IS_NULL\x10\x0e\x12\x1a\n\x16\x46ILTER_KIND_USAGE_KIND\x10\x0f\x12\x1e\n\x1a\x46ILTER_KIND_RESOURCE_GROUP\x10\x10\x12\x18\n\x14\x46ILTER_KIND_POD_NAME\x10\x11\x12#\n\x1f\x46ILTER_KIND_COMPUTATION_CONTEXT\x10\x12\x12\x1a\n\x16\x46ILTER_KIND_TOPIC_NAME\x10\x13\x12!\n\x1d\x46ILTER_KIND_SUBSCRIPTION_NAME\x10\x14\x12\x1e\n\x1a\x46ILTER_KIND_PARTITION_NAME\x10\x15*~\n\x0e\x43omparatorKind\x12\x1f\n\x1b\x43OMPARATOR_KIND_UNSPECIFIED\x10\x00\x12\x16\n\x12\x43OMPARATOR_KIND_EQ\x10\x01\x12\x17\n\x13\x43OMPARATOR_KIND_NEQ\x10\x02\x12\x1a\n\x16\x43OMPARATOR_KIND_ONE_OF\x10\x03*\xec\x03\n\x12WindowFunctionKind\x12$\n WINDOW_FUNCTION_KIND_UNSPECIFIED\x10\x00\x12\x1e\n\x1aWINDOW_FUNCTION_KIND_COUNT\x10\x01\x12\x1d\n\x19WINDOW_FUNCTION_KIND_MEAN\x10\x02\x12\x1c\n\x18WINDOW_FUNCTION_KIND_SUM\x10\x03\x12\x1c\n\x18WINDOW_FUNCTION_KIND_MIN\x10\x04\x12\x1c\n\x18WINDOW_FUNCTION_KIND_MAX\x10\x05\x12&\n"WINDOW_FUNCTION_KIND_PERCENTILE_99\x10\x06\x12&\n"WINDOW_FUNCTION_KIND_PERCENTILE_95\x10\x07\x12&\n"WINDOW_FUNCTION_KIND_PERCENTILE_75\x10\x08\x12&\n"WINDOW_FUNCTION_KIND_PERCENTILE_50\x10\t\x12&\n"WINDOW_FUNCTION_KIND_PERCENTILE_25\x10\n\x12%\n!WINDOW_FUNCTION_KIND_PERCENTILE_5\x10\x0b\x12(\n$WINDOW_FUNCTION_KIND_ALL_PERCENTILES\x10\x0c*\xcc\x04\n\x0bGroupByKind\x12\x1d\n\x19GROUP_BY_KIND_UNSPECIFIED\x10\x00\x12 \n\x1cGROUP_BY_KIND_FEATURE_STATUS\x10\x01\x12\x1e\n\x1aGROUP_BY_KIND_FEATURE_NAME\x10\x02\x12\x19\n\x15GROUP_BY_KIND_IS_NULL\x10\x03\x12!\n\x1dGROUP_BY_KIND_RESOLVER_STATUS\x10\x04\x12\x1f\n\x1bGROUP_BY_KIND_RESOLVER_NAME\x10\x05\x12\x1e\n\x1aGROUP_BY_KIND_QUERY_STATUS\x10\x06\x12\x1c\n\x18GROUP_BY_KIND_QUERY_NAME\x10\x07\x12 \n\x1cGROUP_BY_KIND_ONLINE_OFFLINE\x10\x08\x12\x1b\n\x17GROUP_BY_KIND_CACHE_HIT\x10\t\x12\x1c\n\x18GROUP_BY_KIND_USAGE_KIND\x10\n\x12 \n\x1cGROUP_BY_KIND_RESOURCE_GROUP\x10\x0b\x12\x1f\n\x1bGROUP_BY_KIND_DEPLOYMENT_ID\x10\x0c\x12\x1e\n\x1aGROUP_BY_KIND_OPERATION_ID\x10\r\x12\x1a\n\x16GROUP_BY_KIND_POD_NAME\x10\x0e\x12\x1c\n\x18GROUP_BY_KIND_TOPIC_NAME\x10\x0f\x12#\n\x1fGROUP_BY_KIND_SUBSCRIPTION_NAME\x10\x10\x12 \n\x1cGROUP_BY_KIND_PARTITION_NAME\x10\x11*\xe4\x02\n\x11MetricFormulaKind\x12#\n\x1fMETRIC_FORMULA_KIND_UNSPECIFIED\x10\x00\x12\x1b\n\x17METRIC_FORMULA_KIND_SUM\x10\x01\x12#\n\x1fMETRIC_FORMULA_KIND_TOTAL_RATIO\x10\x02\x12\x1d\n\x19METRIC_FORMULA_KIND_RATIO\x10\x03\x12\x1f\n\x1bMETRIC_FORMULA_KIND_PRODUCT\x10\x04\x12\x1b\n\x17METRIC_FORMULA_KIND_ABS\x10\x05\x12\x1f\n\x1bMETRIC_FORMULA_KIND_KS_STAT\x10\x06\x12\x1f\n\x1bMETRIC_FORMULA_KIND_KS_TEST\x10\x07\x12$\n METRIC_FORMULA_KIND_KS_THRESHOLD\x10\x08\x12#\n\x1fMETRIC_FORMULA_KIND_TIME_OFFSET\x10\t*\xda\x01\n\x11\x41lertSeverityKind\x12#\n\x1f\x41LERT_SEVERITY_KIND_UNSPECIFIED\x10\x00\x12 \n\x1c\x41LERT_SEVERITY_KIND_CRITICAL\x10\x01\x12\x1d\n\x19\x41LERT_SEVERITY_KIND_ERROR\x10\x02\x12\x1f\n\x1b\x41LERT_SEVERITY_KIND_WARNING\x10\x03\x12\x1c\n\x18\x41LERT_SEVERITY_KIND_INFO\x10\x04\x12 \n\x1c\x41LERT_SEVERITY_KIND_RESOLVED\x10\x05*\xdc\x01\n\rThresholdKind\x12\x1e\n\x1aTHRESHOLD_KIND_UNSPECIFIED\x10\x00\x12\x18\n\x14THRESHOLD_KIND_ABOVE\x10\x01\x12\x18\n\x14THRESHOLD_KIND_BELOW\x10\x02\x12 \n\x1cTHRESHOLD_KIND_GREATER_EQUAL\x10\x03\x12\x1d\n\x19THRESHOLD_KIND_LESS_EQUAL\x10\x04\x12\x18\n\x14THRESHOLD_KIND_EQUAL\x10\x05\x12\x1c\n\x18THRESHOLD_KIND_NOT_EQUAL\x10\x06*\xc7\x01\n\rChartLinkKind\x12\x1f\n\x1b\x43HART_LINK_KIND_UNSPECIFIED\x10\x00\x12\x1c\n\x18\x43HART_LINK_KIND_RESOLVER\x10\x01\x12\x1b\n\x17\x43HART_LINK_KIND_FEATURE\x10\x02\x12\x19\n\x15\x43HART_LINK_KIND_QUERY\x10\x03\x12\x1a\n\x16\x43HART_LINK_KIND_MANUAL\x10\x04\x12#\n\x1f\x43HART_LINK_KIND_SCHEDULED_QUERY\x10\x05\x42\x8e\x01\n\x16\x63om.chalk.artifacts.v1B\nChartProtoP\x01\xa2\x02\x03\x43\x41X\xaa\x02\x12\x43halk.Artifacts.V1\xca\x02\x12\x43halk\\Artifacts\\V1\xe2\x02\x1e\x43halk\\Artifacts\\V1\\GPBMetadata\xea\x02\x14\x43halk::Artifacts::V1b\x06proto3'
 )
 _globals = globals()
@@ -31,21 +31,21 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_METRICKIND"]._serialized_start = 2017
     _globals["_METRICKIND"]._serialized_end = 4168
     _globals["_FILTERKIND"]._serialized_start = 4171
-    _globals["_FILTERKIND"]._serialized_end = 4818
-    _globals["_COMPARATORKIND"]._serialized_start = 4820
-    _globals["_COMPARATORKIND"]._serialized_end = 4946
-    _globals["_WINDOWFUNCTIONKIND"]._serialized_start = 4949
-    _globals["_WINDOWFUNCTIONKIND"]._serialized_end = 5441
-    _globals["_GROUPBYKIND"]._serialized_start = 5444
-    _globals["_GROUPBYKIND"]._serialized_end = 5998
-    _globals["_METRICFORMULAKIND"]._serialized_start = 6001
-    _globals["_METRICFORMULAKIND"]._serialized_end = 6357
-    _globals["_ALERTSEVERITYKIND"]._serialized_start = 6360
-    _globals["_ALERTSEVERITYKIND"]._serialized_end = 6578
-    _globals["_THRESHOLDKIND"]._serialized_start = 6581
-    _globals["_THRESHOLDKIND"]._serialized_end = 6801
-    _globals["_CHARTLINKKIND"]._serialized_start = 6804
-    _globals["_CHARTLINKKIND"]._serialized_end = 7003
+    _globals["_FILTERKIND"]._serialized_end = 4850
+    _globals["_COMPARATORKIND"]._serialized_start = 4852
+    _globals["_COMPARATORKIND"]._serialized_end = 4978
+    _globals["_WINDOWFUNCTIONKIND"]._serialized_start = 4981
+    _globals["_WINDOWFUNCTIONKIND"]._serialized_end = 5473
+    _globals["_GROUPBYKIND"]._serialized_start = 5476
+    _globals["_GROUPBYKIND"]._serialized_end = 6064
+    _globals["_METRICFORMULAKIND"]._serialized_start = 6067
+    _globals["_METRICFORMULAKIND"]._serialized_end = 6423
+    _globals["_ALERTSEVERITYKIND"]._serialized_start = 6426
+    _globals["_ALERTSEVERITYKIND"]._serialized_end = 6644
+    _globals["_THRESHOLDKIND"]._serialized_start = 6647
+    _globals["_THRESHOLDKIND"]._serialized_end = 6867
+    _globals["_CHARTLINKKIND"]._serialized_start = 6870
+    _globals["_CHARTLINKKIND"]._serialized_end = 7069
     _globals["_ALERTTRIGGER"]._serialized_start = 87
     _globals["_ALERTTRIGGER"]._serialized_end = 477
     _globals["_DATASETFEATUREOPERAND"]._serialized_start = 479

chalk/_gen/chalk/artifacts/v1/chart_pb2.pyi CHANGED Viewed

@@ -95,6 +95,7 @@ class FilterKind(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     FILTER_KIND_COMPUTATION_CONTEXT: _ClassVar[FilterKind]
     FILTER_KIND_TOPIC_NAME: _ClassVar[FilterKind]
     FILTER_KIND_SUBSCRIPTION_NAME: _ClassVar[FilterKind]
+    FILTER_KIND_PARTITION_NAME: _ClassVar[FilterKind]
 class ComparatorKind(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     __slots__ = ()
@@ -138,6 +139,7 @@ class GroupByKind(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     GROUP_BY_KIND_POD_NAME: _ClassVar[GroupByKind]
     GROUP_BY_KIND_TOPIC_NAME: _ClassVar[GroupByKind]
     GROUP_BY_KIND_SUBSCRIPTION_NAME: _ClassVar[GroupByKind]
+    GROUP_BY_KIND_PARTITION_NAME: _ClassVar[GroupByKind]
 class MetricFormulaKind(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
     __slots__ = ()
@@ -257,6 +259,7 @@ FILTER_KIND_POD_NAME: FilterKind
 FILTER_KIND_COMPUTATION_CONTEXT: FilterKind
 FILTER_KIND_TOPIC_NAME: FilterKind
 FILTER_KIND_SUBSCRIPTION_NAME: FilterKind
+FILTER_KIND_PARTITION_NAME: FilterKind
 COMPARATOR_KIND_UNSPECIFIED: ComparatorKind
 COMPARATOR_KIND_EQ: ComparatorKind
 COMPARATOR_KIND_NEQ: ComparatorKind
@@ -291,6 +294,7 @@ GROUP_BY_KIND_OPERATION_ID: GroupByKind
 GROUP_BY_KIND_POD_NAME: GroupByKind
 GROUP_BY_KIND_TOPIC_NAME: GroupByKind
 GROUP_BY_KIND_SUBSCRIPTION_NAME: GroupByKind
+GROUP_BY_KIND_PARTITION_NAME: GroupByKind
 METRIC_FORMULA_KIND_UNSPECIFIED: MetricFormulaKind
 METRIC_FORMULA_KIND_SUM: MetricFormulaKind
 METRIC_FORMULA_KIND_TOTAL_RATIO: MetricFormulaKind

chalk/_gen/chalk/artifacts/v1/cron_query_pb2.py CHANGED Viewed

@@ -13,11 +13,12 @@ from google.protobuf.internal import builder as _builder
 _sym_db = _symbol_database.Default()
+from google.protobuf import duration_pb2 as google_dot_protobuf_dot_duration__pb2
 from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n#chalk/artifacts/v1/cron_query.proto\x12\x12\x63halk.artifacts.v1\x1a\x1fgoogle/protobuf/timestamp.proto"Y\n\x11RecomputeSettings\x12!\n\x0c\x66\x65\x61ture_fqns\x18\x01 \x03(\tR\x0b\x66\x65\x61tureFqns\x12!\n\x0c\x61ll_features\x18\x02 \x01(\x08R\x0b\x61llFeatures"\xfe\x05\n\tCronQuery\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x12\n\x04\x63ron\x18\x02 \x01(\tR\x04\x63ron\x12\x1b\n\tfile_name\x18\x03 \x01(\tR\x08\x66ileName\x12\x16\n\x06output\x18\x04 \x03(\tR\x06output\x12$\n\x0bmax_samples\x18\x05 \x01(\x03H\x00R\nmaxSamples\x88\x01\x01\x12\x43\n\trecompute\x18\x06 \x01(\x0b\x32%.chalk.artifacts.v1.RecomputeSettingsR\trecompute\x12;\n\x0blower_bound\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\nlowerBound\x12;\n\x0bupper_bound\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\nupperBound\x12\x12\n\x04tags\x18\t \x03(\tR\x04tags\x12\x34\n\x16required_resolver_tags\x18\n \x03(\tR\x14requiredResolverTags\x12!\n\x0cstore_online\x18\x0b \x01(\x08R\x0bstoreOnline\x12#\n\rstore_offline\x18\x0c \x01(\x08R\x0cstoreOffline\x12/\n\x13incremental_sources\x18\r \x03(\tR\x12incrementalSources\x12*\n\x0eresource_group\x18\x0e \x01(\tH\x01R\rresourceGroup\x88\x01\x01\x12Z\n\x0fplanner_options\x18\x0f \x03(\x0b\x32\x31.chalk.artifacts.v1.CronQuery.PlannerOptionsEntryR\x0eplannerOptions\x1a\x41\n\x13PlannerOptionsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x0e\n\x0c_max_samplesB\x11\n\x0f_resource_groupB\x92\x01\n\x16\x63om.chalk.artifacts.v1B\x0e\x43ronQueryProtoP\x01\xa2\x02\x03\x43\x41X\xaa\x02\x12\x43halk.Artifacts.V1\xca\x02\x12\x43halk\\Artifacts\\V1\xe2\x02\x1e\x43halk\\Artifacts\\V1\\GPBMetadata\xea\x02\x14\x43halk::Artifacts::V1b\x06proto3'
+    b'\n#chalk/artifacts/v1/cron_query.proto\x12\x12\x63halk.artifacts.v1\x1a\x1egoogle/protobuf/duration.proto\x1a\x1fgoogle/protobuf/timestamp.proto"Y\n\x11RecomputeSettings\x12!\n\x0c\x66\x65\x61ture_fqns\x18\x01 \x03(\tR\x0b\x66\x65\x61tureFqns\x12!\n\x0c\x61ll_features\x18\x02 \x01(\x08R\x0b\x61llFeatures"\xe7\x06\n\tCronQuery\x12\x12\n\x04name\x18\x01 \x01(\tR\x04name\x12\x12\n\x04\x63ron\x18\x02 \x01(\tR\x04\x63ron\x12\x1b\n\tfile_name\x18\x03 \x01(\tR\x08\x66ileName\x12\x16\n\x06output\x18\x04 \x03(\tR\x06output\x12$\n\x0bmax_samples\x18\x05 \x01(\x03H\x00R\nmaxSamples\x88\x01\x01\x12\x43\n\trecompute\x18\x06 \x01(\x0b\x32%.chalk.artifacts.v1.RecomputeSettingsR\trecompute\x12;\n\x0blower_bound\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\nlowerBound\x12;\n\x0bupper_bound\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\nupperBound\x12\x12\n\x04tags\x18\t \x03(\tR\x04tags\x12\x34\n\x16required_resolver_tags\x18\n \x03(\tR\x14requiredResolverTags\x12!\n\x0cstore_online\x18\x0b \x01(\x08R\x0bstoreOnline\x12#\n\rstore_offline\x18\x0c \x01(\x08R\x0cstoreOffline\x12/\n\x13incremental_sources\x18\r \x03(\tR\x12incrementalSources\x12*\n\x0eresource_group\x18\x0e \x01(\tH\x01R\rresourceGroup\x88\x01\x01\x12Z\n\x0fplanner_options\x18\x0f \x03(\x0b\x32\x31.chalk.artifacts.v1.CronQuery.PlannerOptionsEntryR\x0eplannerOptions\x12O\n\x13\x63ompletion_deadline\x18\x10 \x01(\x0b\x32\x19.google.protobuf.DurationH\x02R\x12\x63ompletionDeadline\x88\x01\x01\x1a\x41\n\x13PlannerOptionsEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\x05value:\x02\x38\x01\x42\x0e\n\x0c_max_samplesB\x11\n\x0f_resource_groupB\x16\n\x14_completion_deadlineB\x92\x01\n\x16\x63om.chalk.artifacts.v1B\x0e\x43ronQueryProtoP\x01\xa2\x02\x03\x43\x41X\xaa\x02\x12\x43halk.Artifacts.V1\xca\x02\x12\x43halk\\Artifacts\\V1\xe2\x02\x1e\x43halk\\Artifacts\\V1\\GPBMetadata\xea\x02\x14\x43halk::Artifacts::V1b\x06proto3'
 )
 _globals = globals()
@@ -30,10 +31,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     ]._serialized_options = b"\n\026com.chalk.artifacts.v1B\016CronQueryProtoP\001\242\002\003CAX\252\002\022Chalk.Artifacts.V1\312\002\022Chalk\\Artifacts\\V1\342\002\036Chalk\\Artifacts\\V1\\GPBMetadata\352\002\024Chalk::Artifacts::V1"
     _globals["_CRONQUERY_PLANNEROPTIONSENTRY"]._options = None
     _globals["_CRONQUERY_PLANNEROPTIONSENTRY"]._serialized_options = b"8\001"
-    _globals["_RECOMPUTESETTINGS"]._serialized_start = 92
-    _globals["_RECOMPUTESETTINGS"]._serialized_end = 181
-    _globals["_CRONQUERY"]._serialized_start = 184
-    _globals["_CRONQUERY"]._serialized_end = 950
-    _globals["_CRONQUERY_PLANNEROPTIONSENTRY"]._serialized_start = 850
-    _globals["_CRONQUERY_PLANNEROPTIONSENTRY"]._serialized_end = 915
+    _globals["_RECOMPUTESETTINGS"]._serialized_start = 124
+    _globals["_RECOMPUTESETTINGS"]._serialized_end = 213
+    _globals["_CRONQUERY"]._serialized_start = 216
+    _globals["_CRONQUERY"]._serialized_end = 1087
+    _globals["_CRONQUERY_PLANNEROPTIONSENTRY"]._serialized_start = 963
+    _globals["_CRONQUERY_PLANNEROPTIONSENTRY"]._serialized_end = 1028
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/artifacts/v1/cron_query_pb2.pyi CHANGED Viewed

@@ -1,3 +1,4 @@
+from google.protobuf import duration_pb2 as _duration_pb2
 from google.protobuf import timestamp_pb2 as _timestamp_pb2
 from google.protobuf.internal import containers as _containers
 from google.protobuf import descriptor as _descriptor
@@ -37,6 +38,7 @@ class CronQuery(_message.Message):
         "incremental_sources",
         "resource_group",
         "planner_options",
+        "completion_deadline",
     )
     class PlannerOptionsEntry(_message.Message):
         __slots__ = ("key", "value")
@@ -61,6 +63,7 @@ class CronQuery(_message.Message):
     INCREMENTAL_SOURCES_FIELD_NUMBER: _ClassVar[int]
     RESOURCE_GROUP_FIELD_NUMBER: _ClassVar[int]
     PLANNER_OPTIONS_FIELD_NUMBER: _ClassVar[int]
+    COMPLETION_DEADLINE_FIELD_NUMBER: _ClassVar[int]
     name: str
     cron: str
     file_name: str
@@ -76,6 +79,7 @@ class CronQuery(_message.Message):
     incremental_sources: _containers.RepeatedScalarFieldContainer[str]
     resource_group: str
     planner_options: _containers.ScalarMap[str, str]
+    completion_deadline: _duration_pb2.Duration
     def __init__(
         self,
         name: _Optional[str] = ...,
@@ -93,4 +97,5 @@ class CronQuery(_message.Message):
         incremental_sources: _Optional[_Iterable[str]] = ...,
         resource_group: _Optional[str] = ...,
         planner_options: _Optional[_Mapping[str, str]] = ...,
+        completion_deadline: _Optional[_Union[_duration_pb2.Duration, _Mapping]] = ...,
     ) -> None: ...

chalk/_gen/chalk/common/v1/offline_query_pb2.py CHANGED Viewed

@@ -21,7 +21,7 @@ from google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
-    b'\n#chalk/common/v1/offline_query.proto\x12\x0f\x63halk.common.v1\x1a!chalk/common/v1/chalk_error.proto\x1a"chalk/common/v1/online_query.proto\x1a$chalk/expression/v1/expression.proto\x1a\x1a\x63halk/graph/v1/graph.proto\x1a\x1cgoogle/protobuf/struct.proto"\xdc\x01\n\x1dOfflineQueryRecomputeFeatures\x12 \n\x0b\x61ll_or_none\x18\x01 \x01(\x08H\x00R\tallOrNone\x12_\n\x0c\x66\x65\x61ture_list\x18\x02 \x01(\x0b\x32:.chalk.common.v1.OfflineQueryRecomputeFeatures.FeatureListH\x00R\x0b\x66\x65\x61tureList\x1a\x30\n\x0b\x46\x65\x61tureList\x12!\n\x0c\x66\x65\x61ture_list\x18\x01 \x03(\tR\x0b\x66\x65\x61tureListB\x06\n\x04impl"\x80\x01\n\x13OfflineQueryExplain\x12\x18\n\x06truthy\x18\x01 \x01(\x08H\x00R\x06truthy\x12?\n\x04only\x18\x02 \x01(\x0b\x32).chalk.common.v1.OfflineQueryExplain.OnlyH\x00R\x04only\x1a\x06\n\x04OnlyB\x06\n\x04impl"a\n\x11OfflineQueryInput\x12\x18\n\x07\x63olumns\x18\x01 \x03(\tR\x07\x63olumns\x12\x32\n\x06values\x18\x02 \x03(\x0b\x32\x1a.google.protobuf.ListValueR\x06values"V\n\x18OfflineQueryInputSharded\x12:\n\x06inputs\x18\x01 \x03(\x0b\x32".chalk.common.v1.OfflineQueryInputR\x06inputs"_\n%OfflineQueryShardedParquetUploadInput\x12\x1c\n\tfilenames\x18\x01 \x03(\tR\tfilenames\x12\x18\n\x07version\x18\x02 \x01(\x05R\x07version"\xba\x03\n\x12OfflineQueryInputs\x12\'\n\x0e\x66\x65\x61ther_inputs\x18\x01 \x01(\x0cH\x00R\rfeatherInputs\x12K\n\tno_inputs\x18\x02 \x01(\x0b\x32,.chalk.common.v1.OfflineQueryInputs.NoInputsH\x00R\x08noInputs\x12I\n\rsingle_inputs\x18\x03 \x01(\x0b\x32".chalk.common.v1.OfflineQueryInputH\x00R\x0csingleInputs\x12R\n\x0esharded_inputs\x18\x04 \x01(\x0b\x32).chalk.common.v1.OfflineQueryInputShardedH\x00R\rshardedInputs\x12{\n\x1dsharded_parquet_upload_inputs\x18\x05 \x01(\x0b\x32\x36.chalk.common.v1.OfflineQueryShardedParquetUploadInputH\x00R\x1ashardedParquetUploadInputs\x1a\n\n\x08NoInputsB\x06\n\x04impl"\x96\x11\n\x13OfflineQueryRequest\x12;\n\x06inputs\x18\x01 \x01(\x0b\x32#.chalk.common.v1.OfflineQueryInputsR\x06inputs\x12\x18\n\x07outputs\x18\x02 \x03(\tR\x07outputs\x12)\n\x10required_outputs\x18\x03 \x03(\tR\x0frequiredOutputs\x12-\n\x12\x64\x65stination_format\x18\x04 \x01(\tR\x11\x64\x65stinationFormat\x12\x1b\n\x06\x62ranch\x18\x05 \x01(\tH\x00R\x06\x62ranch\x88\x01\x01\x12&\n\x0c\x64\x61taset_name\x18\x06 \x01(\tH\x01R\x0b\x64\x61tasetName\x88\x01\x01\x12]\n\x12recompute_features\x18\x07 \x01(\x0b\x32..chalk.common.v1.OfflineQueryRecomputeFeaturesR\x11recomputeFeatures\x12*\n\x11store_plan_stages\x18\x08 \x01(\x08R\x0fstorePlanStages\x12>\n\x07\x66ilters\x18\x0b \x03(\x0b\x32$.chalk.expression.v1.LogicalExprNodeR\x07\x66ilters\x12+\n\x0fspine_sql_query\x18\x0c \x01(\tH\x02R\rspineSqlQuery\x88\x01\x01\x12$\n\x0bmax_samples\x18\x65 \x01(\x05H\x03R\nmaxSamples\x88\x01\x01\x12\x30\n\x12max_cache_age_secs\x18\x66 \x01(\x05H\x04R\x0fmaxCacheAgeSecs\x88\x01\x01\x12\x42\n\x07\x65xplain\x18g \x01(\x0b\x32$.chalk.common.v1.OfflineQueryExplainB\x02\x18\x01R\x07\x65xplain\x12;\n\x08\x65xplain2\x18j \x01(\x0b\x32\x1f.chalk.common.v1.ExplainOptionsR\x08\x65xplain2\x12\x12\n\x04tags\x18h \x03(\tR\x04tags\x12*\n\x0e\x63orrelation_id\x18i \x01(\tH\x05R\rcorrelationId\x88\x01\x01\x12\x34\n\x16required_resolver_tags\x18k \x03(\tR\x14requiredResolverTags\x12;\n\x17observed_at_lower_bound\x18\xc9\x01 \x01(\tH\x06R\x14observedAtLowerBound\x88\x01\x01\x12;\n\x17observed_at_upper_bound\x18\xca\x01 \x01(\tH\x07R\x14observedAtUpperBound\x88\x01\x01\x12\x62\n\x0fplanner_options\x18\xcb\x01 \x03(\x0b\x32\x38.chalk.common.v1.OfflineQueryRequest.PlannerOptionsEntryR\x0eplannerOptions\x12\\\n\renv_overrides\x18\xd5\x01 \x03(\x0b\x32\x36.chalk.common.v1.OfflineQueryRequest.EnvOverridesEntryR\x0c\x65nvOverrides\x12\'\n\x0cstore_online\x18\xcc\x01 \x01(\x08H\x08R\x0bstoreOnline\x88\x01\x01\x12)\n\rstore_offline\x18\xcd\x01 \x01(\x08H\tR\x0cstoreOffline\x88\x01\x01\x12:\n\x16use_multiple_computers\x18\xce\x01 \x01(\x08H\nR\x14useMultipleComputers\x88\x01\x01\x12#\n\nnum_shards\x18\xcf\x01 \x01(\x05H\x0bR\tnumShards\x88\x01\x01\x12%\n\x0bnum_workers\x18\xd0\x01 \x01(\x05H\x0cR\nnumWorkers\x88\x01\x01\x12\\\n\rquery_context\x18\xd1\x01 \x03(\x0b\x32\x36.chalk.common.v1.OfflineQueryRequest.QueryContextEntryR\x0cqueryContext\x12G\n\roverlay_graph\x18\xd2\x01 \x01(\x0b\x32\x1c.chalk.graph.v1.OverlayGraphH\rR\x0coverlayGraph\x88\x01\x01\x12#\n\nquery_name\x18\xd3\x01 \x01(\tH\x0eR\tqueryName\x88\x01\x01\x12\x32\n\x12query_name_version\x18\xd4\x01 \x01(\tH\x0fR\x10queryNameVersion\x88\x01\x01\x1aY\n\x13PlannerOptionsEntry\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\x11\x45nvOverridesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\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\t\n\x07_branchB\x0f\n\r_dataset_nameB\x12\n\x10_spine_sql_queryB\x0e\n\x0c_max_samplesB\x15\n\x13_max_cache_age_secsB\x11\n\x0f_correlation_idB\x1a\n\x18_observed_at_lower_boundB\x1a\n\x18_observed_at_upper_boundB\x0f\n\r_store_onlineB\x10\n\x0e_store_offlineB\x19\n\x17_use_multiple_computersB\r\n\x0b_num_shardsB\x0e\n\x0c_num_workersB\x10\n\x0e_overlay_graphB\r\n\x0b_query_nameB\x15\n\x13_query_name_version"\xce\x01\n\x12\x43olumnMetadataList\x12N\n\x08metadata\x18\x01 \x03(\x0b\x32\x32.chalk.common.v1.ColumnMetadataList.ColumnMetadataR\x08metadata\x1ah\n\x0e\x43olumnMetadata\x12\x1f\n\x0b\x66\x65\x61ture_fqn\x18\x01 \x01(\tR\nfeatureFqn\x12\x1f\n\x0b\x63olumn_name\x18\x02 \x01(\tR\ncolumnName\x12\x14\n\x05\x64type\x18\x03 \x01(\tR\x05\x64type"\xf0\x01\n\x1aGetOfflineQueryJobResponse\x12\x1f\n\x0bis_finished\x18\x01 \x01(\x08R\nisFinished\x12\x18\n\x07version\x18\x02 \x01(\x05R\x07version\x12\x12\n\x04urls\x18\x03 \x03(\tR\x04urls\x12\x33\n\x06\x65rrors\x18\x04 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors\x12\x42\n\x07\x63olumns\x18\x05 \x01(\x0b\x32#.chalk.common.v1.ColumnMetadataListH\x00R\x07\x63olumns\x88\x01\x01\x42\n\n\x08_columnsB\x86\x01\n\x13\x63om.chalk.common.v1B\x11OfflineQueryProtoP\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/offline_query.proto\x12\x0f\x63halk.common.v1\x1a!chalk/common/v1/chalk_error.proto\x1a"chalk/common/v1/online_query.proto\x1a$chalk/expression/v1/expression.proto\x1a\x1a\x63halk/graph/v1/graph.proto\x1a\x1cgoogle/protobuf/struct.proto"\xdc\x01\n\x1dOfflineQueryRecomputeFeatures\x12 \n\x0b\x61ll_or_none\x18\x01 \x01(\x08H\x00R\tallOrNone\x12_\n\x0c\x66\x65\x61ture_list\x18\x02 \x01(\x0b\x32:.chalk.common.v1.OfflineQueryRecomputeFeatures.FeatureListH\x00R\x0b\x66\x65\x61tureList\x1a\x30\n\x0b\x46\x65\x61tureList\x12!\n\x0c\x66\x65\x61ture_list\x18\x01 \x03(\tR\x0b\x66\x65\x61tureListB\x06\n\x04impl"\x80\x01\n\x13OfflineQueryExplain\x12\x18\n\x06truthy\x18\x01 \x01(\x08H\x00R\x06truthy\x12?\n\x04only\x18\x02 \x01(\x0b\x32).chalk.common.v1.OfflineQueryExplain.OnlyH\x00R\x04only\x1a\x06\n\x04OnlyB\x06\n\x04impl"a\n\x11OfflineQueryInput\x12\x18\n\x07\x63olumns\x18\x01 \x03(\tR\x07\x63olumns\x12\x32\n\x06values\x18\x02 \x03(\x0b\x32\x1a.google.protobuf.ListValueR\x06values"V\n\x18OfflineQueryInputSharded\x12:\n\x06inputs\x18\x01 \x03(\x0b\x32".chalk.common.v1.OfflineQueryInputR\x06inputs"_\n%OfflineQueryShardedParquetUploadInput\x12\x1c\n\tfilenames\x18\x01 \x03(\tR\tfilenames\x12\x18\n\x07version\x18\x02 \x01(\x05R\x07version"\xba\x03\n\x12OfflineQueryInputs\x12\'\n\x0e\x66\x65\x61ther_inputs\x18\x01 \x01(\x0cH\x00R\rfeatherInputs\x12K\n\tno_inputs\x18\x02 \x01(\x0b\x32,.chalk.common.v1.OfflineQueryInputs.NoInputsH\x00R\x08noInputs\x12I\n\rsingle_inputs\x18\x03 \x01(\x0b\x32".chalk.common.v1.OfflineQueryInputH\x00R\x0csingleInputs\x12R\n\x0esharded_inputs\x18\x04 \x01(\x0b\x32).chalk.common.v1.OfflineQueryInputShardedH\x00R\rshardedInputs\x12{\n\x1dsharded_parquet_upload_inputs\x18\x05 \x01(\x0b\x32\x36.chalk.common.v1.OfflineQueryShardedParquetUploadInputH\x00R\x1ashardedParquetUploadInputs\x1a\n\n\x08NoInputsB\x06\n\x04impl"\xb3\x02\n\x10ResourceRequests\x12\x15\n\x03\x63pu\x18\x01 \x01(\tH\x00R\x03\x63pu\x88\x01\x01\x12\x1b\n\x06memory\x18\x02 \x01(\tH\x01R\x06memory\x88\x01\x01\x12\x37\n\x15\x65phemeral_volume_size\x18\x03 \x01(\tH\x02R\x13\x65phemeralVolumeSize\x88\x01\x01\x12\x30\n\x11\x65phemeral_storage\x18\x04 \x01(\tH\x03R\x10\x65phemeralStorage\x88\x01\x01\x12*\n\x0eresource_group\x18\x05 \x01(\tH\x04R\rresourceGroup\x88\x01\x01\x42\x06\n\x04_cpuB\t\n\x07_memoryB\x18\n\x16_ephemeral_volume_sizeB\x14\n\x12_ephemeral_storageB\x11\n\x0f_resource_group"\xeb\x11\n\x13OfflineQueryRequest\x12;\n\x06inputs\x18\x01 \x01(\x0b\x32#.chalk.common.v1.OfflineQueryInputsR\x06inputs\x12\x18\n\x07outputs\x18\x02 \x03(\tR\x07outputs\x12)\n\x10required_outputs\x18\x03 \x03(\tR\x0frequiredOutputs\x12-\n\x12\x64\x65stination_format\x18\x04 \x01(\tR\x11\x64\x65stinationFormat\x12\x1b\n\x06\x62ranch\x18\x05 \x01(\tH\x00R\x06\x62ranch\x88\x01\x01\x12&\n\x0c\x64\x61taset_name\x18\x06 \x01(\tH\x01R\x0b\x64\x61tasetName\x88\x01\x01\x12]\n\x12recompute_features\x18\x07 \x01(\x0b\x32..chalk.common.v1.OfflineQueryRecomputeFeaturesR\x11recomputeFeatures\x12*\n\x11store_plan_stages\x18\x08 \x01(\x08R\x0fstorePlanStages\x12>\n\x07\x66ilters\x18\x0b \x03(\x0b\x32$.chalk.expression.v1.LogicalExprNodeR\x07\x66ilters\x12+\n\x0fspine_sql_query\x18\x0c \x01(\tH\x02R\rspineSqlQuery\x88\x01\x01\x12$\n\x0bmax_samples\x18\x65 \x01(\x05H\x03R\nmaxSamples\x88\x01\x01\x12\x30\n\x12max_cache_age_secs\x18\x66 \x01(\x05H\x04R\x0fmaxCacheAgeSecs\x88\x01\x01\x12\x42\n\x07\x65xplain\x18g \x01(\x0b\x32$.chalk.common.v1.OfflineQueryExplainB\x02\x18\x01R\x07\x65xplain\x12;\n\x08\x65xplain2\x18j \x01(\x0b\x32\x1f.chalk.common.v1.ExplainOptionsR\x08\x65xplain2\x12\x12\n\x04tags\x18h \x03(\tR\x04tags\x12*\n\x0e\x63orrelation_id\x18i \x01(\tH\x05R\rcorrelationId\x88\x01\x01\x12\x34\n\x16required_resolver_tags\x18k \x03(\tR\x14requiredResolverTags\x12;\n\x17observed_at_lower_bound\x18\xc9\x01 \x01(\tH\x06R\x14observedAtLowerBound\x88\x01\x01\x12;\n\x17observed_at_upper_bound\x18\xca\x01 \x01(\tH\x07R\x14observedAtUpperBound\x88\x01\x01\x12\x62\n\x0fplanner_options\x18\xcb\x01 \x03(\x0b\x32\x38.chalk.common.v1.OfflineQueryRequest.PlannerOptionsEntryR\x0eplannerOptions\x12\\\n\renv_overrides\x18\xd5\x01 \x03(\x0b\x32\x36.chalk.common.v1.OfflineQueryRequest.EnvOverridesEntryR\x0c\x65nvOverrides\x12\'\n\x0cstore_online\x18\xcc\x01 \x01(\x08H\x08R\x0bstoreOnline\x88\x01\x01\x12)\n\rstore_offline\x18\xcd\x01 \x01(\x08H\tR\x0cstoreOffline\x88\x01\x01\x12:\n\x16use_multiple_computers\x18\xce\x01 \x01(\x08H\nR\x14useMultipleComputers\x88\x01\x01\x12#\n\nnum_shards\x18\xcf\x01 \x01(\x05H\x0bR\tnumShards\x88\x01\x01\x12%\n\x0bnum_workers\x18\xd0\x01 \x01(\x05H\x0cR\nnumWorkers\x88\x01\x01\x12\\\n\rquery_context\x18\xd1\x01 \x03(\x0b\x32\x36.chalk.common.v1.OfflineQueryRequest.QueryContextEntryR\x0cqueryContext\x12G\n\roverlay_graph\x18\xd2\x01 \x01(\x0b\x32\x1c.chalk.graph.v1.OverlayGraphH\rR\x0coverlayGraph\x88\x01\x01\x12#\n\nquery_name\x18\xd3\x01 \x01(\tH\x0eR\tqueryName\x88\x01\x01\x12\x32\n\x12query_name_version\x18\xd4\x01 \x01(\tH\x0fR\x10queryNameVersion\x88\x01\x01\x12\x45\n\tresources\x18\xd6\x01 \x01(\x0b\x32!.chalk.common.v1.ResourceRequestsH\x10R\tresources\x88\x01\x01\x1aY\n\x13PlannerOptionsEntry\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\x11\x45nvOverridesEntry\x12\x10\n\x03key\x18\x01 \x01(\tR\x03key\x12\x14\n\x05value\x18\x02 \x01(\tR\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\t\n\x07_branchB\x0f\n\r_dataset_nameB\x12\n\x10_spine_sql_queryB\x0e\n\x0c_max_samplesB\x15\n\x13_max_cache_age_secsB\x11\n\x0f_correlation_idB\x1a\n\x18_observed_at_lower_boundB\x1a\n\x18_observed_at_upper_boundB\x0f\n\r_store_onlineB\x10\n\x0e_store_offlineB\x19\n\x17_use_multiple_computersB\r\n\x0b_num_shardsB\x0e\n\x0c_num_workersB\x10\n\x0e_overlay_graphB\r\n\x0b_query_nameB\x15\n\x13_query_name_versionB\x0c\n\n_resources"\xce\x01\n\x12\x43olumnMetadataList\x12N\n\x08metadata\x18\x01 \x03(\x0b\x32\x32.chalk.common.v1.ColumnMetadataList.ColumnMetadataR\x08metadata\x1ah\n\x0e\x43olumnMetadata\x12\x1f\n\x0b\x66\x65\x61ture_fqn\x18\x01 \x01(\tR\nfeatureFqn\x12\x1f\n\x0b\x63olumn_name\x18\x02 \x01(\tR\ncolumnName\x12\x14\n\x05\x64type\x18\x03 \x01(\tR\x05\x64type"\xf0\x01\n\x1aGetOfflineQueryJobResponse\x12\x1f\n\x0bis_finished\x18\x01 \x01(\x08R\nisFinished\x12\x18\n\x07version\x18\x02 \x01(\x05R\x07version\x12\x12\n\x04urls\x18\x03 \x03(\tR\x04urls\x12\x33\n\x06\x65rrors\x18\x04 \x03(\x0b\x32\x1b.chalk.common.v1.ChalkErrorR\x06\x65rrors\x12\x42\n\x07\x63olumns\x18\x05 \x01(\x0b\x32#.chalk.common.v1.ColumnMetadataListH\x00R\x07\x63olumns\x88\x01\x01\x42\n\n\x08_columnsB\x86\x01\n\x13\x63om.chalk.common.v1B\x11OfflineQueryProtoP\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()
@@ -58,18 +58,20 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_OFFLINEQUERYINPUTS"]._serialized_end = 1304
     _globals["_OFFLINEQUERYINPUTS_NOINPUTS"]._serialized_start = 1286
     _globals["_OFFLINEQUERYINPUTS_NOINPUTS"]._serialized_end = 1296
-    _globals["_OFFLINEQUERYREQUEST"]._serialized_start = 1307
-    _globals["_OFFLINEQUERYREQUEST"]._serialized_end = 3505
-    _globals["_OFFLINEQUERYREQUEST_PLANNEROPTIONSENTRY"]._serialized_start = 2951
-    _globals["_OFFLINEQUERYREQUEST_PLANNEROPTIONSENTRY"]._serialized_end = 3040
-    _globals["_OFFLINEQUERYREQUEST_ENVOVERRIDESENTRY"]._serialized_start = 3042
-    _globals["_OFFLINEQUERYREQUEST_ENVOVERRIDESENTRY"]._serialized_end = 3105
-    _globals["_OFFLINEQUERYREQUEST_QUERYCONTEXTENTRY"]._serialized_start = 3107
-    _globals["_OFFLINEQUERYREQUEST_QUERYCONTEXTENTRY"]._serialized_end = 3194
-    _globals["_COLUMNMETADATALIST"]._serialized_start = 3508
-    _globals["_COLUMNMETADATALIST"]._serialized_end = 3714
-    _globals["_COLUMNMETADATALIST_COLUMNMETADATA"]._serialized_start = 3610
-    _globals["_COLUMNMETADATALIST_COLUMNMETADATA"]._serialized_end = 3714
-    _globals["_GETOFFLINEQUERYJOBRESPONSE"]._serialized_start = 3717
-    _globals["_GETOFFLINEQUERYJOBRESPONSE"]._serialized_end = 3957
+    _globals["_RESOURCEREQUESTS"]._serialized_start = 1307
+    _globals["_RESOURCEREQUESTS"]._serialized_end = 1614
+    _globals["_OFFLINEQUERYREQUEST"]._serialized_start = 1617
+    _globals["_OFFLINEQUERYREQUEST"]._serialized_end = 3900
+    _globals["_OFFLINEQUERYREQUEST_PLANNEROPTIONSENTRY"]._serialized_start = 3332
+    _globals["_OFFLINEQUERYREQUEST_PLANNEROPTIONSENTRY"]._serialized_end = 3421
+    _globals["_OFFLINEQUERYREQUEST_ENVOVERRIDESENTRY"]._serialized_start = 3423
+    _globals["_OFFLINEQUERYREQUEST_ENVOVERRIDESENTRY"]._serialized_end = 3486
+    _globals["_OFFLINEQUERYREQUEST_QUERYCONTEXTENTRY"]._serialized_start = 3488
+    _globals["_OFFLINEQUERYREQUEST_QUERYCONTEXTENTRY"]._serialized_end = 3575
+    _globals["_COLUMNMETADATALIST"]._serialized_start = 3903
+    _globals["_COLUMNMETADATALIST"]._serialized_end = 4109
+    _globals["_COLUMNMETADATALIST_COLUMNMETADATA"]._serialized_start = 4005
+    _globals["_COLUMNMETADATALIST_COLUMNMETADATA"]._serialized_end = 4109
+    _globals["_GETOFFLINEQUERYJOBRESPONSE"]._serialized_start = 4112
+    _globals["_GETOFFLINEQUERYJOBRESPONSE"]._serialized_end = 4352
 # @@protoc_insertion_point(module_scope)

chalk/_gen/chalk/common/v1/offline_query_pb2.pyi CHANGED Viewed

@@ -99,6 +99,27 @@ class OfflineQueryInputs(_message.Message):
         sharded_parquet_upload_inputs: _Optional[_Union[OfflineQueryShardedParquetUploadInput, _Mapping]] = ...,
     ) -> None: ...
+class ResourceRequests(_message.Message):
+    __slots__ = ("cpu", "memory", "ephemeral_volume_size", "ephemeral_storage", "resource_group")
+    CPU_FIELD_NUMBER: _ClassVar[int]
+    MEMORY_FIELD_NUMBER: _ClassVar[int]
+    EPHEMERAL_VOLUME_SIZE_FIELD_NUMBER: _ClassVar[int]
+    EPHEMERAL_STORAGE_FIELD_NUMBER: _ClassVar[int]
+    RESOURCE_GROUP_FIELD_NUMBER: _ClassVar[int]
+    cpu: str
+    memory: str
+    ephemeral_volume_size: str
+    ephemeral_storage: str
+    resource_group: str
+    def __init__(
+        self,
+        cpu: _Optional[str] = ...,
+        memory: _Optional[str] = ...,
+        ephemeral_volume_size: _Optional[str] = ...,
+        ephemeral_storage: _Optional[str] = ...,
+        resource_group: _Optional[str] = ...,
+    ) -> None: ...
 class OfflineQueryRequest(_message.Message):
     __slots__ = (
         "inputs",
@@ -131,6 +152,7 @@ class OfflineQueryRequest(_message.Message):
         "overlay_graph",
         "query_name",
         "query_name_version",
+        "resources",
     )
     class PlannerOptionsEntry(_message.Message):
         __slots__ = ("key", "value")
@@ -190,6 +212,7 @@ class OfflineQueryRequest(_message.Message):
     OVERLAY_GRAPH_FIELD_NUMBER: _ClassVar[int]
     QUERY_NAME_FIELD_NUMBER: _ClassVar[int]
     QUERY_NAME_VERSION_FIELD_NUMBER: _ClassVar[int]
+    RESOURCES_FIELD_NUMBER: _ClassVar[int]
     inputs: OfflineQueryInputs
     outputs: _containers.RepeatedScalarFieldContainer[str]
     required_outputs: _containers.RepeatedScalarFieldContainer[str]
@@ -220,6 +243,7 @@ class OfflineQueryRequest(_message.Message):
     overlay_graph: _graph_pb2.OverlayGraph
     query_name: str
     query_name_version: str
+    resources: ResourceRequests
     def __init__(
         self,
         inputs: _Optional[_Union[OfflineQueryInputs, _Mapping]] = ...,
@@ -252,6 +276,7 @@ class OfflineQueryRequest(_message.Message):
         overlay_graph: _Optional[_Union[_graph_pb2.OverlayGraph, _Mapping]] = ...,
         query_name: _Optional[str] = ...,
         query_name_version: _Optional[str] = ...,
+        resources: _Optional[_Union[ResourceRequests, _Mapping]] = ...,
     ) -> None: ...
 class ColumnMetadataList(_message.Message):

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

@@ -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":\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*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()
@@ -32,8 +32,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
     _globals["_SCRIPTTASKREQUEST_ENVOVERRIDESENTRY"]._serialized_options = b"8\001"
     _globals["_SCRIPTTASKSTATUS"]._serialized_start = 1304
     _globals["_SCRIPTTASKSTATUS"]._serialized_end = 1519
-    _globals["_SCRIPTTASKKIND"]._serialized_start = 1521
-    _globals["_SCRIPTTASKKIND"]._serialized_end = 1642
+    _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

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

@@ -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,7 @@ 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",)

chalk/_gen/chalk/dataframe/__init__.py ADDED Viewed

File without changes

chalk/_gen/chalk/dataframe/v1/__init__.py ADDED Viewed

File without changes

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

@@ -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)

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

chalkpy 2.90.1py3-none-any.whl → 2.95.3py3-none-any.whl