metaflow-stubs 2.12.39__py2.py3-none-any.whl → 2.13.1__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.
Files changed (144) hide show
  1. metaflow-stubs/__init__.pyi +266 -265
  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 +2 -2
  9. metaflow-stubs/events.pyi +2 -2
  10. metaflow-stubs/exception.pyi +3 -3
  11. metaflow-stubs/flowspec.pyi +7 -7
  12. metaflow-stubs/generated_for.txt +1 -1
  13. metaflow-stubs/includefile.pyi +4 -4
  14. metaflow-stubs/info_file.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 +2 -2
  18. metaflow-stubs/metadata_provider/util.pyi +2 -2
  19. metaflow-stubs/metaflow_config.pyi +6 -2
  20. metaflow-stubs/metaflow_current.pyi +27 -27
  21. metaflow-stubs/multicore_utils.pyi +2 -2
  22. metaflow-stubs/parameters.pyi +5 -5
  23. metaflow-stubs/plugins/__init__.pyi +14 -14
  24. metaflow-stubs/plugins/airflow/__init__.pyi +2 -2
  25. metaflow-stubs/plugins/airflow/airflow_decorator.pyi +2 -2
  26. metaflow-stubs/plugins/airflow/airflow_utils.pyi +2 -2
  27. metaflow-stubs/plugins/airflow/exception.pyi +2 -2
  28. metaflow-stubs/plugins/airflow/sensors/__init__.pyi +2 -2
  29. metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi +2 -2
  30. metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi +2 -2
  31. metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi +2 -2
  32. metaflow-stubs/plugins/argo/__init__.pyi +2 -2
  33. metaflow-stubs/plugins/argo/argo_client.pyi +2 -2
  34. metaflow-stubs/plugins/argo/argo_events.pyi +2 -2
  35. metaflow-stubs/plugins/argo/argo_workflows.pyi +3 -9
  36. metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +3 -5
  37. metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +4 -4
  38. metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi +31 -3
  39. metaflow-stubs/plugins/aws/__init__.pyi +3 -3
  40. metaflow-stubs/plugins/aws/aws_client.pyi +2 -2
  41. metaflow-stubs/plugins/aws/aws_utils.pyi +2 -2
  42. metaflow-stubs/plugins/aws/batch/__init__.pyi +2 -2
  43. metaflow-stubs/plugins/aws/batch/batch.pyi +2 -2
  44. metaflow-stubs/plugins/aws/batch/batch_client.pyi +2 -2
  45. metaflow-stubs/plugins/aws/batch/batch_decorator.pyi +9 -10
  46. metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi +2 -2
  47. metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi +16 -14
  48. metaflow-stubs/plugins/aws/step_functions/__init__.pyi +2 -2
  49. metaflow-stubs/plugins/aws/step_functions/dynamo_db_client.pyi +2 -2
  50. metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi +2 -2
  51. metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi +2 -2
  52. metaflow-stubs/plugins/aws/step_functions/step_functions.pyi +2 -2
  53. metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi +2 -2
  54. metaflow-stubs/plugins/aws/step_functions/step_functions_decorator.pyi +2 -2
  55. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi +4 -4
  56. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi +3 -3
  57. metaflow-stubs/plugins/azure/__init__.pyi +2 -2
  58. metaflow-stubs/plugins/azure/azure_credential.pyi +2 -2
  59. metaflow-stubs/plugins/azure/azure_exceptions.pyi +2 -2
  60. metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi +4 -4
  61. metaflow-stubs/plugins/azure/azure_utils.pyi +2 -2
  62. metaflow-stubs/plugins/azure/blob_service_client_factory.pyi +2 -2
  63. metaflow-stubs/plugins/azure/includefile_support.pyi +2 -2
  64. metaflow-stubs/plugins/cards/__init__.pyi +2 -2
  65. metaflow-stubs/plugins/cards/card_client.pyi +3 -3
  66. metaflow-stubs/plugins/cards/card_creator.pyi +2 -2
  67. metaflow-stubs/plugins/cards/card_datastore.pyi +2 -2
  68. metaflow-stubs/plugins/cards/card_decorator.pyi +4 -2
  69. metaflow-stubs/plugins/cards/card_modules/__init__.pyi +2 -2
  70. metaflow-stubs/plugins/cards/card_modules/basic.pyi +3 -3
  71. metaflow-stubs/plugins/cards/card_modules/card.pyi +2 -2
  72. metaflow-stubs/plugins/cards/card_modules/components.pyi +4 -4
  73. metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi +2 -2
  74. metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi +2 -2
  75. metaflow-stubs/plugins/cards/card_modules/test_cards.pyi +2 -2
  76. metaflow-stubs/plugins/cards/card_resolver.pyi +2 -2
  77. metaflow-stubs/plugins/cards/component_serializer.pyi +2 -2
  78. metaflow-stubs/plugins/cards/exception.pyi +2 -2
  79. metaflow-stubs/plugins/catch_decorator.pyi +2 -2
  80. metaflow-stubs/plugins/datatools/__init__.pyi +2 -2
  81. metaflow-stubs/plugins/datatools/local.pyi +2 -2
  82. metaflow-stubs/plugins/datatools/s3/__init__.pyi +2 -2
  83. metaflow-stubs/plugins/datatools/s3/s3.pyi +3 -3
  84. metaflow-stubs/plugins/datatools/s3/s3tail.pyi +2 -2
  85. metaflow-stubs/plugins/datatools/s3/s3util.pyi +2 -2
  86. metaflow-stubs/plugins/debug_logger.pyi +2 -2
  87. metaflow-stubs/plugins/debug_monitor.pyi +2 -2
  88. metaflow-stubs/plugins/environment_decorator.pyi +2 -2
  89. metaflow-stubs/plugins/events_decorator.pyi +2 -2
  90. metaflow-stubs/plugins/frameworks/__init__.pyi +2 -2
  91. metaflow-stubs/plugins/frameworks/pytorch.pyi +2 -2
  92. metaflow-stubs/plugins/gcp/__init__.pyi +2 -2
  93. metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi +4 -4
  94. metaflow-stubs/plugins/gcp/gs_exceptions.pyi +2 -2
  95. metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi +2 -2
  96. metaflow-stubs/plugins/gcp/gs_utils.pyi +2 -2
  97. metaflow-stubs/plugins/gcp/includefile_support.pyi +2 -2
  98. metaflow-stubs/plugins/kubernetes/__init__.pyi +2 -2
  99. metaflow-stubs/plugins/kubernetes/kube_utils.pyi +25 -2
  100. metaflow-stubs/plugins/kubernetes/kubernetes.pyi +4 -21
  101. metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +2 -2
  102. metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +12 -3
  103. metaflow-stubs/plugins/kubernetes/kubernetes_job.pyi +3 -7
  104. metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi +10 -4
  105. metaflow-stubs/plugins/parallel_decorator.pyi +2 -2
  106. metaflow-stubs/plugins/project_decorator.pyi +2 -2
  107. metaflow-stubs/plugins/pypi/__init__.pyi +3 -3
  108. metaflow-stubs/plugins/pypi/conda_decorator.pyi +2 -2
  109. metaflow-stubs/plugins/pypi/conda_environment.pyi +5 -5
  110. metaflow-stubs/plugins/pypi/pypi_decorator.pyi +2 -2
  111. metaflow-stubs/plugins/pypi/pypi_environment.pyi +2 -2
  112. metaflow-stubs/plugins/pypi/utils.pyi +6 -2
  113. metaflow-stubs/plugins/resources_decorator.pyi +2 -2
  114. metaflow-stubs/plugins/retry_decorator.pyi +2 -2
  115. metaflow-stubs/plugins/secrets/__init__.pyi +3 -3
  116. metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +3 -3
  117. metaflow-stubs/plugins/secrets/secrets_decorator.pyi +2 -2
  118. metaflow-stubs/plugins/storage_executor.pyi +2 -2
  119. metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +3 -3
  120. metaflow-stubs/plugins/timeout_decorator.pyi +2 -2
  121. metaflow-stubs/pylint_wrapper.pyi +2 -2
  122. metaflow-stubs/runner/__init__.pyi +2 -2
  123. metaflow-stubs/runner/deployer.pyi +28 -28
  124. metaflow-stubs/runner/deployer_impl.pyi +3 -3
  125. metaflow-stubs/runner/metaflow_runner.pyi +3 -3
  126. metaflow-stubs/runner/nbdeploy.pyi +2 -2
  127. metaflow-stubs/runner/nbrun.pyi +2 -2
  128. metaflow-stubs/runner/subprocess_manager.pyi +2 -2
  129. metaflow-stubs/runner/utils.pyi +4 -4
  130. metaflow-stubs/system/__init__.pyi +2 -2
  131. metaflow-stubs/system/system_logger.pyi +3 -3
  132. metaflow-stubs/system/system_monitor.pyi +2 -2
  133. metaflow-stubs/tagging_util.pyi +2 -2
  134. metaflow-stubs/tuple_util.pyi +2 -2
  135. metaflow-stubs/user_configs/__init__.pyi +2 -2
  136. metaflow-stubs/user_configs/config_decorators.pyi +6 -6
  137. metaflow-stubs/user_configs/config_options.pyi +9 -4
  138. metaflow-stubs/user_configs/config_parameters.pyi +7 -7
  139. metaflow-stubs/version.pyi +2 -2
  140. {metaflow_stubs-2.12.39.dist-info → metaflow_stubs-2.13.1.dist-info}/METADATA +2 -2
  141. metaflow_stubs-2.13.1.dist-info/RECORD +144 -0
  142. {metaflow_stubs-2.12.39.dist-info → metaflow_stubs-2.13.1.dist-info}/WHEEL +1 -1
  143. metaflow_stubs-2.12.39.dist-info/RECORD +0 -144
  144. {metaflow_stubs-2.12.39.dist-info → metaflow_stubs-2.13.1.dist-info}/top_level.txt +0 -0
metaflow-stubs/plugins/argo/argo_workflows.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.672815 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.361800 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,8 +9,8 @@ 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.exception
13
+ import metaflow.graph
14
14
 
15
15
  from ...parameters import JSONType as JSONType
16
16
  from ...metaflow_current import current as current
@@ -18,8 +18,6 @@ from ...exception import MetaflowException as MetaflowException
18
18
  from ...includefile import FilePathClass as FilePathClass
19
19
  from ...parameters import deploy_time_eval as deploy_time_eval
20
20
  from ..kubernetes.kube_utils import qos_requests_and_limits as qos_requests_and_limits
21
- from ..kubernetes.kubernetes import parse_kube_keyvalue_list as parse_kube_keyvalue_list
22
- from ..kubernetes.kubernetes import validate_kube_labels as validate_kube_labels
23
21
  from ..kubernetes.kubernetes_jobsets import KubernetesArgoJobSet as KubernetesArgoJobSet
24
22
  from ...user_configs.config_options import ConfigInput as ConfigInput
25
23
  from .argo_client import ArgoClient as ArgoClient
@@ -74,12 +72,8 @@ GCP_SECRET_MANAGER_PREFIX: None
74
72
 
75
73
  KUBERNETES_FETCH_EC2_METADATA: bool
76
74
 
77
- KUBERNETES_LABELS: str
78
-
79
75
  KUBERNETES_NAMESPACE: str
80
76
 
81
- KUBERNETES_NODE_SELECTOR: str
82
-
83
77
  KUBERNETES_SANDBOX_INIT_SCRIPT: None
84
78
 
85
79
  KUBERNETES_SECRETS: str
metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.636802 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.334955 #
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.flowspec
13
12
  import metaflow.graph
14
13
  import metaflow.decorators
14
+ import metaflow.flowspec
15
15
 
16
16
  from ...metaflow_current import current as current
17
17
  from ...events import Trigger as Trigger
@@ -19,8 +19,6 @@ from ...metadata_provider.metadata import MetaDatum as MetaDatum
19
19
  from ...flowspec import FlowSpec as FlowSpec
20
20
  from .argo_events import ArgoEvent as ArgoEvent
21
21
 
22
- ARGO_EVENTS_WEBHOOK_URL: None
23
-
24
22
  class ArgoWorkflowsInternalDecorator(metaflow.decorators.StepDecorator, metaclass=type):
25
23
  def task_pre_step(self, step_name, task_datastore, metadata, run_id, task_id, flow, graph, retry_count, max_user_code_retries, ubf_context, inputs):
26
24
  ...
metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.637320 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.335408 #
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.plugins.argo.argo_workflows_deployer_objects
13
- import typing
14
12
  import metaflow.runner.deployer_impl
13
+ import typing
14
+ import metaflow.plugins.argo.argo_workflows_deployer_objects
15
15
 
16
16
  from ...runner.deployer_impl import DeployerImpl as DeployerImpl
17
17
 
metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.660753 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.351331 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.plugins.argo.argo_workflows_deployer_objects
13
12
  import metaflow.runner.deployer
13
+ import metaflow.plugins.argo.argo_workflows_deployer_objects
14
14
 
15
15
  from ...client.core import get_metadata as get_metadata
16
16
  from ...exception import MetaflowException as MetaflowException
@@ -77,6 +77,34 @@ class ArgoWorkflowsTriggeredRun(metaflow.runner.deployer.TriggeredRun, metaclass
77
77
  True if the command was successful, False otherwise.
78
78
  """
79
79
  ...
80
+ def wait_for_completion(self, timeout: typing.Optional[int] = None):
81
+ """
82
+ Wait for the workflow to complete or timeout.
83
+
84
+ Parameters
85
+ ----------
86
+ timeout : int, optional, default None
87
+ Maximum time in seconds to wait for workflow completion.
88
+ If None, waits indefinitely.
89
+
90
+ Raises
91
+ ------
92
+ TimeoutError
93
+ If the workflow does not complete within the specified timeout period.
94
+ """
95
+ ...
96
+ @property
97
+ def is_running(self):
98
+ """
99
+ Check if the workflow is currently running.
100
+
101
+ Returns
102
+ -------
103
+ bool
104
+ True if the workflow status is either 'Pending' or 'Running',
105
+ False otherwise.
106
+ """
107
+ ...
80
108
  @property
81
109
  def status(self) -> typing.Optional[str]:
82
110
  """
metaflow-stubs/plugins/aws/__init__.pyi CHANGED
@@ -1,15 +1,15 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.619678 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.315197 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
9
 
10
- from . import batch as batch
11
10
  from . import aws_utils as aws_utils
12
11
  from . import step_functions as step_functions
12
+ from . import batch as batch
13
13
  from . import aws_client as aws_client
14
14
  from . import secrets_manager as secrets_manager
15
15
 
metaflow-stubs/plugins/aws/aws_client.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.640912 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.341116 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/aws_utils.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.640360 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.340392 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/batch/__init__.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.639994 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.340839 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/batch/batch.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.663338 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.354775 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/batch/batch_client.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.662244 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.353902 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/batch/batch_decorator.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.664023 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.355378 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -12,27 +12,26 @@ if typing.TYPE_CHECKING:
12
12
  import metaflow.decorators
13
13
 
14
14
  from ....metaflow_current import current as current
15
- from ...resources_decorator import ResourcesDecorator as ResourcesDecorator
16
- from ...timeout_decorator import get_run_time_limit_for_task as get_run_time_limit_for_task
17
15
  from ....metadata_provider.metadata import MetaDatum as MetaDatum
18
16
  from ....metadata_provider.util import sync_local_metadata_to_datastore as sync_local_metadata_to_datastore
19
- from .batch import BatchException as BatchException
17
+ from ...timeout_decorator import get_run_time_limit_for_task as get_run_time_limit_for_task
20
18
  from ..aws_utils import compute_resource_attributes as compute_resource_attributes
21
19
  from ..aws_utils import get_docker_registry as get_docker_registry
22
20
  from ..aws_utils import get_ec2_instance_metadata as get_ec2_instance_metadata
23
-
24
- ECS_S3_ACCESS_IAM_ROLE: None
25
-
26
- BATCH_JOB_QUEUE: None
21
+ from .batch import BatchException as BatchException
27
22
 
28
23
  BATCH_CONTAINER_IMAGE: None
29
24
 
30
25
  BATCH_CONTAINER_REGISTRY: None
31
26
 
32
- ECS_FARGATE_EXECUTION_ROLE: None
27
+ BATCH_JOB_QUEUE: None
33
28
 
34
29
  DATASTORE_LOCAL_DIR: str
35
30
 
31
+ ECS_FARGATE_EXECUTION_ROLE: None
32
+
33
+ ECS_S3_ACCESS_IAM_ROLE: None
34
+
36
35
  UBF_CONTROL: str
37
36
 
38
37
  class BatchDecorator(metaflow.decorators.StepDecorator, metaclass=type):
metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.641191 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.341328 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi CHANGED
@@ -1,17 +1,17 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.665761 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.355705 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import metaflow
10
9
  import abc
10
+ import metaflow
11
11
  import typing
12
12
  if typing.TYPE_CHECKING:
13
- import abc
14
13
  import metaflow.exception
14
+ import abc
15
15
  import metaflow.plugins.secrets
16
16
 
17
17
  from ....exception import MetaflowException as MetaflowException
@@ -51,23 +51,25 @@ class AwsSecretsManagerSecretsProvider(metaflow.plugins.secrets.SecretsProvider,
51
51
  The secret payload from AWS is EITHER a string OR a binary blob.
52
52
 
53
53
  If the secret contains a string payload ("SecretString"):
54
- - if the `parse_secret_string_as_json` option is True (default):
54
+ - if the `json` option is True (default):
55
55
  {SecretString} will be parsed as a JSON. If successfully parsed, AND the JSON contains a
56
56
  top-level object, each entry K/V in the object will also be converted to an entry in the result. V will
57
57
  always be casted to a string (if not already a string).
58
- - If `parse_secret_string_as_json` option is False:
59
- {SecretString} will be returned as a single entry in the result, with the key being the secret_id.
58
+ - If `json` option is False:
59
+ {SecretString} will be returned as a single entry in the result, where the key is either:
60
+ - the `secret_id`, OR
61
+ - the value set by `options={"env_var_name": custom_env_var_name}`.
60
62
 
61
- Otherwise, the secret contains a binary blob payload ("SecretBinary"). In this case
62
- - The result dic contains '{SecretName}': '{SecretBinary}', where {SecretBinary} is a base64-encoded string
63
+ Otherwise, if the secret contains a binary blob payload ("SecretBinary"):
64
+ - The result dict contains '{SecretName}': '{SecretBinary}', where {SecretBinary} is a base64-encoded string.
63
65
 
64
- All keys in the result are sanitized to be more valid environment variable names. This is done on a best effort
66
+ All keys in the result are sanitized to be more valid environment variable names. This is done on a best-effort
65
67
  basis. Further validation is expected to be done by the invoking @secrets decorator itself.
66
68
 
67
- :param secret_id: ARN or friendly name of the secret
68
- :param options: unused
69
- :param role: AWS IAM Role ARN to assume before reading the secret
70
- :return: dict of environment variables. All keys and values are strings.
69
+ :param secret_id: ARN or friendly name of the secret.
70
+ :param options: Dictionary of additional options. E.g., `options={"env_var_name": custom_env_var_name}`.
71
+ :param role: AWS IAM Role ARN to assume before reading the secret.
72
+ :return: Dictionary of environment variables. All keys and values are strings.
71
73
  """
72
74
  ...
73
75
  ...
metaflow-stubs/plugins/aws/step_functions/__init__.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.640599 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.340612 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/step_functions/dynamo_db_client.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.664314 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.351609 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.677018 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.365407 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.664841 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.352089 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/step_functions/step_functions.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.676661 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.365074 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.677434 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.365756 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/step_functions/step_functions_decorator.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.664604 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.351873 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.665394 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.352548 #
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 typing
13
- import metaflow.plugins.aws.step_functions.step_functions_deployer_objects
14
12
  import metaflow.runner.deployer_impl
13
+ import metaflow.plugins.aws.step_functions.step_functions_deployer_objects
14
+ import typing
15
15
 
16
16
  from ....runner.deployer_impl import DeployerImpl as DeployerImpl
17
17
 
metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.675024 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.363611 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.plugins.aws.step_functions.step_functions_deployer_objects
13
12
  import metaflow.runner.deployer
13
+ import metaflow.plugins.aws.step_functions.step_functions_deployer_objects
14
14
 
15
15
  from .step_functions import StepFunctions as StepFunctions
16
16
  from ....runner.deployer import DeployedFlow as DeployedFlow
metaflow-stubs/plugins/azure/__init__.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.623810 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.317725 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/azure/azure_credential.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.655575 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.345320 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/azure/azure_exceptions.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.655753 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.345488 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi CHANGED
@@ -1,17 +1,17 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.657338 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.346917 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import metaflow
10
9
  import abc
10
+ import metaflow
11
11
  import typing
12
12
  if typing.TYPE_CHECKING:
13
- import abc
14
13
  import metaflow.exception
14
+ import abc
15
15
  import metaflow.plugins.secrets
16
16
 
17
17
  from ..secrets import SecretsProvider as SecretsProvider
metaflow-stubs/plugins/azure/azure_utils.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.656101 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.345811 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/azure/blob_service_client_factory.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.656417 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.346097 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/azure/includefile_support.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.656950 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.346599 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/cards/__init__.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.620612 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.314337 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/cards/card_client.pyi CHANGED
@@ -1,15 +1,15 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.630032 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.323735 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
9
  import typing
10
10
  if typing.TYPE_CHECKING:
11
- import metaflow
12
11
  import metaflow.plugins.cards.card_client
12
+ import metaflow
13
13
 
14
14
  from .card_resolver import resolve_paths_from_task as resolve_paths_from_task
15
15
  from .card_resolver import resumed_info as resumed_info
metaflow-stubs/plugins/cards/card_creator.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.644854 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.332659 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/plugins/cards/card_datastore.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39 #
4
- # Generated on 2024-12-10T16:02:32.646255 #
3
+ # MF version: 2.13.1 #
4
+ # Generated on 2025-01-06T13:29:15.333906 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations