metaflow-stubs 2.18.1__py2.py3-none-any.whl → 2.18.3__py2.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 metaflow-stubs might be problematic. Click here for more details.

Files changed (166) hide show
  1. metaflow-stubs/__init__.pyi +462 -462
  2. metaflow-stubs/cards.pyi +2 -2
  3. metaflow-stubs/cli.pyi +2 -2
  4. metaflow-stubs/cli_components/__init__.pyi +2 -2
  5. metaflow-stubs/cli_components/utils.pyi +2 -2
  6. metaflow-stubs/client/__init__.pyi +2 -2
  7. metaflow-stubs/client/core.pyi +5 -5
  8. metaflow-stubs/client/filecache.pyi +3 -3
  9. metaflow-stubs/events.pyi +3 -3
  10. metaflow-stubs/exception.pyi +2 -2
  11. metaflow-stubs/flowspec.pyi +5 -5
  12. metaflow-stubs/generated_for.txt +1 -1
  13. metaflow-stubs/includefile.pyi +4 -4
  14. metaflow-stubs/meta_files.pyi +2 -2
  15. metaflow-stubs/metadata_provider/__init__.pyi +2 -2
  16. metaflow-stubs/metadata_provider/heartbeat.pyi +2 -2
  17. metaflow-stubs/metadata_provider/metadata.pyi +3 -3
  18. metaflow-stubs/metadata_provider/util.pyi +2 -2
  19. metaflow-stubs/metaflow_config.pyi +6 -2
  20. metaflow-stubs/metaflow_current.pyi +18 -18
  21. metaflow-stubs/metaflow_git.pyi +2 -2
  22. metaflow-stubs/multicore_utils.pyi +2 -2
  23. metaflow-stubs/packaging_sys/__init__.pyi +5 -5
  24. metaflow-stubs/packaging_sys/backend.pyi +3 -3
  25. metaflow-stubs/packaging_sys/distribution_support.pyi +4 -4
  26. metaflow-stubs/packaging_sys/tar_backend.pyi +6 -6
  27. metaflow-stubs/packaging_sys/utils.pyi +2 -2
  28. metaflow-stubs/packaging_sys/v1.pyi +4 -4
  29. metaflow-stubs/parameters.pyi +4 -4
  30. metaflow-stubs/plugins/__init__.pyi +14 -14
  31. metaflow-stubs/plugins/airflow/__init__.pyi +2 -2
  32. metaflow-stubs/plugins/airflow/airflow_decorator.pyi +2 -2
  33. metaflow-stubs/plugins/airflow/airflow_utils.pyi +2 -2
  34. metaflow-stubs/plugins/airflow/exception.pyi +2 -2
  35. metaflow-stubs/plugins/airflow/sensors/__init__.pyi +2 -2
  36. metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi +2 -2
  37. metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi +2 -2
  38. metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi +2 -2
  39. metaflow-stubs/plugins/argo/__init__.pyi +2 -2
  40. metaflow-stubs/plugins/argo/argo_client.pyi +6 -4
  41. metaflow-stubs/plugins/argo/argo_events.pyi +2 -2
  42. metaflow-stubs/plugins/argo/argo_workflows.pyi +10 -3
  43. metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +3 -3
  44. metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +5 -5
  45. metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi +3 -3
  46. metaflow-stubs/plugins/argo/exit_hooks.pyi +3 -3
  47. metaflow-stubs/plugins/aws/__init__.pyi +2 -2
  48. metaflow-stubs/plugins/aws/aws_client.pyi +2 -2
  49. metaflow-stubs/plugins/aws/aws_utils.pyi +2 -2
  50. metaflow-stubs/plugins/aws/batch/__init__.pyi +2 -2
  51. metaflow-stubs/plugins/aws/batch/batch.pyi +4 -4
  52. metaflow-stubs/plugins/aws/batch/batch_client.pyi +2 -2
  53. metaflow-stubs/plugins/aws/batch/batch_decorator.pyi +2 -2
  54. metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi +2 -2
  55. metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi +5 -5
  56. metaflow-stubs/plugins/aws/step_functions/__init__.pyi +2 -2
  57. metaflow-stubs/plugins/aws/step_functions/dynamo_db_client.pyi +2 -2
  58. metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi +2 -2
  59. metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi +2 -2
  60. metaflow-stubs/plugins/aws/step_functions/step_functions.pyi +3 -3
  61. metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi +2 -2
  62. metaflow-stubs/plugins/aws/step_functions/step_functions_decorator.pyi +2 -2
  63. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi +8 -5
  64. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi +3 -3
  65. metaflow-stubs/plugins/azure/__init__.pyi +2 -2
  66. metaflow-stubs/plugins/azure/azure_credential.pyi +2 -2
  67. metaflow-stubs/plugins/azure/azure_exceptions.pyi +2 -2
  68. metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi +5 -5
  69. metaflow-stubs/plugins/azure/azure_utils.pyi +2 -2
  70. metaflow-stubs/plugins/azure/blob_service_client_factory.pyi +2 -2
  71. metaflow-stubs/plugins/azure/includefile_support.pyi +2 -2
  72. metaflow-stubs/plugins/cards/__init__.pyi +2 -2
  73. metaflow-stubs/plugins/cards/card_client.pyi +3 -3
  74. metaflow-stubs/plugins/cards/card_creator.pyi +2 -2
  75. metaflow-stubs/plugins/cards/card_datastore.pyi +2 -2
  76. metaflow-stubs/plugins/cards/card_decorator.pyi +3 -3
  77. metaflow-stubs/plugins/cards/card_modules/__init__.pyi +2 -2
  78. metaflow-stubs/plugins/cards/card_modules/basic.pyi +3 -3
  79. metaflow-stubs/plugins/cards/card_modules/card.pyi +2 -2
  80. metaflow-stubs/plugins/cards/card_modules/components.pyi +5 -5
  81. metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi +2 -2
  82. metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi +2 -2
  83. metaflow-stubs/plugins/cards/card_modules/test_cards.pyi +2 -2
  84. metaflow-stubs/plugins/cards/card_resolver.pyi +2 -2
  85. metaflow-stubs/plugins/cards/component_serializer.pyi +2 -2
  86. metaflow-stubs/plugins/cards/exception.pyi +2 -2
  87. metaflow-stubs/plugins/catch_decorator.pyi +3 -3
  88. metaflow-stubs/plugins/datatools/__init__.pyi +2 -2
  89. metaflow-stubs/plugins/datatools/local.pyi +2 -2
  90. metaflow-stubs/plugins/datatools/s3/__init__.pyi +2 -2
  91. metaflow-stubs/plugins/datatools/s3/s3.pyi +5 -5
  92. metaflow-stubs/plugins/datatools/s3/s3tail.pyi +2 -2
  93. metaflow-stubs/plugins/datatools/s3/s3util.pyi +2 -2
  94. metaflow-stubs/plugins/debug_logger.pyi +2 -2
  95. metaflow-stubs/plugins/debug_monitor.pyi +2 -2
  96. metaflow-stubs/plugins/environment_decorator.pyi +2 -2
  97. metaflow-stubs/plugins/events_decorator.pyi +2 -2
  98. metaflow-stubs/plugins/exit_hook/__init__.pyi +2 -2
  99. metaflow-stubs/plugins/exit_hook/exit_hook_decorator.pyi +2 -2
  100. metaflow-stubs/plugins/frameworks/__init__.pyi +2 -2
  101. metaflow-stubs/plugins/frameworks/pytorch.pyi +2 -2
  102. metaflow-stubs/plugins/gcp/__init__.pyi +2 -2
  103. metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi +5 -5
  104. metaflow-stubs/plugins/gcp/gs_exceptions.pyi +2 -2
  105. metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi +2 -2
  106. metaflow-stubs/plugins/gcp/gs_utils.pyi +2 -2
  107. metaflow-stubs/plugins/gcp/includefile_support.pyi +2 -2
  108. metaflow-stubs/plugins/kubernetes/__init__.pyi +2 -2
  109. metaflow-stubs/plugins/kubernetes/kube_utils.pyi +3 -3
  110. metaflow-stubs/plugins/kubernetes/kubernetes.pyi +2 -2
  111. metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +2 -2
  112. metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +2 -2
  113. metaflow-stubs/plugins/kubernetes/kubernetes_job.pyi +2 -2
  114. metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi +2 -2
  115. metaflow-stubs/plugins/kubernetes/spot_monitor_sidecar.pyi +2 -2
  116. metaflow-stubs/plugins/parallel_decorator.pyi +2 -2
  117. metaflow-stubs/plugins/project_decorator.pyi +2 -2
  118. metaflow-stubs/plugins/pypi/__init__.pyi +3 -3
  119. metaflow-stubs/plugins/pypi/conda_decorator.pyi +2 -2
  120. metaflow-stubs/plugins/pypi/conda_environment.pyi +7 -7
  121. metaflow-stubs/plugins/pypi/parsers.pyi +2 -2
  122. metaflow-stubs/plugins/pypi/pypi_decorator.pyi +2 -2
  123. metaflow-stubs/plugins/pypi/pypi_environment.pyi +2 -2
  124. metaflow-stubs/plugins/pypi/utils.pyi +2 -2
  125. metaflow-stubs/plugins/resources_decorator.pyi +2 -2
  126. metaflow-stubs/plugins/retry_decorator.pyi +2 -2
  127. metaflow-stubs/plugins/secrets/__init__.pyi +2 -2
  128. metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +4 -4
  129. metaflow-stubs/plugins/secrets/secrets_decorator.pyi +2 -2
  130. metaflow-stubs/plugins/secrets/secrets_func.pyi +2 -2
  131. metaflow-stubs/plugins/secrets/secrets_spec.pyi +2 -2
  132. metaflow-stubs/plugins/secrets/utils.pyi +2 -2
  133. metaflow-stubs/plugins/storage_executor.pyi +2 -2
  134. metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +2 -2
  135. metaflow-stubs/plugins/timeout_decorator.pyi +3 -3
  136. metaflow-stubs/plugins/uv/__init__.pyi +2 -2
  137. metaflow-stubs/plugins/uv/uv_environment.pyi +3 -3
  138. metaflow-stubs/pylint_wrapper.pyi +2 -2
  139. metaflow-stubs/runner/__init__.pyi +2 -2
  140. metaflow-stubs/runner/deployer.pyi +33 -33
  141. metaflow-stubs/runner/deployer_impl.pyi +3 -3
  142. metaflow-stubs/runner/metaflow_runner.pyi +4 -4
  143. metaflow-stubs/runner/nbdeploy.pyi +2 -2
  144. metaflow-stubs/runner/nbrun.pyi +2 -2
  145. metaflow-stubs/runner/subprocess_manager.pyi +2 -2
  146. metaflow-stubs/runner/utils.pyi +4 -4
  147. metaflow-stubs/system/__init__.pyi +2 -2
  148. metaflow-stubs/system/system_logger.pyi +3 -3
  149. metaflow-stubs/system/system_monitor.pyi +2 -2
  150. metaflow-stubs/tagging_util.pyi +2 -2
  151. metaflow-stubs/tuple_util.pyi +2 -2
  152. metaflow-stubs/user_configs/__init__.pyi +2 -2
  153. metaflow-stubs/user_configs/config_options.pyi +4 -4
  154. metaflow-stubs/user_configs/config_parameters.pyi +6 -6
  155. metaflow-stubs/user_decorators/__init__.pyi +2 -2
  156. metaflow-stubs/user_decorators/common.pyi +2 -2
  157. metaflow-stubs/user_decorators/mutable_flow.pyi +4 -4
  158. metaflow-stubs/user_decorators/mutable_step.pyi +4 -4
  159. metaflow-stubs/user_decorators/user_flow_decorator.pyi +5 -5
  160. metaflow-stubs/user_decorators/user_step_decorator.pyi +6 -6
  161. metaflow-stubs/version.pyi +2 -2
  162. {metaflow_stubs-2.18.1.dist-info → metaflow_stubs-2.18.3.dist-info}/METADATA +2 -2
  163. metaflow_stubs-2.18.3.dist-info/RECORD +166 -0
  164. metaflow_stubs-2.18.1.dist-info/RECORD +0 -166
  165. {metaflow_stubs-2.18.1.dist-info → metaflow_stubs-2.18.3.dist-info}/WHEEL +0 -0
  166. {metaflow_stubs-2.18.1.dist-info → metaflow_stubs-2.18.3.dist-info}/top_level.txt +0 -0
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.217652 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.113216 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.217931 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.113511 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.218162 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.113753 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.165698 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.061113 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.218803 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.114452 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -14,6 +14,8 @@ if typing.TYPE_CHECKING:
14
14
  from ...exception import MetaflowException as MetaflowException
15
15
  from ..kubernetes.kubernetes_client import KubernetesClient as KubernetesClient
16
16
 
17
+ ARGO_EVENTS_SENSOR_NAMESPACE: str
18
+
17
19
  class ArgoClientException(metaflow.exception.MetaflowException, metaclass=type):
18
20
  ...
19
21
 
@@ -58,9 +60,9 @@ class ArgoClient(object, metaclass=type):
58
60
  ...
59
61
  def schedule_workflow_template(self, name, schedule = None, timezone = None):
60
62
  ...
61
- def register_sensor(self, name, sensor = None):
63
+ def register_sensor(self, name, sensor = None, sensor_namespace = 'default'):
62
64
  ...
63
- def delete_sensor(self, name):
65
+ def delete_sensor(self, name, sensor_namespace):
64
66
  """
65
67
  Issues an API call for deleting a sensor
66
68
 
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.181809 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.086152 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,13 +1,13 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.224048 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.120140 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import typing
10
9
  import metaflow
10
+ import typing
11
11
  if typing.TYPE_CHECKING:
12
12
  import metaflow.graph
13
13
  import metaflow.exception
@@ -35,6 +35,8 @@ ARGO_EVENTS_EVENT_SOURCE: None
35
35
 
36
36
  ARGO_EVENTS_INTERNAL_WEBHOOK_URL: None
37
37
 
38
+ ARGO_EVENTS_SENSOR_NAMESPACE: str
39
+
38
40
  ARGO_EVENTS_SERVICE_ACCOUNT: None
39
41
 
40
42
  ARGO_EVENTS_WEBHOOK_AUTH: str
@@ -100,6 +102,9 @@ UBF_TASK: str
100
102
  class ArgoWorkflowsException(metaflow.exception.MetaflowException, metaclass=type):
101
103
  ...
102
104
 
105
+ class ArgoWorkflowsSensorCleanupException(metaflow.exception.MetaflowException, metaclass=type):
106
+ ...
107
+
103
108
  class ArgoWorkflowsSchedulingException(metaflow.exception.MetaflowException, metaclass=type):
104
109
  ...
105
110
 
@@ -110,6 +115,8 @@ class ArgoWorkflows(object, metaclass=type):
110
115
  ...
111
116
  def deploy(self):
112
117
  ...
118
+ def cleanup_previous_sensors(self):
119
+ ...
113
120
  @staticmethod
114
121
  def list_templates(flow_name, all = False, page_size = 100):
115
122
  ...
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.182125 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.086471 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,9 +9,9 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.graph
13
12
  import metaflow.decorators
14
13
  import metaflow.flowspec
14
+ import metaflow.graph
15
15
 
16
16
  from ...metaflow_current import current as current
17
17
  from ...events import Trigger as Trigger
@@ -1,17 +1,17 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.182590 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.086989 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import typing
10
9
  import metaflow
10
+ import typing
11
11
  if typing.TYPE_CHECKING:
12
- import typing
13
- import metaflow.runner.deployer_impl
14
12
  import metaflow.plugins.argo.argo_workflows_deployer_objects
13
+ import metaflow.runner.deployer_impl
14
+ import typing
15
15
 
16
16
  from ...runner.deployer_impl import DeployerImpl as DeployerImpl
17
17
 
@@ -1,13 +1,13 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.212279 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.107516 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import typing
10
9
  import metaflow
10
+ import typing
11
11
  if typing.TYPE_CHECKING:
12
12
  import metaflow.plugins.argo.argo_workflows_deployer_objects
13
13
  import metaflow.runner.deployer
@@ -1,13 +1,13 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.226941 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.123730 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import typing
10
9
  import metaflow
10
+ import typing
11
11
  if typing.TYPE_CHECKING:
12
12
  import metaflow.plugins.argo.exit_hooks
13
13
 
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.165946 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.061552 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.187068 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.088259 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.186565 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.087707 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.182812 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.087257 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.214637 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.109976 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -59,13 +59,13 @@ class BatchKilledException(metaflow.exception.MetaflowException, metaclass=type)
59
59
  ...
60
60
 
61
61
  class Batch(object, metaclass=type):
62
- def __init__(self, metadata, environment):
62
+ def __init__(self, metadata, environment, flow_datastore = None):
63
63
  ...
64
64
  def list_jobs(self, flow_name, run_id, user, echo):
65
65
  ...
66
66
  def kill_jobs(self, flow_name, run_id, user, echo):
67
67
  ...
68
- def create_job(self, step_name, step_cli, task_spec, code_package_metadata, code_package_sha, code_package_url, code_package_ds, image, queue, iam_role = None, execution_role = None, cpu = None, gpu = None, memory = None, run_time_limit = None, shared_memory = None, max_swap = None, swappiness = None, inferentia = None, efa = None, env = {}, attrs = {}, host_volumes = None, efs_volumes = None, use_tmpfs = None, aws_batch_tags = None, tmpfs_tempdir = None, tmpfs_size = None, tmpfs_path = None, num_parallel = 0, ephemeral_storage = None, log_driver = None, log_options = None):
68
+ def create_job(self, step_name, step_cli, task_spec, code_package_metadata, code_package_sha, code_package_url, code_package_ds, image, queue, iam_role = None, execution_role = None, cpu = None, gpu = None, memory = None, run_time_limit = None, shared_memory = None, max_swap = None, swappiness = None, inferentia = None, efa = None, env = {}, attrs = {}, host_volumes = None, efs_volumes = None, use_tmpfs = None, aws_batch_tags = None, tmpfs_tempdir = None, tmpfs_size = None, tmpfs_path = None, num_parallel = 0, ephemeral_storage = None, log_driver = None, log_options = None, offload_command_to_s3 = False):
69
69
  ...
70
70
  def launch_job(self, step_name, step_cli, task_spec, code_package_metadata, code_package_sha, code_package_url, code_package_ds, image, queue, iam_role = None, execution_role = None, cpu = None, gpu = None, memory = None, run_time_limit = None, shared_memory = None, max_swap = None, swappiness = None, inferentia = None, efa = None, host_volumes = None, efs_volumes = None, use_tmpfs = None, aws_batch_tags = None, tmpfs_tempdir = None, tmpfs_size = None, tmpfs_path = None, num_parallel = 0, env = {}, attrs = {}, ephemeral_storage = None, log_driver = None, log_options = None):
71
71
  ...
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.213806 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.109095 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.215297 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.110663 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.187260 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.088469 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,18 +1,18 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.216894 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.112371 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import abc
10
9
  import metaflow
10
+ import abc
11
11
  import typing
12
12
  if typing.TYPE_CHECKING:
13
- import abc
14
- import metaflow.exception
15
13
  import metaflow.plugins.secrets
14
+ import metaflow.exception
15
+ import abc
16
16
 
17
17
  from ....exception import MetaflowException as MetaflowException
18
18
  from ...secrets import SecretsProvider as SecretsProvider
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.186788 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.087963 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.215592 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.110973 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.228977 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.125826 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.216077 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.111473 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.228628 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.125458 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -37,7 +37,7 @@ class StepFunctionsSchedulingException(metaflow.exception.MetaflowException, met
37
37
  ...
38
38
 
39
39
  class StepFunctions(object, metaclass=type):
40
- def __init__(self, name, graph, flow, code_package_metadata, code_package_sha, code_package_url, production_token, metadata, flow_datastore, environment, event_logger, monitor, tags = None, aws_batch_tags = None, namespace = None, username = None, max_workers = None, workflow_timeout = None, is_project = False, use_distributed_map = False):
40
+ def __init__(self, name, graph, flow, code_package_metadata, code_package_sha, code_package_url, production_token, metadata, flow_datastore, environment, event_logger, monitor, tags = None, aws_batch_tags = None, namespace = None, username = None, max_workers = None, workflow_timeout = None, is_project = False, use_distributed_map = False, compress_state_machine = False):
41
41
  ...
42
42
  def to_json(self):
43
43
  ...
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.229382 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.126232 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.215863 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.111247 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,17 +1,17 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.216543 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.112000 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import typing
10
9
  import metaflow
10
+ import typing
11
11
  if typing.TYPE_CHECKING:
12
- import typing
13
- import metaflow.runner.deployer_impl
14
12
  import metaflow.plugins.aws.step_functions.step_functions_deployer_objects
13
+ import metaflow.runner.deployer_impl
14
+ import typing
15
15
 
16
16
  from ....runner.deployer_impl import DeployerImpl as DeployerImpl
17
17
 
@@ -72,6 +72,9 @@ class StepFunctionsDeployer(metaflow.runner.deployer_impl.DeployerImpl, metaclas
72
72
  use_distributed_map : bool, optional, default False
73
73
  Use AWS Step Functions Distributed Map instead of Inline Map for defining foreach
74
74
  tasks in Amazon State Language.
75
+ compress_state_machine : bool, optional, default False
76
+ Compress AWS Step Functions state machine to fit within the 8K limit.
77
+
75
78
  deployer_attribute_file : str, optional, default None
76
79
  Write the workflow name to the specified file. Used internally for Metaflow's Deployer API.
77
80
 
@@ -1,13 +1,13 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.226319 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.123096 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import typing
10
9
  import metaflow
10
+ import typing
11
11
  if typing.TYPE_CHECKING:
12
12
  import metaflow.plugins.aws.step_functions.step_functions_deployer_objects
13
13
  import metaflow.runner.deployer
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.171667 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.064397 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.206223 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.098748 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.206393 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.098945 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,18 +1,18 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.207870 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.100494 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import abc
10
9
  import metaflow
10
+ import abc
11
11
  import typing
12
12
  if typing.TYPE_CHECKING:
13
- import abc
14
- import metaflow.exception
15
13
  import metaflow.plugins.secrets
14
+ import metaflow.exception
15
+ import abc
16
16
 
17
17
  from ..secrets import SecretsProvider as SecretsProvider
18
18
  from ...exception import MetaflowException as MetaflowException
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.206736 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.099280 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.207017 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.099569 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.207510 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.100119 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.1 #
4
- # Generated on 2025-08-29T13:35:56.168368 #
3
+ # MF version: 2.18.3 #
4
+ # Generated on 2025-09-08T23:52:16.059529 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations