chalkpy 2.96.5__py3-none-any.whl → 2.96.6__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (11) hide show
  1. chalk/_gen/chalk/artifacts/v1/cron_query_pb2.py +4 -4
  2. chalk/_gen/chalk/artifacts/v1/cron_query_pb2.pyi +8 -0
  3. chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py +9 -3
  4. chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi +119 -0
  5. chalk/_version.py +1 -1
  6. chalk/parsed/_proto/export.py +2 -0
  7. {chalkpy-2.96.5.dist-info → chalkpy-2.96.6.dist-info}/METADATA +1 -1
  8. {chalkpy-2.96.5.dist-info → chalkpy-2.96.6.dist-info}/RECORD +11 -11
  9. {chalkpy-2.96.5.dist-info → chalkpy-2.96.6.dist-info}/WHEEL +0 -0
  10. {chalkpy-2.96.5.dist-info → chalkpy-2.96.6.dist-info}/entry_points.txt +0 -0
  11. {chalkpy-2.96.5.dist-info → chalkpy-2.96.6.dist-info}/top_level.txt +0 -0
chalk/_gen/chalk/artifacts/v1/cron_query_pb2.py CHANGED
@@ -18,7 +18,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
18
18
 
19
19
 
20
20
  DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
21
- 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'
21
+ 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"\xd0\x07\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\x12"\n\nnum_shards\x18\x11 \x01(\x05H\x03R\tnumShards\x88\x01\x01\x12$\n\x0bnum_workers\x18\x12 \x01(\x05H\x04R\nnumWorkers\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\r\n\x0b_num_shardsB\x0e\n\x0c_num_workersB\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'
22
22
  )
23
23
 
24
24
  _globals = globals()
@@ -34,7 +34,7 @@ if _descriptor._USE_C_DESCRIPTORS == False:
34
34
  _globals["_RECOMPUTESETTINGS"]._serialized_start = 124
35
35
  _globals["_RECOMPUTESETTINGS"]._serialized_end = 213
36
36
  _globals["_CRONQUERY"]._serialized_start = 216
37
- _globals["_CRONQUERY"]._serialized_end = 1087
38
- _globals["_CRONQUERY_PLANNEROPTIONSENTRY"]._serialized_start = 963
39
- _globals["_CRONQUERY_PLANNEROPTIONSENTRY"]._serialized_end = 1028
37
+ _globals["_CRONQUERY"]._serialized_end = 1192
38
+ _globals["_CRONQUERY_PLANNEROPTIONSENTRY"]._serialized_start = 1037
39
+ _globals["_CRONQUERY_PLANNEROPTIONSENTRY"]._serialized_end = 1102
40
40
  # @@protoc_insertion_point(module_scope)
chalk/_gen/chalk/artifacts/v1/cron_query_pb2.pyi CHANGED
@@ -39,6 +39,8 @@ class CronQuery(_message.Message):
39
39
  "resource_group",
40
40
  "planner_options",
41
41
  "completion_deadline",
42
+ "num_shards",
43
+ "num_workers",
42
44
  )
43
45
  class PlannerOptionsEntry(_message.Message):
44
46
  __slots__ = ("key", "value")
@@ -64,6 +66,8 @@ class CronQuery(_message.Message):
64
66
  RESOURCE_GROUP_FIELD_NUMBER: _ClassVar[int]
65
67
  PLANNER_OPTIONS_FIELD_NUMBER: _ClassVar[int]
66
68
  COMPLETION_DEADLINE_FIELD_NUMBER: _ClassVar[int]
69
+ NUM_SHARDS_FIELD_NUMBER: _ClassVar[int]
70
+ NUM_WORKERS_FIELD_NUMBER: _ClassVar[int]
67
71
  name: str
68
72
  cron: str
69
73
  file_name: str
@@ -80,6 +84,8 @@ class CronQuery(_message.Message):
80
84
  resource_group: str
81
85
  planner_options: _containers.ScalarMap[str, str]
82
86
  completion_deadline: _duration_pb2.Duration
87
+ num_shards: int
88
+ num_workers: int
83
89
  def __init__(
84
90
  self,
85
91
  name: _Optional[str] = ...,
@@ -98,4 +104,6 @@ class CronQuery(_message.Message):
98
104
  resource_group: _Optional[str] = ...,
99
105
  planner_options: _Optional[_Mapping[str, str]] = ...,
100
106
  completion_deadline: _Optional[_Union[_duration_pb2.Duration, _Mapping]] = ...,
107
+ num_shards: _Optional[int] = ...,
108
+ num_workers: _Optional[int] = ...,
101
109
  ) -> None: ...
chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py CHANGED
@@ -19,7 +19,7 @@ from google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__
19
19
 
20
20
 
21
21
  DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(
22
- b'\n)chalk/server/v1/scheduled_query_run.proto\x12\x0f\x63halk.server.v1\x1a%chalk/server/v1/offline_queries.proto\x1a google/protobuf/field_mask.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xa4\x06\n\x11ScheduledQueryRun\x12\x0e\n\x02id\x18\x01 \x01(\x03R\x02id\x12%\n\x0e\x65nvironment_id\x18\x02 \x01(\tR\renvironmentId\x12(\n\rdeployment_id\x18\x03 \x01(\tH\x00R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x1a\n\x06run_id\x18\x04 \x01(\tH\x01R\x05runId\x88\x01\x01\x12"\n\rcron_query_id\x18\x05 \x01(\x03R\x0b\x63ronQueryId\x12\x33\n\x16\x63ron_query_schedule_id\x18\x06 \x01(\x03R\x13\x63ronQueryScheduleId\x12\x1b\n\tcron_name\x18\x07 \x01(\tR\x08\x63ronName\x12-\n\x10gcr_execution_id\x18\x08 \x01(\tH\x02R\x0egcrExecutionId\x88\x01\x01\x12%\n\x0cgcr_job_name\x18\t \x01(\tH\x03R\ngcrJobName\x88\x01\x01\x12(\n\x10offline_query_id\x18\n \x01(\tR\x0eofflineQueryId\x12\x39\n\ncreated_at\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12@\n\x06status\x18\r \x01(\x0e\x32(.chalk.server.v1.ScheduledQueryRunStatusR\x06status\x12\x35\n\x14\x62locker_operation_id\x18\x0e \x01(\tH\x04R\x12\x62lockerOperationId\x88\x01\x01\x12\x37\n\x15workflow_execution_id\x18\x0f \x01(\tH\x05R\x13workflowExecutionId\x88\x01\x01\x42\x10\n\x0e_deployment_idB\t\n\x07_run_idB\x13\n\x11_gcr_execution_idB\x0f\n\r_gcr_job_nameB\x17\n\x15_blocker_operation_idB\x18\n\x16_workflow_execution_id"\xdd\x01\n\x1bGetScheduledQueryRunRequest\x12\x17\n\x06run_id\x18\x01 \x01(\x03H\x00R\x05runId\x12*\n\x10offline_query_id\x18\x02 \x01(\tH\x00R\x0eofflineQueryId\x12\x34\n\x15workflow_execution_id\x18\x04 \x01(\tH\x00R\x13workflowExecutionId\x12\x35\n\x08get_mask\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskR\x07getMaskB\x0c\n\nidentifier"\xd1\x01\n\x1cGetScheduledQueryRunResponse\x12R\n\x13scheduled_query_run\x18\x01 \x01(\x0b\x32".chalk.server.v1.ScheduledQueryRunR\x11scheduledQueryRun\x12K\n\roffline_query\x18\x02 \x01(\x0b\x32!.chalk.server.v1.OfflineQueryMetaH\x00R\x0cofflineQuery\x88\x01\x01\x42\x10\n\x0e_offline_query"\x9d\x01\n\x1cGetScheduledQueryRunsRequest\x12"\n\rcron_query_id\x18\x01 \x01(\x05R\x0b\x63ronQueryId\x12\x1d\n\tcron_name\x18\x04 \x01(\tH\x00R\x08\x63ronName\x12\x16\n\x06\x63ursor\x18\x02 \x01(\tR\x06\x63ursor\x12\x14\n\x05limit\x18\x03 \x01(\x05R\x05limitB\x0c\n\nidentifier"o\n\x1dGetScheduledQueryRunsResponse\x12\x36\n\x04runs\x18\x01 \x03(\x0b\x32".chalk.server.v1.ScheduledQueryRunR\x04runs\x12\x16\n\x06\x63ursor\x18\x02 \x01(\tR\x06\x63ursor*\x8f\x03\n\x17ScheduledQueryRunStatus\x12*\n&SCHEDULED_QUERY_RUN_STATUS_UNSPECIFIED\x10\x00\x12+\n\'SCHEDULED_QUERY_RUN_STATUS_INITIALIZING\x10\x01\x12*\n&SCHEDULED_QUERY_RUN_STATUS_INIT_FAILED\x10\x02\x12&\n"SCHEDULED_QUERY_RUN_STATUS_SKIPPED\x10\x03\x12%\n!SCHEDULED_QUERY_RUN_STATUS_QUEUED\x10\x04\x12&\n"SCHEDULED_QUERY_RUN_STATUS_WORKING\x10\x05\x12(\n$SCHEDULED_QUERY_RUN_STATUS_COMPLETED\x10\x06\x12%\n!SCHEDULED_QUERY_RUN_STATUS_FAILED\x10\x07\x12\'\n#SCHEDULED_QUERY_RUN_STATUS_CANCELED\x10\x08\x42\x9f\x01\n\x13\x63om.chalk.server.v1B\x16ScheduledQueryRunProtoP\x01Z\x12server/v1;serverv1\xa2\x02\x03\x43SX\xaa\x02\x0f\x43halk.Server.V1\xca\x02\x0f\x43halk\\Server\\V1\xe2\x02\x1b\x43halk\\Server\\V1\\GPBMetadata\xea\x02\x11\x43halk::Server::V1b\x06proto3'
22
+ b'\n)chalk/server/v1/scheduled_query_run.proto\x12\x0f\x63halk.server.v1\x1a%chalk/server/v1/offline_queries.proto\x1a google/protobuf/field_mask.proto\x1a\x1fgoogle/protobuf/timestamp.proto"\xa4\x06\n\x11ScheduledQueryRun\x12\x0e\n\x02id\x18\x01 \x01(\x03R\x02id\x12%\n\x0e\x65nvironment_id\x18\x02 \x01(\tR\renvironmentId\x12(\n\rdeployment_id\x18\x03 \x01(\tH\x00R\x0c\x64\x65ploymentId\x88\x01\x01\x12\x1a\n\x06run_id\x18\x04 \x01(\tH\x01R\x05runId\x88\x01\x01\x12"\n\rcron_query_id\x18\x05 \x01(\x03R\x0b\x63ronQueryId\x12\x33\n\x16\x63ron_query_schedule_id\x18\x06 \x01(\x03R\x13\x63ronQueryScheduleId\x12\x1b\n\tcron_name\x18\x07 \x01(\tR\x08\x63ronName\x12-\n\x10gcr_execution_id\x18\x08 \x01(\tH\x02R\x0egcrExecutionId\x88\x01\x01\x12%\n\x0cgcr_job_name\x18\t \x01(\tH\x03R\ngcrJobName\x88\x01\x01\x12(\n\x10offline_query_id\x18\n \x01(\tR\x0eofflineQueryId\x12\x39\n\ncreated_at\x18\x0b \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12@\n\x06status\x18\r \x01(\x0e\x32(.chalk.server.v1.ScheduledQueryRunStatusR\x06status\x12\x35\n\x14\x62locker_operation_id\x18\x0e \x01(\tH\x04R\x12\x62lockerOperationId\x88\x01\x01\x12\x37\n\x15workflow_execution_id\x18\x0f \x01(\tH\x05R\x13workflowExecutionId\x88\x01\x01\x42\x10\n\x0e_deployment_idB\t\n\x07_run_idB\x13\n\x11_gcr_execution_idB\x0f\n\r_gcr_job_nameB\x17\n\x15_blocker_operation_idB\x18\n\x16_workflow_execution_id"\xdd\x01\n\x1bGetScheduledQueryRunRequest\x12\x17\n\x06run_id\x18\x01 \x01(\x03H\x00R\x05runId\x12*\n\x10offline_query_id\x18\x02 \x01(\tH\x00R\x0eofflineQueryId\x12\x34\n\x15workflow_execution_id\x18\x04 \x01(\tH\x00R\x13workflowExecutionId\x12\x35\n\x08get_mask\x18\x03 \x01(\x0b\x32\x1a.google.protobuf.FieldMaskR\x07getMaskB\x0c\n\nidentifier"\xd1\x01\n\x1cGetScheduledQueryRunResponse\x12R\n\x13scheduled_query_run\x18\x01 \x01(\x0b\x32".chalk.server.v1.ScheduledQueryRunR\x11scheduledQueryRun\x12K\n\roffline_query\x18\x02 \x01(\x0b\x32!.chalk.server.v1.OfflineQueryMetaH\x00R\x0cofflineQuery\x88\x01\x01\x42\x10\n\x0e_offline_query"\x9d\x01\n\x1cGetScheduledQueryRunsRequest\x12"\n\rcron_query_id\x18\x01 \x01(\x05R\x0b\x63ronQueryId\x12\x1d\n\tcron_name\x18\x04 \x01(\tH\x00R\x08\x63ronName\x12\x16\n\x06\x63ursor\x18\x02 \x01(\tR\x06\x63ursor\x12\x14\n\x05limit\x18\x03 \x01(\x05R\x05limitB\x0c\n\nidentifier"o\n\x1dGetScheduledQueryRunsResponse\x12\x36\n\x04runs\x18\x01 \x03(\x0b\x32".chalk.server.v1.ScheduledQueryRunR\x04runs\x12\x16\n\x06\x63ursor\x18\x02 \x01(\tR\x06\x63ursor"\xf5\x02\n\x15ScheduledQueryControl\x12"\n\rcron_query_id\x18\x01 \x01(\x05R\x0b\x63ronQueryId\x12&\n\x0f\x63ron_query_name\x18\x02 \x01(\tR\rcronQueryName\x12:\n\x06status\x18\x03 \x01(\x0e\x32".chalk.server.v1.CronControlStatusR\x06status\x12\x1e\n\x08\x61gent_id\x18\x04 \x01(\tH\x00R\x07\x61gentId\x88\x01\x01\x12\x39\n\ncreated_at\x18\x05 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x06 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12"\n\njob_config\x18\x07 \x01(\tH\x01R\tjobConfig\x88\x01\x01\x42\x0b\n\t_agent_idB\r\n\x0b_job_config"\xd4\x07\n\x16ScheduledQuerySchedule\x12\x0e\n\x02id\x18\x01 \x01(\x05R\x02id\x12"\n\rcron_query_id\x18\x02 \x01(\x05R\x0b\x63ronQueryId\x12#\n\rdeployment_id\x18\x03 \x01(\tR\x0c\x64\x65ploymentId\x12\x12\n\x04name\x18\x04 \x01(\tR\x04name\x12\x12\n\x04\x63ron\x18\x05 \x01(\tR\x04\x63ron\x12\x1a\n\x08\x66ilename\x18\x06 \x01(\tR\x08\x66ilename\x12\x39\n\ncreated_at\x18\x07 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tcreatedAt\x12\x39\n\nupdated_at\x18\x08 \x01(\x0b\x32\x1a.google.protobuf.TimestampR\tupdatedAt\x12\x1b\n\x06output\x18\t \x01(\tH\x00R\x06output\x88\x01\x01\x12$\n\x0bmax_samples\x18\n \x01(\x05H\x01R\nmaxSamples\x88\x01\x01\x12\x32\n\x12recompute_features\x18\x0b \x01(\tH\x02R\x11recomputeFeatures\x88\x01\x01\x12@\n\x0blower_bound\x18\x0c \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x03R\nlowerBound\x88\x01\x01\x12@\n\x0bupper_bound\x18\r \x01(\x0b\x32\x1a.google.protobuf.TimestampH\x04R\nupperBound\x88\x01\x01\x12\x17\n\x04tags\x18\x0e \x01(\tH\x05R\x04tags\x88\x01\x01\x12\x39\n\x16required_resolver_tags\x18\x0f \x01(\tH\x06R\x14requiredResolverTags\x88\x01\x01\x12,\n\x0fplanner_options\x18\x10 \x01(\tH\x07R\x0eplannerOptions\x88\x01\x01\x12&\n\x0c\x64\x61taset_name\x18\x11 \x01(\tH\x08R\x0b\x64\x61tasetName\x88\x01\x01\x12"\n\nnum_shards\x18\x12 \x01(\x05H\tR\tnumShards\x88\x01\x01\x12$\n\x0bnum_workers\x18\x13 \x01(\x05H\nR\nnumWorkers\x88\x01\x01\x42\t\n\x07_outputB\x0e\n\x0c_max_samplesB\x15\n\x13_recompute_featuresB\x0e\n\x0c_lower_boundB\x0e\n\x0c_upper_boundB\x07\n\x05_tagsB\x19\n\x17_required_resolver_tagsB\x12\n\x10_planner_optionsB\x0f\n\r_dataset_nameB\r\n\x0b_num_shardsB\x0e\n\x0c_num_workers*\x8f\x03\n\x17ScheduledQueryRunStatus\x12*\n&SCHEDULED_QUERY_RUN_STATUS_UNSPECIFIED\x10\x00\x12+\n\'SCHEDULED_QUERY_RUN_STATUS_INITIALIZING\x10\x01\x12*\n&SCHEDULED_QUERY_RUN_STATUS_INIT_FAILED\x10\x02\x12&\n"SCHEDULED_QUERY_RUN_STATUS_SKIPPED\x10\x03\x12%\n!SCHEDULED_QUERY_RUN_STATUS_QUEUED\x10\x04\x12&\n"SCHEDULED_QUERY_RUN_STATUS_WORKING\x10\x05\x12(\n$SCHEDULED_QUERY_RUN_STATUS_COMPLETED\x10\x06\x12%\n!SCHEDULED_QUERY_RUN_STATUS_FAILED\x10\x07\x12\'\n#SCHEDULED_QUERY_RUN_STATUS_CANCELED\x10\x08*u\n\x11\x43ronControlStatus\x12#\n\x1f\x43RON_CONTROL_STATUS_UNSPECIFIED\x10\x00\x12\x1c\n\x18\x43RON_CONTROL_STATUS_PLAY\x10\x01\x12\x1d\n\x19\x43RON_CONTROL_STATUS_PAUSE\x10\x02\x42\x9f\x01\n\x13\x63om.chalk.server.v1B\x16ScheduledQueryRunProtoP\x01Z\x12server/v1;serverv1\xa2\x02\x03\x43SX\xaa\x02\x0f\x43halk.Server.V1\xca\x02\x0f\x43halk\\Server\\V1\xe2\x02\x1b\x43halk\\Server\\V1\\GPBMetadata\xea\x02\x11\x43halk::Server::V1b\x06proto3'
23
23
  )
24
24
 
25
25
  _globals = globals()
@@ -30,8 +30,10 @@ if _descriptor._USE_C_DESCRIPTORS == False:
30
30
  _globals[
31
31
  "DESCRIPTOR"
32
32
  ]._serialized_options = b"\n\023com.chalk.server.v1B\026ScheduledQueryRunProtoP\001Z\022server/v1;serverv1\242\002\003CSX\252\002\017Chalk.Server.V1\312\002\017Chalk\\Server\\V1\342\002\033Chalk\\Server\\V1\\GPBMetadata\352\002\021Chalk::Server::V1"
33
- _globals["_SCHEDULEDQUERYRUNSTATUS"]._serialized_start = 1685
34
- _globals["_SCHEDULEDQUERYRUNSTATUS"]._serialized_end = 2084
33
+ _globals["_SCHEDULEDQUERYRUNSTATUS"]._serialized_start = 3044
34
+ _globals["_SCHEDULEDQUERYRUNSTATUS"]._serialized_end = 3443
35
+ _globals["_CRONCONTROLSTATUS"]._serialized_start = 3445
36
+ _globals["_CRONCONTROLSTATUS"]._serialized_end = 3562
35
37
  _globals["_SCHEDULEDQUERYRUN"]._serialized_start = 169
36
38
  _globals["_SCHEDULEDQUERYRUN"]._serialized_end = 973
37
39
  _globals["_GETSCHEDULEDQUERYRUNREQUEST"]._serialized_start = 976
@@ -42,4 +44,8 @@ if _descriptor._USE_C_DESCRIPTORS == False:
42
44
  _globals["_GETSCHEDULEDQUERYRUNSREQUEST"]._serialized_end = 1569
43
45
  _globals["_GETSCHEDULEDQUERYRUNSRESPONSE"]._serialized_start = 1571
44
46
  _globals["_GETSCHEDULEDQUERYRUNSRESPONSE"]._serialized_end = 1682
47
+ _globals["_SCHEDULEDQUERYCONTROL"]._serialized_start = 1685
48
+ _globals["_SCHEDULEDQUERYCONTROL"]._serialized_end = 2058
49
+ _globals["_SCHEDULEDQUERYSCHEDULE"]._serialized_start = 2061
50
+ _globals["_SCHEDULEDQUERYSCHEDULE"]._serialized_end = 3041
45
51
  # @@protoc_insertion_point(module_scope)
chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi CHANGED
@@ -27,6 +27,12 @@ class ScheduledQueryRunStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper)
27
27
  SCHEDULED_QUERY_RUN_STATUS_FAILED: _ClassVar[ScheduledQueryRunStatus]
28
28
  SCHEDULED_QUERY_RUN_STATUS_CANCELED: _ClassVar[ScheduledQueryRunStatus]
29
29
 
30
+ class CronControlStatus(int, metaclass=_enum_type_wrapper.EnumTypeWrapper):
31
+ __slots__ = ()
32
+ CRON_CONTROL_STATUS_UNSPECIFIED: _ClassVar[CronControlStatus]
33
+ CRON_CONTROL_STATUS_PLAY: _ClassVar[CronControlStatus]
34
+ CRON_CONTROL_STATUS_PAUSE: _ClassVar[CronControlStatus]
35
+
30
36
  SCHEDULED_QUERY_RUN_STATUS_UNSPECIFIED: ScheduledQueryRunStatus
31
37
  SCHEDULED_QUERY_RUN_STATUS_INITIALIZING: ScheduledQueryRunStatus
32
38
  SCHEDULED_QUERY_RUN_STATUS_INIT_FAILED: ScheduledQueryRunStatus
@@ -36,6 +42,9 @@ SCHEDULED_QUERY_RUN_STATUS_WORKING: ScheduledQueryRunStatus
36
42
  SCHEDULED_QUERY_RUN_STATUS_COMPLETED: ScheduledQueryRunStatus
37
43
  SCHEDULED_QUERY_RUN_STATUS_FAILED: ScheduledQueryRunStatus
38
44
  SCHEDULED_QUERY_RUN_STATUS_CANCELED: ScheduledQueryRunStatus
45
+ CRON_CONTROL_STATUS_UNSPECIFIED: CronControlStatus
46
+ CRON_CONTROL_STATUS_PLAY: CronControlStatus
47
+ CRON_CONTROL_STATUS_PAUSE: CronControlStatus
39
48
 
40
49
  class ScheduledQueryRun(_message.Message):
41
50
  __slots__ = (
@@ -161,3 +170,113 @@ class GetScheduledQueryRunsResponse(_message.Message):
161
170
  def __init__(
162
171
  self, runs: _Optional[_Iterable[_Union[ScheduledQueryRun, _Mapping]]] = ..., cursor: _Optional[str] = ...
163
172
  ) -> None: ...
173
+
174
+ class ScheduledQueryControl(_message.Message):
175
+ __slots__ = ("cron_query_id", "cron_query_name", "status", "agent_id", "created_at", "updated_at", "job_config")
176
+ CRON_QUERY_ID_FIELD_NUMBER: _ClassVar[int]
177
+ CRON_QUERY_NAME_FIELD_NUMBER: _ClassVar[int]
178
+ STATUS_FIELD_NUMBER: _ClassVar[int]
179
+ AGENT_ID_FIELD_NUMBER: _ClassVar[int]
180
+ CREATED_AT_FIELD_NUMBER: _ClassVar[int]
181
+ UPDATED_AT_FIELD_NUMBER: _ClassVar[int]
182
+ JOB_CONFIG_FIELD_NUMBER: _ClassVar[int]
183
+ cron_query_id: int
184
+ cron_query_name: str
185
+ status: CronControlStatus
186
+ agent_id: str
187
+ created_at: _timestamp_pb2.Timestamp
188
+ updated_at: _timestamp_pb2.Timestamp
189
+ job_config: str
190
+ def __init__(
191
+ self,
192
+ cron_query_id: _Optional[int] = ...,
193
+ cron_query_name: _Optional[str] = ...,
194
+ status: _Optional[_Union[CronControlStatus, str]] = ...,
195
+ agent_id: _Optional[str] = ...,
196
+ created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
197
+ updated_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
198
+ job_config: _Optional[str] = ...,
199
+ ) -> None: ...
200
+
201
+ class ScheduledQuerySchedule(_message.Message):
202
+ __slots__ = (
203
+ "id",
204
+ "cron_query_id",
205
+ "deployment_id",
206
+ "name",
207
+ "cron",
208
+ "filename",
209
+ "created_at",
210
+ "updated_at",
211
+ "output",
212
+ "max_samples",
213
+ "recompute_features",
214
+ "lower_bound",
215
+ "upper_bound",
216
+ "tags",
217
+ "required_resolver_tags",
218
+ "planner_options",
219
+ "dataset_name",
220
+ "num_shards",
221
+ "num_workers",
222
+ )
223
+ ID_FIELD_NUMBER: _ClassVar[int]
224
+ CRON_QUERY_ID_FIELD_NUMBER: _ClassVar[int]
225
+ DEPLOYMENT_ID_FIELD_NUMBER: _ClassVar[int]
226
+ NAME_FIELD_NUMBER: _ClassVar[int]
227
+ CRON_FIELD_NUMBER: _ClassVar[int]
228
+ FILENAME_FIELD_NUMBER: _ClassVar[int]
229
+ CREATED_AT_FIELD_NUMBER: _ClassVar[int]
230
+ UPDATED_AT_FIELD_NUMBER: _ClassVar[int]
231
+ OUTPUT_FIELD_NUMBER: _ClassVar[int]
232
+ MAX_SAMPLES_FIELD_NUMBER: _ClassVar[int]
233
+ RECOMPUTE_FEATURES_FIELD_NUMBER: _ClassVar[int]
234
+ LOWER_BOUND_FIELD_NUMBER: _ClassVar[int]
235
+ UPPER_BOUND_FIELD_NUMBER: _ClassVar[int]
236
+ TAGS_FIELD_NUMBER: _ClassVar[int]
237
+ REQUIRED_RESOLVER_TAGS_FIELD_NUMBER: _ClassVar[int]
238
+ PLANNER_OPTIONS_FIELD_NUMBER: _ClassVar[int]
239
+ DATASET_NAME_FIELD_NUMBER: _ClassVar[int]
240
+ NUM_SHARDS_FIELD_NUMBER: _ClassVar[int]
241
+ NUM_WORKERS_FIELD_NUMBER: _ClassVar[int]
242
+ id: int
243
+ cron_query_id: int
244
+ deployment_id: str
245
+ name: str
246
+ cron: str
247
+ filename: str
248
+ created_at: _timestamp_pb2.Timestamp
249
+ updated_at: _timestamp_pb2.Timestamp
250
+ output: str
251
+ max_samples: int
252
+ recompute_features: str
253
+ lower_bound: _timestamp_pb2.Timestamp
254
+ upper_bound: _timestamp_pb2.Timestamp
255
+ tags: str
256
+ required_resolver_tags: str
257
+ planner_options: str
258
+ dataset_name: str
259
+ num_shards: int
260
+ num_workers: int
261
+ def __init__(
262
+ self,
263
+ id: _Optional[int] = ...,
264
+ cron_query_id: _Optional[int] = ...,
265
+ deployment_id: _Optional[str] = ...,
266
+ name: _Optional[str] = ...,
267
+ cron: _Optional[str] = ...,
268
+ filename: _Optional[str] = ...,
269
+ created_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
270
+ updated_at: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
271
+ output: _Optional[str] = ...,
272
+ max_samples: _Optional[int] = ...,
273
+ recompute_features: _Optional[str] = ...,
274
+ lower_bound: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
275
+ upper_bound: _Optional[_Union[_timestamp_pb2.Timestamp, _Mapping]] = ...,
276
+ tags: _Optional[str] = ...,
277
+ required_resolver_tags: _Optional[str] = ...,
278
+ planner_options: _Optional[str] = ...,
279
+ dataset_name: _Optional[str] = ...,
280
+ num_shards: _Optional[int] = ...,
281
+ num_workers: _Optional[int] = ...,
282
+ ) -> None: ...
chalk/_version.py CHANGED
@@ -1 +1 @@
1
- __version__ = "2.96.5"
1
+ __version__ = "2.96.6"
chalk/parsed/_proto/export.py CHANGED
@@ -215,6 +215,8 @@ def export_from_registry() -> export_pb.Export:
215
215
  completion_deadline=timedelta_to_proto_duration(cron.completion_deadline)
216
216
  if cron.completion_deadline is not None
217
217
  else cron.completion_deadline,
218
+ num_shards=cron.num_shards,
219
+ num_workers=cron.num_workers,
218
220
  )
219
221
  )
220
222
 
{chalkpy-2.96.5.dist-info → chalkpy-2.96.6.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: chalkpy
3
- Version: 2.96.5
3
+ Version: 2.96.6
4
4
  Summary: Python SDK for Chalk
5
5
  Author: Chalk AI, Inc.
6
6
  Project-URL: Homepage, https://chalk.ai
{chalkpy-2.96.5.dist-info → chalkpy-2.96.6.dist-info}/RECORD RENAMED
@@ -1,5 +1,5 @@
1
1
  chalk/__init__.py,sha256=vKsx9-cl5kImlVWGHVRYO6bweBm79NAzGs3l36u71wM,2657
2
- chalk/_version.py,sha256=8ZvKPpEPaNnATdFe98glfnwM5s6vGpw6tcR9AeaiP58,23
2
+ chalk/_version.py,sha256=Y-Kl04ucuWNOt_WOW6dvhcamZanMwDrSw7NrZ8zKeSs,23
3
3
  chalk/cli.py,sha256=ckqqfOI-A2mT23-rnZzDMmblYj-2x1VBX8ebHlIEn9A,5873
4
4
  chalk/importer.py,sha256=m4lMn1lSYj_euDq8CS7LYTBnek9JOcjGJf9-82dJHbA,64441
5
5
  chalk/prompts.py,sha256=2H9UomLAamdfRTNUdKs9i3VTpiossuyRhntqsAXUhhg,16117
@@ -41,8 +41,8 @@ chalk/_gen/chalk/artifacts/v1/chart_pb2.py,sha256=XOXGa60UkIn6g2I0KWZihDF0PSiHp3
41
41
  chalk/_gen/chalk/artifacts/v1/chart_pb2.pyi,sha256=BDs1o5LFC1gqMdO67wimsCI4sVfX-42aKwDQqDb7DpQ,21653
42
42
  chalk/_gen/chalk/artifacts/v1/chart_pb2_grpc.py,sha256=VCyAf0skoHSgQPkD4n8rKQPYesinqHqN8TEwyu7XGUo,159
43
43
  chalk/_gen/chalk/artifacts/v1/chart_pb2_grpc.pyi,sha256=ff2TSiLVnG6IVQcTGzb2DIH3XRSoAvAo_RMcvbMFyc0,76
44
- chalk/_gen/chalk/artifacts/v1/cron_query_pb2.py,sha256=VyJU2lT1_DqrWJG9Ce6mIiTQHJM2oDANlsOaiRMEtK8,3776
45
- chalk/_gen/chalk/artifacts/v1/cron_query_pb2.pyi,sha256=SR2_0nsD61s9EutuTo6UrtrHa--FRi-qPle53C5Ul8A,3867
44
+ chalk/_gen/chalk/artifacts/v1/cron_query_pb2.py,sha256=HGcz2kkIV7V7e7DKV_fTxjSl9d0Qm36ZtdwdSp4ze_k,3956
45
+ chalk/_gen/chalk/artifacts/v1/cron_query_pb2.pyi,sha256=xlL_r_iy4WUHMYIY3uay2oiFWNY7pJq1anK15KRX04Y,4127
46
46
  chalk/_gen/chalk/artifacts/v1/cron_query_pb2_grpc.py,sha256=VCyAf0skoHSgQPkD4n8rKQPYesinqHqN8TEwyu7XGUo,159
47
47
  chalk/_gen/chalk/artifacts/v1/cron_query_pb2_grpc.pyi,sha256=ff2TSiLVnG6IVQcTGzb2DIH3XRSoAvAo_RMcvbMFyc0,76
48
48
  chalk/_gen/chalk/artifacts/v1/deployment_pb2.py,sha256=7ifnsrurEyVGcUMX41tD3XGNveBGrCfqG21GEdhrRZE,2877
@@ -504,8 +504,8 @@ chalk/_gen/chalk/server/v1/scheduled_query_pb2.py,sha256=aU7YjTJ2H9dp_ZRNmh_91NC
504
504
  chalk/_gen/chalk/server/v1/scheduled_query_pb2.pyi,sha256=4o__cQ3plq1LucDRH03pMf9LBSjk1eE7XY51gyv435c,299
505
505
  chalk/_gen/chalk/server/v1/scheduled_query_pb2_grpc.py,sha256=Pb0WD51FjkvUmPcYINjkSgYX7Z2HCmakFb5wp_Ulqxw,5037
506
506
  chalk/_gen/chalk/server/v1/scheduled_query_pb2_grpc.pyi,sha256=w98vRRLSiFl6Z_8c06DDAVH2593_dAgbhdlI-vY0tJw,1337
507
- chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py,sha256=WJ-nWBvaVoAUy57jZ1kG6_Q4xvQgwpW83xoIE1K_2CI,5619
508
- chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi,sha256=ePknNWz-VY73vA60QqzyX6ZgimWyJQ6rO_f_DUlNziY,6590
507
+ chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.py,sha256=oFIhmStu504lzlE9Lrc7zGzK_T21pq_EDl54rG0_87Y,8412
508
+ chalk/_gen/chalk/server/v1/scheduled_query_run_pb2.pyi,sha256=b1KQf_j5joN92W2_Q2ha04wCyj5jIdqrRWEKGJpjFJA,10952
509
509
  chalk/_gen/chalk/server/v1/scheduled_query_run_pb2_grpc.py,sha256=VCyAf0skoHSgQPkD4n8rKQPYesinqHqN8TEwyu7XGUo,159
510
510
  chalk/_gen/chalk/server/v1/scheduled_query_run_pb2_grpc.pyi,sha256=ff2TSiLVnG6IVQcTGzb2DIH3XRSoAvAo_RMcvbMFyc0,76
511
511
  chalk/_gen/chalk/server/v1/scheduler_pb2.py,sha256=kfmYnzdx22eNeemjrfuANsiyI1oARmfSRw3MrURaFAI,11969
@@ -731,7 +731,7 @@ chalk/parsed/to_proto.py,sha256=ZSwEYkuwr3UhMVabt6FM02fBN_oBy_mSMjDYS_crJw0,7553
731
731
  chalk/parsed/user_types_to_json.py,sha256=ZJWdYFqyhr5InvItQybtHadXQjW3vjHrv8hjMGtL3Bc,13318
732
732
  chalk/parsed/validation_from_registries.py,sha256=nfiAj1tvWRu0RrkhkGtElhAsL8V7ayEKUisrKQF-wYc,7900
733
733
  chalk/parsed/_proto/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
734
- chalk/parsed/_proto/export.py,sha256=hsHKIEHl6-Bd-lMPU1zMzu8MEA1lal8FNWBFX4bRFPw,11676
734
+ chalk/parsed/_proto/export.py,sha256=mRPDyXNOqHLE9GcpmADjQgzHTCeYpZW5x8qp41UIlMQ,11766
735
735
  chalk/parsed/_proto/lsp.py,sha256=hAv20bo8edVDhn2EgJRo9PmzoIi2JlR6bBcvkUPF4YM,3711
736
736
  chalk/parsed/_proto/utils.py,sha256=X6vZp3OAFMxWS5BaWVeKuNhF5ob4Ws9yAz4RiNalCx4,10680
737
737
  chalk/parsed/_proto/validation.py,sha256=D6dOnCoeARHOLPHQb8zYGXDc0eWJblSQ4tO4rb_HKNU,35367
@@ -827,8 +827,8 @@ chalk/utils/tracing.py,sha256=NiiM-9dbuJhSCv6R1npR1uYNKWlkqTR6Ygm0Voi2NrY,13078
827
827
  chalk/utils/weak_set_by_identity.py,sha256=VmikA_laYwFeOphCwXJIuyOIkrdlQe0bSzaXq7onoQw,953
828
828
  chalk/utils/pydanticutil/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
829
829
  chalk/utils/pydanticutil/pydantic_compat.py,sha256=O575lLYJ5GvZC4HMzR9yATxf9XwjC6NrDUXbNwZidlE,3031
830
- chalkpy-2.96.5.dist-info/METADATA,sha256=dYguQl-353EIvqauFcbZX5hG1RrMj45QqCLsi2la9Fw,27754
831
- chalkpy-2.96.5.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
832
- chalkpy-2.96.5.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
833
- chalkpy-2.96.5.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
834
- chalkpy-2.96.5.dist-info/RECORD,,
830
+ chalkpy-2.96.6.dist-info/METADATA,sha256=5dqjMynKBZ6H6Y6tbYNazvRRwyCLZblyJFsi6E_PA1E,27754
831
+ chalkpy-2.96.6.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
832
+ chalkpy-2.96.6.dist-info/entry_points.txt,sha256=Vg23sd8icwq-morJrljVFr-kQnMbm95rZfZj5wsZGis,42
833
+ chalkpy-2.96.6.dist-info/top_level.txt,sha256=1Q6_19IGYfNxSw50W8tYKEJ2t5HKQ3W9Wiw4ia5yg2c,6
834
+ chalkpy-2.96.6.dist-info/RECORD,,