flyte 0.0.1b0__py3-none-any.whl → 0.0.1b1__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.

Potentially problematic release.


This version of flyte might be problematic. Click here for more details.

Files changed (209) hide show
  1. flyte/_cli/_common.py +12 -0
  2. {union → flyte}/_cli/_params.py +106 -147
  3. flyte/_cli/_run.py +24 -2
  4. flyte/_cli/main.py +28 -2
  5. flyte/_image.py +11 -10
  6. flyte/_initialize.py +24 -15
  7. flyte/_internal/runtime/convert.py +6 -0
  8. flyte/_run.py +0 -1
  9. flyte/_version.py +2 -2
  10. flyte/config/__init__.py +168 -0
  11. flyte/config/_config.py +196 -0
  12. flyte/config/_internal.py +64 -0
  13. flyte/remote/_console.py +1 -1
  14. flyte/types/_type_engine.py +4 -3
  15. {flyte-0.0.1b0.dist-info → flyte-0.0.1b1.dist-info}/METADATA +1 -1
  16. flyte-0.0.1b1.dist-info/RECORD +204 -0
  17. flyte-0.0.1b0.dist-info/RECORD +0 -390
  18. union/__init__.py +0 -54
  19. union/_api_commons.py +0 -3
  20. union/_bin/__init__.py +0 -0
  21. union/_bin/runtime.py +0 -113
  22. union/_build.py +0 -25
  23. union/_cache/__init__.py +0 -12
  24. union/_cache/cache.py +0 -141
  25. union/_cache/defaults.py +0 -9
  26. union/_cache/policy_function_body.py +0 -42
  27. union/_cli/__init__.py +0 -0
  28. union/_cli/_common.py +0 -263
  29. union/_cli/_create.py +0 -40
  30. union/_cli/_delete.py +0 -23
  31. union/_cli/_deploy.py +0 -120
  32. union/_cli/_get.py +0 -162
  33. union/_cli/_run.py +0 -150
  34. union/_cli/main.py +0 -72
  35. union/_code_bundle/__init__.py +0 -8
  36. union/_code_bundle/_ignore.py +0 -113
  37. union/_code_bundle/_packaging.py +0 -187
  38. union/_code_bundle/_utils.py +0 -342
  39. union/_code_bundle/bundle.py +0 -176
  40. union/_context.py +0 -146
  41. union/_datastructures.py +0 -295
  42. union/_deploy.py +0 -185
  43. union/_doc.py +0 -29
  44. union/_docstring.py +0 -26
  45. union/_environment.py +0 -43
  46. union/_group.py +0 -31
  47. union/_hash.py +0 -23
  48. union/_image.py +0 -760
  49. union/_initialize.py +0 -585
  50. union/_interface.py +0 -84
  51. union/_internal/__init__.py +0 -3
  52. union/_internal/controllers/__init__.py +0 -77
  53. union/_internal/controllers/_local_controller.py +0 -77
  54. union/_internal/controllers/pbhash.py +0 -39
  55. union/_internal/controllers/remote/__init__.py +0 -40
  56. union/_internal/controllers/remote/_action.py +0 -131
  57. union/_internal/controllers/remote/_client.py +0 -43
  58. union/_internal/controllers/remote/_controller.py +0 -169
  59. union/_internal/controllers/remote/_core.py +0 -341
  60. union/_internal/controllers/remote/_informer.py +0 -260
  61. union/_internal/controllers/remote/_service_protocol.py +0 -44
  62. union/_internal/imagebuild/__init__.py +0 -11
  63. union/_internal/imagebuild/docker_builder.py +0 -416
  64. union/_internal/imagebuild/image_builder.py +0 -243
  65. union/_internal/imagebuild/remote_builder.py +0 -0
  66. union/_internal/resolvers/__init__.py +0 -0
  67. union/_internal/resolvers/_task_module.py +0 -31
  68. union/_internal/resolvers/common.py +0 -24
  69. union/_internal/resolvers/default.py +0 -27
  70. union/_internal/runtime/__init__.py +0 -0
  71. union/_internal/runtime/convert.py +0 -163
  72. union/_internal/runtime/entrypoints.py +0 -121
  73. union/_internal/runtime/io.py +0 -136
  74. union/_internal/runtime/resources_serde.py +0 -134
  75. union/_internal/runtime/task_serde.py +0 -202
  76. union/_internal/runtime/taskrunner.py +0 -179
  77. union/_internal/runtime/types_serde.py +0 -53
  78. union/_logging.py +0 -124
  79. union/_protos/__init__.py +0 -0
  80. union/_protos/common/authorization_pb2.py +0 -66
  81. union/_protos/common/authorization_pb2.pyi +0 -106
  82. union/_protos/common/authorization_pb2_grpc.py +0 -4
  83. union/_protos/common/identifier_pb2.py +0 -71
  84. union/_protos/common/identifier_pb2.pyi +0 -82
  85. union/_protos/common/identifier_pb2_grpc.py +0 -4
  86. union/_protos/common/identity_pb2.py +0 -48
  87. union/_protos/common/identity_pb2.pyi +0 -72
  88. union/_protos/common/identity_pb2_grpc.py +0 -4
  89. union/_protos/common/list_pb2.py +0 -36
  90. union/_protos/common/list_pb2.pyi +0 -69
  91. union/_protos/common/list_pb2_grpc.py +0 -4
  92. union/_protos/common/policy_pb2.py +0 -37
  93. union/_protos/common/policy_pb2.pyi +0 -27
  94. union/_protos/common/policy_pb2_grpc.py +0 -4
  95. union/_protos/common/role_pb2.py +0 -37
  96. union/_protos/common/role_pb2.pyi +0 -51
  97. union/_protos/common/role_pb2_grpc.py +0 -4
  98. union/_protos/common/runtime_version_pb2.py +0 -28
  99. union/_protos/common/runtime_version_pb2.pyi +0 -24
  100. union/_protos/common/runtime_version_pb2_grpc.py +0 -4
  101. union/_protos/logs/dataplane/payload_pb2.py +0 -96
  102. union/_protos/logs/dataplane/payload_pb2.pyi +0 -168
  103. union/_protos/logs/dataplane/payload_pb2_grpc.py +0 -4
  104. union/_protos/secret/definition_pb2.py +0 -49
  105. union/_protos/secret/definition_pb2.pyi +0 -93
  106. union/_protos/secret/definition_pb2_grpc.py +0 -4
  107. union/_protos/secret/payload_pb2.py +0 -62
  108. union/_protos/secret/payload_pb2.pyi +0 -94
  109. union/_protos/secret/payload_pb2_grpc.py +0 -4
  110. union/_protos/secret/secret_pb2.py +0 -38
  111. union/_protos/secret/secret_pb2.pyi +0 -6
  112. union/_protos/secret/secret_pb2_grpc.py +0 -198
  113. union/_protos/validate/validate/validate_pb2.py +0 -76
  114. union/_protos/workflow/node_execution_service_pb2.py +0 -26
  115. union/_protos/workflow/node_execution_service_pb2.pyi +0 -4
  116. union/_protos/workflow/node_execution_service_pb2_grpc.py +0 -32
  117. union/_protos/workflow/queue_service_pb2.py +0 -75
  118. union/_protos/workflow/queue_service_pb2.pyi +0 -103
  119. union/_protos/workflow/queue_service_pb2_grpc.py +0 -172
  120. union/_protos/workflow/run_definition_pb2.py +0 -100
  121. union/_protos/workflow/run_definition_pb2.pyi +0 -256
  122. union/_protos/workflow/run_definition_pb2_grpc.py +0 -4
  123. union/_protos/workflow/run_logs_service_pb2.py +0 -41
  124. union/_protos/workflow/run_logs_service_pb2.pyi +0 -28
  125. union/_protos/workflow/run_logs_service_pb2_grpc.py +0 -69
  126. union/_protos/workflow/run_service_pb2.py +0 -133
  127. union/_protos/workflow/run_service_pb2.pyi +0 -173
  128. union/_protos/workflow/run_service_pb2_grpc.py +0 -412
  129. union/_protos/workflow/state_service_pb2.py +0 -58
  130. union/_protos/workflow/state_service_pb2.pyi +0 -69
  131. union/_protos/workflow/state_service_pb2_grpc.py +0 -138
  132. union/_protos/workflow/task_definition_pb2.py +0 -72
  133. union/_protos/workflow/task_definition_pb2.pyi +0 -65
  134. union/_protos/workflow/task_definition_pb2_grpc.py +0 -4
  135. union/_protos/workflow/task_service_pb2.py +0 -44
  136. union/_protos/workflow/task_service_pb2.pyi +0 -31
  137. union/_protos/workflow/task_service_pb2_grpc.py +0 -104
  138. union/_resources.py +0 -226
  139. union/_retry.py +0 -32
  140. union/_reusable_environment.py +0 -25
  141. union/_run.py +0 -374
  142. union/_secret.py +0 -61
  143. union/_task.py +0 -354
  144. union/_task_environment.py +0 -186
  145. union/_timeout.py +0 -47
  146. union/_tools.py +0 -27
  147. union/_utils/__init__.py +0 -11
  148. union/_utils/asyn.py +0 -119
  149. union/_utils/file_handling.py +0 -71
  150. union/_utils/helpers.py +0 -46
  151. union/_utils/lazy_module.py +0 -54
  152. union/_utils/uv_script_parser.py +0 -49
  153. union/_version.py +0 -21
  154. union/connectors/__init__.py +0 -0
  155. union/errors.py +0 -128
  156. union/extras/__init__.py +0 -5
  157. union/extras/_container.py +0 -263
  158. union/io/__init__.py +0 -11
  159. union/io/_dataframe.py +0 -0
  160. union/io/_dir.py +0 -425
  161. union/io/_file.py +0 -418
  162. union/io/pickle/__init__.py +0 -0
  163. union/io/pickle/transformer.py +0 -117
  164. union/io/structured_dataset/__init__.py +0 -122
  165. union/io/structured_dataset/basic_dfs.py +0 -219
  166. union/io/structured_dataset/structured_dataset.py +0 -1057
  167. union/py.typed +0 -0
  168. union/remote/__init__.py +0 -23
  169. union/remote/_client/__init__.py +0 -0
  170. union/remote/_client/_protocols.py +0 -129
  171. union/remote/_client/auth/__init__.py +0 -12
  172. union/remote/_client/auth/_authenticators/__init__.py +0 -0
  173. union/remote/_client/auth/_authenticators/base.py +0 -391
  174. union/remote/_client/auth/_authenticators/client_credentials.py +0 -73
  175. union/remote/_client/auth/_authenticators/device_code.py +0 -120
  176. union/remote/_client/auth/_authenticators/external_command.py +0 -77
  177. union/remote/_client/auth/_authenticators/factory.py +0 -200
  178. union/remote/_client/auth/_authenticators/pkce.py +0 -515
  179. union/remote/_client/auth/_channel.py +0 -184
  180. union/remote/_client/auth/_client_config.py +0 -83
  181. union/remote/_client/auth/_default_html.py +0 -32
  182. union/remote/_client/auth/_grpc_utils/__init__.py +0 -0
  183. union/remote/_client/auth/_grpc_utils/auth_interceptor.py +0 -204
  184. union/remote/_client/auth/_grpc_utils/default_metadata_interceptor.py +0 -144
  185. union/remote/_client/auth/_keyring.py +0 -154
  186. union/remote/_client/auth/_token_client.py +0 -258
  187. union/remote/_client/auth/errors.py +0 -16
  188. union/remote/_client/controlplane.py +0 -86
  189. union/remote/_data.py +0 -149
  190. union/remote/_logs.py +0 -74
  191. union/remote/_project.py +0 -86
  192. union/remote/_run.py +0 -820
  193. union/remote/_secret.py +0 -132
  194. union/remote/_task.py +0 -193
  195. union/report/__init__.py +0 -3
  196. union/report/_report.py +0 -178
  197. union/report/_template.html +0 -124
  198. union/storage/__init__.py +0 -24
  199. union/storage/_remote_fs.py +0 -34
  200. union/storage/_storage.py +0 -247
  201. union/storage/_utils.py +0 -5
  202. union/types/__init__.py +0 -11
  203. union/types/_renderer.py +0 -162
  204. union/types/_string_literals.py +0 -120
  205. union/types/_type_engine.py +0 -2131
  206. union/types/_utils.py +0 -80
  207. {flyte-0.0.1b0.dist-info → flyte-0.0.1b1.dist-info}/WHEEL +0 -0
  208. {flyte-0.0.1b0.dist-info → flyte-0.0.1b1.dist-info}/entry_points.txt +0 -0
  209. {flyte-0.0.1b0.dist-info → flyte-0.0.1b1.dist-info}/top_level.txt +0 -0
@@ -1,138 +0,0 @@
1
- # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
- """Client and server classes corresponding to protobuf-defined services."""
3
- import grpc
4
-
5
- from union._protos.workflow import state_service_pb2 as workflow_dot_state__service__pb2
6
-
7
-
8
- class StateServiceStub(object):
9
- """provides an interface for managing the state of actions.
10
- """
11
-
12
- def __init__(self, channel):
13
- """Constructor.
14
-
15
- Args:
16
- channel: A grpc.Channel.
17
- """
18
- self.Store = channel.unary_unary(
19
- '/cloudidl.workflow.StateService/Store',
20
- request_serializer=workflow_dot_state__service__pb2.StoreRequest.SerializeToString,
21
- response_deserializer=workflow_dot_state__service__pb2.StoreResponse.FromString,
22
- )
23
- self.Load = channel.unary_unary(
24
- '/cloudidl.workflow.StateService/Load',
25
- request_serializer=workflow_dot_state__service__pb2.LoadRequest.SerializeToString,
26
- response_deserializer=workflow_dot_state__service__pb2.LoadResponse.FromString,
27
- )
28
- self.Watch = channel.unary_stream(
29
- '/cloudidl.workflow.StateService/Watch',
30
- request_serializer=workflow_dot_state__service__pb2.WatchRequest.SerializeToString,
31
- response_deserializer=workflow_dot_state__service__pb2.WatchResponse.FromString,
32
- )
33
-
34
-
35
- class StateServiceServicer(object):
36
- """provides an interface for managing the state of actions.
37
- """
38
-
39
- def Store(self, request, context):
40
- """store the state of an action.
41
- """
42
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
43
- context.set_details('Method not implemented!')
44
- raise NotImplementedError('Method not implemented!')
45
-
46
- def Load(self, request, context):
47
- """load the state of an action.
48
- """
49
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
50
- context.set_details('Method not implemented!')
51
- raise NotImplementedError('Method not implemented!')
52
-
53
- def Watch(self, request, context):
54
- """watch for updates to the state of actions. this api guarantees at-least-once delivery semantics.
55
- """
56
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
57
- context.set_details('Method not implemented!')
58
- raise NotImplementedError('Method not implemented!')
59
-
60
-
61
- def add_StateServiceServicer_to_server(servicer, server):
62
- rpc_method_handlers = {
63
- 'Store': grpc.unary_unary_rpc_method_handler(
64
- servicer.Store,
65
- request_deserializer=workflow_dot_state__service__pb2.StoreRequest.FromString,
66
- response_serializer=workflow_dot_state__service__pb2.StoreResponse.SerializeToString,
67
- ),
68
- 'Load': grpc.unary_unary_rpc_method_handler(
69
- servicer.Load,
70
- request_deserializer=workflow_dot_state__service__pb2.LoadRequest.FromString,
71
- response_serializer=workflow_dot_state__service__pb2.LoadResponse.SerializeToString,
72
- ),
73
- 'Watch': grpc.unary_stream_rpc_method_handler(
74
- servicer.Watch,
75
- request_deserializer=workflow_dot_state__service__pb2.WatchRequest.FromString,
76
- response_serializer=workflow_dot_state__service__pb2.WatchResponse.SerializeToString,
77
- ),
78
- }
79
- generic_handler = grpc.method_handlers_generic_handler(
80
- 'cloudidl.workflow.StateService', rpc_method_handlers)
81
- server.add_generic_rpc_handlers((generic_handler,))
82
-
83
-
84
- # This class is part of an EXPERIMENTAL API.
85
- class StateService(object):
86
- """provides an interface for managing the state of actions.
87
- """
88
-
89
- @staticmethod
90
- def Store(request,
91
- target,
92
- options=(),
93
- channel_credentials=None,
94
- call_credentials=None,
95
- insecure=False,
96
- compression=None,
97
- wait_for_ready=None,
98
- timeout=None,
99
- metadata=None):
100
- return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.StateService/Store',
101
- workflow_dot_state__service__pb2.StoreRequest.SerializeToString,
102
- workflow_dot_state__service__pb2.StoreResponse.FromString,
103
- options, channel_credentials,
104
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
105
-
106
- @staticmethod
107
- def Load(request,
108
- target,
109
- options=(),
110
- channel_credentials=None,
111
- call_credentials=None,
112
- insecure=False,
113
- compression=None,
114
- wait_for_ready=None,
115
- timeout=None,
116
- metadata=None):
117
- return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.StateService/Load',
118
- workflow_dot_state__service__pb2.LoadRequest.SerializeToString,
119
- workflow_dot_state__service__pb2.LoadResponse.FromString,
120
- options, channel_credentials,
121
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
122
-
123
- @staticmethod
124
- def Watch(request,
125
- target,
126
- options=(),
127
- channel_credentials=None,
128
- call_credentials=None,
129
- insecure=False,
130
- compression=None,
131
- wait_for_ready=None,
132
- timeout=None,
133
- metadata=None):
134
- return grpc.experimental.unary_stream(request, target, '/cloudidl.workflow.StateService/Watch',
135
- workflow_dot_state__service__pb2.WatchRequest.SerializeToString,
136
- workflow_dot_state__service__pb2.WatchResponse.FromString,
137
- options, channel_credentials,
138
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
@@ -1,72 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
- # Generated by the protocol buffer compiler. DO NOT EDIT!
3
- # source: workflow/task_definition.proto
4
- """Generated protocol buffer code."""
5
- from google.protobuf import descriptor as _descriptor
6
- from google.protobuf import descriptor_pool as _descriptor_pool
7
- from google.protobuf import symbol_database as _symbol_database
8
- from google.protobuf.internal import builder as _builder
9
- # @@protoc_insertion_point(imports)
10
-
11
- _sym_db = _symbol_database.Default()
12
-
13
-
14
- from union._protos.common import identifier_pb2 as common_dot_identifier__pb2
15
- from union._protos.common import identity_pb2 as common_dot_identity__pb2
16
- from flyteidl.core import tasks_pb2 as flyteidl_dot_core_dot_tasks__pb2
17
- from union._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
18
-
19
-
20
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1eworkflow/task_definition.proto\x12\x11\x63loudidl.workflow\x1a\x17\x63ommon/identifier.proto\x1a\x15\x63ommon/identity.proto\x1a\x19\x66lyteidl/core/tasks.proto\x1a\x17validate/validate.proto\"\x88\x01\n\x08TaskName\x12\x19\n\x03org\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x03org\x12!\n\x07project\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x07project\x12\x1f\n\x06\x64omain\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x06\x64omain\x12\x1d\n\x04name\x18\x04 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18\x1eR\x04name\"\xb1\x01\n\x0eTaskIdentifier\x12\x19\n\x03org\x18\x01 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x03org\x12!\n\x07project\x18\x02 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x07project\x12\x1f\n\x06\x64omain\x18\x03 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x06\x64omain\x12\x1d\n\x04name\x18\x04 \x01(\tB\t\xfa\x42\x06r\x04\x10\x01\x18\x1eR\x04name\x12!\n\x07version\x18\x05 \x01(\tB\x07\xfa\x42\x04r\x02\x10\x01R\x07version\"\\\n\x0cTaskMetadata\x12L\n\x0b\x64\x65ployed_by\x18\x01 \x01(\x0b\x32!.cloudidl.common.EnrichedIdentityB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\ndeployedBy\"\x93\x01\n\x04Task\x12\x44\n\x07task_id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06taskId\x12\x45\n\x08metadata\x18\x02 \x01(\x0b\x32\x1f.cloudidl.workflow.TaskMetadataB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08metadata\"V\n\x08TaskSpec\x12J\n\rtask_template\x18\x01 \x01(\x0b\x32\x1b.flyteidl.core.TaskTemplateB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x0ctaskTemplate\"\xd5\x01\n\x0bTaskDetails\x12\x44\n\x07task_id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06taskId\x12\x45\n\x08metadata\x18\x02 \x01(\x0b\x32\x1f.cloudidl.workflow.TaskMetadataB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x08metadata\x12\x39\n\x04spec\x18\x03 \x01(\x0b\x32\x1b.cloudidl.workflow.TaskSpecB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x04specB\xc0\x01\n\x15\x63om.cloudidl.workflowB\x13TaskDefinitionProtoH\x02P\x01Z+github.com/unionai/cloud/gen/pb-go/workflow\xa2\x02\x03\x43WX\xaa\x02\x11\x43loudidl.Workflow\xca\x02\x11\x43loudidl\\Workflow\xe2\x02\x1d\x43loudidl\\Workflow\\GPBMetadata\xea\x02\x12\x43loudidl::Workflowb\x06proto3')
21
-
22
- _globals = globals()
23
- _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
24
- _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'workflow.task_definition_pb2', _globals)
25
- if _descriptor._USE_C_DESCRIPTORS == False:
26
- DESCRIPTOR._options = None
27
- DESCRIPTOR._serialized_options = b'\n\025com.cloudidl.workflowB\023TaskDefinitionProtoH\002P\001Z+github.com/unionai/cloud/gen/pb-go/workflow\242\002\003CWX\252\002\021Cloudidl.Workflow\312\002\021Cloudidl\\Workflow\342\002\035Cloudidl\\Workflow\\GPBMetadata\352\002\022Cloudidl::Workflow'
28
- _TASKNAME.fields_by_name['org']._options = None
29
- _TASKNAME.fields_by_name['org']._serialized_options = b'\372B\004r\002\020\001'
30
- _TASKNAME.fields_by_name['project']._options = None
31
- _TASKNAME.fields_by_name['project']._serialized_options = b'\372B\004r\002\020\001'
32
- _TASKNAME.fields_by_name['domain']._options = None
33
- _TASKNAME.fields_by_name['domain']._serialized_options = b'\372B\004r\002\020\001'
34
- _TASKNAME.fields_by_name['name']._options = None
35
- _TASKNAME.fields_by_name['name']._serialized_options = b'\372B\006r\004\020\001\030\036'
36
- _TASKIDENTIFIER.fields_by_name['org']._options = None
37
- _TASKIDENTIFIER.fields_by_name['org']._serialized_options = b'\372B\004r\002\020\001'
38
- _TASKIDENTIFIER.fields_by_name['project']._options = None
39
- _TASKIDENTIFIER.fields_by_name['project']._serialized_options = b'\372B\004r\002\020\001'
40
- _TASKIDENTIFIER.fields_by_name['domain']._options = None
41
- _TASKIDENTIFIER.fields_by_name['domain']._serialized_options = b'\372B\004r\002\020\001'
42
- _TASKIDENTIFIER.fields_by_name['name']._options = None
43
- _TASKIDENTIFIER.fields_by_name['name']._serialized_options = b'\372B\006r\004\020\001\030\036'
44
- _TASKIDENTIFIER.fields_by_name['version']._options = None
45
- _TASKIDENTIFIER.fields_by_name['version']._serialized_options = b'\372B\004r\002\020\001'
46
- _TASKMETADATA.fields_by_name['deployed_by']._options = None
47
- _TASKMETADATA.fields_by_name['deployed_by']._serialized_options = b'\372B\005\212\001\002\020\001'
48
- _TASK.fields_by_name['task_id']._options = None
49
- _TASK.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
50
- _TASK.fields_by_name['metadata']._options = None
51
- _TASK.fields_by_name['metadata']._serialized_options = b'\372B\005\212\001\002\020\001'
52
- _TASKSPEC.fields_by_name['task_template']._options = None
53
- _TASKSPEC.fields_by_name['task_template']._serialized_options = b'\372B\005\212\001\002\020\001'
54
- _TASKDETAILS.fields_by_name['task_id']._options = None
55
- _TASKDETAILS.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
56
- _TASKDETAILS.fields_by_name['metadata']._options = None
57
- _TASKDETAILS.fields_by_name['metadata']._serialized_options = b'\372B\005\212\001\002\020\001'
58
- _TASKDETAILS.fields_by_name['spec']._options = None
59
- _TASKDETAILS.fields_by_name['spec']._serialized_options = b'\372B\005\212\001\002\020\001'
60
- _globals['_TASKNAME']._serialized_start=154
61
- _globals['_TASKNAME']._serialized_end=290
62
- _globals['_TASKIDENTIFIER']._serialized_start=293
63
- _globals['_TASKIDENTIFIER']._serialized_end=470
64
- _globals['_TASKMETADATA']._serialized_start=472
65
- _globals['_TASKMETADATA']._serialized_end=564
66
- _globals['_TASK']._serialized_start=567
67
- _globals['_TASK']._serialized_end=714
68
- _globals['_TASKSPEC']._serialized_start=716
69
- _globals['_TASKSPEC']._serialized_end=802
70
- _globals['_TASKDETAILS']._serialized_start=805
71
- _globals['_TASKDETAILS']._serialized_end=1018
72
- # @@protoc_insertion_point(module_scope)
@@ -1,65 +0,0 @@
1
- from union._protos.common import identifier_pb2 as _identifier_pb2
2
- from union._protos.common import identity_pb2 as _identity_pb2
3
- from flyteidl.core import tasks_pb2 as _tasks_pb2
4
- from union._protos.validate.validate import validate_pb2 as _validate_pb2
5
- from google.protobuf import descriptor as _descriptor
6
- from google.protobuf import message as _message
7
- from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union
8
-
9
- DESCRIPTOR: _descriptor.FileDescriptor
10
-
11
- class TaskName(_message.Message):
12
- __slots__ = ["org", "project", "domain", "name"]
13
- ORG_FIELD_NUMBER: _ClassVar[int]
14
- PROJECT_FIELD_NUMBER: _ClassVar[int]
15
- DOMAIN_FIELD_NUMBER: _ClassVar[int]
16
- NAME_FIELD_NUMBER: _ClassVar[int]
17
- org: str
18
- project: str
19
- domain: str
20
- name: str
21
- def __init__(self, org: _Optional[str] = ..., project: _Optional[str] = ..., domain: _Optional[str] = ..., name: _Optional[str] = ...) -> None: ...
22
-
23
- class TaskIdentifier(_message.Message):
24
- __slots__ = ["org", "project", "domain", "name", "version"]
25
- ORG_FIELD_NUMBER: _ClassVar[int]
26
- PROJECT_FIELD_NUMBER: _ClassVar[int]
27
- DOMAIN_FIELD_NUMBER: _ClassVar[int]
28
- NAME_FIELD_NUMBER: _ClassVar[int]
29
- VERSION_FIELD_NUMBER: _ClassVar[int]
30
- org: str
31
- project: str
32
- domain: str
33
- name: str
34
- version: str
35
- def __init__(self, org: _Optional[str] = ..., project: _Optional[str] = ..., domain: _Optional[str] = ..., name: _Optional[str] = ..., version: _Optional[str] = ...) -> None: ...
36
-
37
- class TaskMetadata(_message.Message):
38
- __slots__ = ["deployed_by"]
39
- DEPLOYED_BY_FIELD_NUMBER: _ClassVar[int]
40
- deployed_by: _identity_pb2.EnrichedIdentity
41
- def __init__(self, deployed_by: _Optional[_Union[_identity_pb2.EnrichedIdentity, _Mapping]] = ...) -> None: ...
42
-
43
- class Task(_message.Message):
44
- __slots__ = ["task_id", "metadata"]
45
- TASK_ID_FIELD_NUMBER: _ClassVar[int]
46
- METADATA_FIELD_NUMBER: _ClassVar[int]
47
- task_id: TaskIdentifier
48
- metadata: TaskMetadata
49
- def __init__(self, task_id: _Optional[_Union[TaskIdentifier, _Mapping]] = ..., metadata: _Optional[_Union[TaskMetadata, _Mapping]] = ...) -> None: ...
50
-
51
- class TaskSpec(_message.Message):
52
- __slots__ = ["task_template"]
53
- TASK_TEMPLATE_FIELD_NUMBER: _ClassVar[int]
54
- task_template: _tasks_pb2.TaskTemplate
55
- def __init__(self, task_template: _Optional[_Union[_tasks_pb2.TaskTemplate, _Mapping]] = ...) -> None: ...
56
-
57
- class TaskDetails(_message.Message):
58
- __slots__ = ["task_id", "metadata", "spec"]
59
- TASK_ID_FIELD_NUMBER: _ClassVar[int]
60
- METADATA_FIELD_NUMBER: _ClassVar[int]
61
- SPEC_FIELD_NUMBER: _ClassVar[int]
62
- task_id: TaskIdentifier
63
- metadata: TaskMetadata
64
- spec: TaskSpec
65
- def __init__(self, task_id: _Optional[_Union[TaskIdentifier, _Mapping]] = ..., metadata: _Optional[_Union[TaskMetadata, _Mapping]] = ..., spec: _Optional[_Union[TaskSpec, _Mapping]] = ...) -> None: ...
@@ -1,4 +0,0 @@
1
- # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
- """Client and server classes corresponding to protobuf-defined services."""
3
- import grpc
4
-
@@ -1,44 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
- # Generated by the protocol buffer compiler. DO NOT EDIT!
3
- # source: workflow/task_service.proto
4
- """Generated protocol buffer code."""
5
- from google.protobuf import descriptor as _descriptor
6
- from google.protobuf import descriptor_pool as _descriptor_pool
7
- from google.protobuf import symbol_database as _symbol_database
8
- from google.protobuf.internal import builder as _builder
9
- # @@protoc_insertion_point(imports)
10
-
11
- _sym_db = _symbol_database.Default()
12
-
13
-
14
- from union._protos.validate.validate import validate_pb2 as validate_dot_validate__pb2
15
- from union._protos.workflow import task_definition_pb2 as workflow_dot_task__definition__pb2
16
-
17
-
18
- DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1bworkflow/task_service.proto\x12\x11\x63loudidl.workflow\x1a\x17validate/validate.proto\x1a\x1eworkflow/task_definition.proto\"\x94\x01\n\x11\x44\x65ployTaskRequest\x12\x44\n\x07task_id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06taskId\x12\x39\n\x04spec\x18\x02 \x01(\x0b\x32\x1b.cloudidl.workflow.TaskSpecB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x04spec\"\x14\n\x12\x44\x65ployTaskResponse\"]\n\x15GetTaskDetailsRequest\x12\x44\n\x07task_id\x18\x01 \x01(\x0b\x32!.cloudidl.workflow.TaskIdentifierB\x08\xfa\x42\x05\x8a\x01\x02\x10\x01R\x06taskId\"R\n\x16GetTaskDetailsResponse\x12\x38\n\x07\x64\x65tails\x18\x01 \x01(\x0b\x32\x1e.cloudidl.workflow.TaskDetailsR\x07\x64\x65tails2\xd6\x01\n\x0bTaskService\x12[\n\nDeployTask\x12$.cloudidl.workflow.DeployTaskRequest\x1a%.cloudidl.workflow.DeployTaskResponse\"\x00\x12j\n\x0eGetTaskDetails\x12(.cloudidl.workflow.GetTaskDetailsRequest\x1a).cloudidl.workflow.GetTaskDetailsResponse\"\x03\x90\x02\x01\x42\xbd\x01\n\x15\x63om.cloudidl.workflowB\x10TaskServiceProtoH\x02P\x01Z+github.com/unionai/cloud/gen/pb-go/workflow\xa2\x02\x03\x43WX\xaa\x02\x11\x43loudidl.Workflow\xca\x02\x11\x43loudidl\\Workflow\xe2\x02\x1d\x43loudidl\\Workflow\\GPBMetadata\xea\x02\x12\x43loudidl::Workflowb\x06proto3')
19
-
20
- _globals = globals()
21
- _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
22
- _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'workflow.task_service_pb2', _globals)
23
- if _descriptor._USE_C_DESCRIPTORS == False:
24
- DESCRIPTOR._options = None
25
- DESCRIPTOR._serialized_options = b'\n\025com.cloudidl.workflowB\020TaskServiceProtoH\002P\001Z+github.com/unionai/cloud/gen/pb-go/workflow\242\002\003CWX\252\002\021Cloudidl.Workflow\312\002\021Cloudidl\\Workflow\342\002\035Cloudidl\\Workflow\\GPBMetadata\352\002\022Cloudidl::Workflow'
26
- _DEPLOYTASKREQUEST.fields_by_name['task_id']._options = None
27
- _DEPLOYTASKREQUEST.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
28
- _DEPLOYTASKREQUEST.fields_by_name['spec']._options = None
29
- _DEPLOYTASKREQUEST.fields_by_name['spec']._serialized_options = b'\372B\005\212\001\002\020\001'
30
- _GETTASKDETAILSREQUEST.fields_by_name['task_id']._options = None
31
- _GETTASKDETAILSREQUEST.fields_by_name['task_id']._serialized_options = b'\372B\005\212\001\002\020\001'
32
- _TASKSERVICE.methods_by_name['GetTaskDetails']._options = None
33
- _TASKSERVICE.methods_by_name['GetTaskDetails']._serialized_options = b'\220\002\001'
34
- _globals['_DEPLOYTASKREQUEST']._serialized_start=108
35
- _globals['_DEPLOYTASKREQUEST']._serialized_end=256
36
- _globals['_DEPLOYTASKRESPONSE']._serialized_start=258
37
- _globals['_DEPLOYTASKRESPONSE']._serialized_end=278
38
- _globals['_GETTASKDETAILSREQUEST']._serialized_start=280
39
- _globals['_GETTASKDETAILSREQUEST']._serialized_end=373
40
- _globals['_GETTASKDETAILSRESPONSE']._serialized_start=375
41
- _globals['_GETTASKDETAILSRESPONSE']._serialized_end=457
42
- _globals['_TASKSERVICE']._serialized_start=460
43
- _globals['_TASKSERVICE']._serialized_end=674
44
- # @@protoc_insertion_point(module_scope)
@@ -1,31 +0,0 @@
1
- from union._protos.validate.validate import validate_pb2 as _validate_pb2
2
- from union._protos.workflow import task_definition_pb2 as _task_definition_pb2
3
- from google.protobuf import descriptor as _descriptor
4
- from google.protobuf import message as _message
5
- from typing import ClassVar as _ClassVar, Mapping as _Mapping, Optional as _Optional, Union as _Union
6
-
7
- DESCRIPTOR: _descriptor.FileDescriptor
8
-
9
- class DeployTaskRequest(_message.Message):
10
- __slots__ = ["task_id", "spec"]
11
- TASK_ID_FIELD_NUMBER: _ClassVar[int]
12
- SPEC_FIELD_NUMBER: _ClassVar[int]
13
- task_id: _task_definition_pb2.TaskIdentifier
14
- spec: _task_definition_pb2.TaskSpec
15
- def __init__(self, task_id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ..., spec: _Optional[_Union[_task_definition_pb2.TaskSpec, _Mapping]] = ...) -> None: ...
16
-
17
- class DeployTaskResponse(_message.Message):
18
- __slots__ = []
19
- def __init__(self) -> None: ...
20
-
21
- class GetTaskDetailsRequest(_message.Message):
22
- __slots__ = ["task_id"]
23
- TASK_ID_FIELD_NUMBER: _ClassVar[int]
24
- task_id: _task_definition_pb2.TaskIdentifier
25
- def __init__(self, task_id: _Optional[_Union[_task_definition_pb2.TaskIdentifier, _Mapping]] = ...) -> None: ...
26
-
27
- class GetTaskDetailsResponse(_message.Message):
28
- __slots__ = ["details"]
29
- DETAILS_FIELD_NUMBER: _ClassVar[int]
30
- details: _task_definition_pb2.TaskDetails
31
- def __init__(self, details: _Optional[_Union[_task_definition_pb2.TaskDetails, _Mapping]] = ...) -> None: ...
@@ -1,104 +0,0 @@
1
- # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
2
- """Client and server classes corresponding to protobuf-defined services."""
3
- import grpc
4
-
5
- from union._protos.workflow import task_service_pb2 as workflow_dot_task__service__pb2
6
-
7
-
8
- class TaskServiceStub(object):
9
- """TaskService provides an interface for managing tasks.
10
- """
11
-
12
- def __init__(self, channel):
13
- """Constructor.
14
-
15
- Args:
16
- channel: A grpc.Channel.
17
- """
18
- self.DeployTask = channel.unary_unary(
19
- '/cloudidl.workflow.TaskService/DeployTask',
20
- request_serializer=workflow_dot_task__service__pb2.DeployTaskRequest.SerializeToString,
21
- response_deserializer=workflow_dot_task__service__pb2.DeployTaskResponse.FromString,
22
- )
23
- self.GetTaskDetails = channel.unary_unary(
24
- '/cloudidl.workflow.TaskService/GetTaskDetails',
25
- request_serializer=workflow_dot_task__service__pb2.GetTaskDetailsRequest.SerializeToString,
26
- response_deserializer=workflow_dot_task__service__pb2.GetTaskDetailsResponse.FromString,
27
- )
28
-
29
-
30
- class TaskServiceServicer(object):
31
- """TaskService provides an interface for managing tasks.
32
- """
33
-
34
- def DeployTask(self, request, context):
35
- """Deploy a task.
36
- """
37
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
38
- context.set_details('Method not implemented!')
39
- raise NotImplementedError('Method not implemented!')
40
-
41
- def GetTaskDetails(self, request, context):
42
- """Get detailed information about a task.
43
- """
44
- context.set_code(grpc.StatusCode.UNIMPLEMENTED)
45
- context.set_details('Method not implemented!')
46
- raise NotImplementedError('Method not implemented!')
47
-
48
-
49
- def add_TaskServiceServicer_to_server(servicer, server):
50
- rpc_method_handlers = {
51
- 'DeployTask': grpc.unary_unary_rpc_method_handler(
52
- servicer.DeployTask,
53
- request_deserializer=workflow_dot_task__service__pb2.DeployTaskRequest.FromString,
54
- response_serializer=workflow_dot_task__service__pb2.DeployTaskResponse.SerializeToString,
55
- ),
56
- 'GetTaskDetails': grpc.unary_unary_rpc_method_handler(
57
- servicer.GetTaskDetails,
58
- request_deserializer=workflow_dot_task__service__pb2.GetTaskDetailsRequest.FromString,
59
- response_serializer=workflow_dot_task__service__pb2.GetTaskDetailsResponse.SerializeToString,
60
- ),
61
- }
62
- generic_handler = grpc.method_handlers_generic_handler(
63
- 'cloudidl.workflow.TaskService', rpc_method_handlers)
64
- server.add_generic_rpc_handlers((generic_handler,))
65
-
66
-
67
- # This class is part of an EXPERIMENTAL API.
68
- class TaskService(object):
69
- """TaskService provides an interface for managing tasks.
70
- """
71
-
72
- @staticmethod
73
- def DeployTask(request,
74
- target,
75
- options=(),
76
- channel_credentials=None,
77
- call_credentials=None,
78
- insecure=False,
79
- compression=None,
80
- wait_for_ready=None,
81
- timeout=None,
82
- metadata=None):
83
- return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.TaskService/DeployTask',
84
- workflow_dot_task__service__pb2.DeployTaskRequest.SerializeToString,
85
- workflow_dot_task__service__pb2.DeployTaskResponse.FromString,
86
- options, channel_credentials,
87
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
88
-
89
- @staticmethod
90
- def GetTaskDetails(request,
91
- target,
92
- options=(),
93
- channel_credentials=None,
94
- call_credentials=None,
95
- insecure=False,
96
- compression=None,
97
- wait_for_ready=None,
98
- timeout=None,
99
- metadata=None):
100
- return grpc.experimental.unary_unary(request, target, '/cloudidl.workflow.TaskService/GetTaskDetails',
101
- workflow_dot_task__service__pb2.GetTaskDetailsRequest.SerializeToString,
102
- workflow_dot_task__service__pb2.GetTaskDetailsResponse.FromString,
103
- options, channel_credentials,
104
- insecure, call_credentials, compression, wait_for_ready, timeout, metadata)