metaflow-stubs 2.15.15__py2.py3-none-any.whl → 2.15.17__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 (149) hide show
  1. metaflow-stubs/__init__.pyi +456 -452
  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 +26 -26
  8. metaflow-stubs/client/filecache.pyi +3 -3
  9. metaflow-stubs/events.pyi +6 -6
  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 +5 -5
  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 +3 -3
  18. metaflow-stubs/metadata_provider/util.pyi +2 -2
  19. metaflow-stubs/metaflow_config.pyi +4 -2
  20. metaflow-stubs/metaflow_current.pyi +22 -22
  21. metaflow-stubs/metaflow_git.pyi +2 -2
  22. metaflow-stubs/multicore_utils.pyi +2 -2
  23. metaflow-stubs/parameters.pyi +4 -4
  24. metaflow-stubs/plugins/__init__.pyi +11 -11
  25. metaflow-stubs/plugins/airflow/__init__.pyi +2 -2
  26. metaflow-stubs/plugins/airflow/airflow_decorator.pyi +2 -2
  27. metaflow-stubs/plugins/airflow/airflow_utils.pyi +2 -2
  28. metaflow-stubs/plugins/airflow/exception.pyi +2 -2
  29. metaflow-stubs/plugins/airflow/sensors/__init__.pyi +2 -2
  30. metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi +2 -2
  31. metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi +2 -2
  32. metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi +2 -2
  33. metaflow-stubs/plugins/argo/__init__.pyi +2 -2
  34. metaflow-stubs/plugins/argo/argo_client.pyi +2 -2
  35. metaflow-stubs/plugins/argo/argo_events.pyi +2 -2
  36. metaflow-stubs/plugins/argo/argo_workflows.pyi +5 -3
  37. metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +4 -4
  38. metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +4 -4
  39. metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi +4 -6
  40. metaflow-stubs/plugins/aws/__init__.pyi +2 -2
  41. metaflow-stubs/plugins/aws/aws_client.pyi +2 -2
  42. metaflow-stubs/plugins/aws/aws_utils.pyi +2 -2
  43. metaflow-stubs/plugins/aws/batch/__init__.pyi +2 -2
  44. metaflow-stubs/plugins/aws/batch/batch.pyi +2 -2
  45. metaflow-stubs/plugins/aws/batch/batch_client.pyi +2 -2
  46. metaflow-stubs/plugins/aws/batch/batch_decorator.pyi +2 -2
  47. metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi +2 -2
  48. metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi +3 -3
  49. metaflow-stubs/plugins/aws/step_functions/__init__.pyi +2 -2
  50. metaflow-stubs/plugins/aws/step_functions/dynamo_db_client.pyi +2 -2
  51. metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi +2 -2
  52. metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi +2 -2
  53. metaflow-stubs/plugins/aws/step_functions/step_functions.pyi +2 -2
  54. metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi +2 -2
  55. metaflow-stubs/plugins/aws/step_functions/step_functions_decorator.pyi +2 -2
  56. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi +4 -4
  57. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi +4 -4
  58. metaflow-stubs/plugins/azure/__init__.pyi +2 -2
  59. metaflow-stubs/plugins/azure/azure_credential.pyi +2 -2
  60. metaflow-stubs/plugins/azure/azure_exceptions.pyi +2 -2
  61. metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi +3 -3
  62. metaflow-stubs/plugins/azure/azure_utils.pyi +2 -2
  63. metaflow-stubs/plugins/azure/blob_service_client_factory.pyi +2 -2
  64. metaflow-stubs/plugins/azure/includefile_support.pyi +2 -2
  65. metaflow-stubs/plugins/cards/__init__.pyi +2 -2
  66. metaflow-stubs/plugins/cards/card_client.pyi +2 -2
  67. metaflow-stubs/plugins/cards/card_creator.pyi +2 -2
  68. metaflow-stubs/plugins/cards/card_datastore.pyi +2 -2
  69. metaflow-stubs/plugins/cards/card_decorator.pyi +2 -2
  70. metaflow-stubs/plugins/cards/card_modules/__init__.pyi +2 -2
  71. metaflow-stubs/plugins/cards/card_modules/basic.pyi +3 -3
  72. metaflow-stubs/plugins/cards/card_modules/card.pyi +2 -2
  73. metaflow-stubs/plugins/cards/card_modules/components.pyi +5 -5
  74. metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi +2 -2
  75. metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi +2 -2
  76. metaflow-stubs/plugins/cards/card_modules/test_cards.pyi +2 -2
  77. metaflow-stubs/plugins/cards/card_resolver.pyi +2 -2
  78. metaflow-stubs/plugins/cards/component_serializer.pyi +2 -2
  79. metaflow-stubs/plugins/cards/exception.pyi +2 -2
  80. metaflow-stubs/plugins/catch_decorator.pyi +3 -3
  81. metaflow-stubs/plugins/datatools/__init__.pyi +2 -2
  82. metaflow-stubs/plugins/datatools/local.pyi +2 -2
  83. metaflow-stubs/plugins/datatools/s3/__init__.pyi +2 -2
  84. metaflow-stubs/plugins/datatools/s3/s3.pyi +17 -17
  85. metaflow-stubs/plugins/datatools/s3/s3tail.pyi +2 -2
  86. metaflow-stubs/plugins/datatools/s3/s3util.pyi +2 -2
  87. metaflow-stubs/plugins/debug_logger.pyi +2 -2
  88. metaflow-stubs/plugins/debug_monitor.pyi +2 -2
  89. metaflow-stubs/plugins/environment_decorator.pyi +2 -2
  90. metaflow-stubs/plugins/events_decorator.pyi +2 -2
  91. metaflow-stubs/plugins/frameworks/__init__.pyi +2 -2
  92. metaflow-stubs/plugins/frameworks/pytorch.pyi +2 -2
  93. metaflow-stubs/plugins/gcp/__init__.pyi +2 -2
  94. metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi +3 -3
  95. metaflow-stubs/plugins/gcp/gs_exceptions.pyi +2 -2
  96. metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi +2 -2
  97. metaflow-stubs/plugins/gcp/gs_utils.pyi +2 -2
  98. metaflow-stubs/plugins/gcp/includefile_support.pyi +2 -2
  99. metaflow-stubs/plugins/kubernetes/__init__.pyi +2 -2
  100. metaflow-stubs/plugins/kubernetes/kube_utils.pyi +3 -3
  101. metaflow-stubs/plugins/kubernetes/kubernetes.pyi +4 -4
  102. metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +2 -2
  103. metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +8 -2
  104. metaflow-stubs/plugins/kubernetes/kubernetes_job.pyi +2 -2
  105. metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi +2 -2
  106. metaflow-stubs/plugins/kubernetes/spot_monitor_sidecar.pyi +2 -2
  107. metaflow-stubs/plugins/parallel_decorator.pyi +2 -2
  108. metaflow-stubs/plugins/project_decorator.pyi +2 -2
  109. metaflow-stubs/plugins/pypi/__init__.pyi +2 -2
  110. metaflow-stubs/plugins/pypi/conda_decorator.pyi +2 -2
  111. metaflow-stubs/plugins/pypi/conda_environment.pyi +5 -5
  112. metaflow-stubs/plugins/pypi/parsers.pyi +2 -2
  113. metaflow-stubs/plugins/pypi/pypi_decorator.pyi +2 -2
  114. metaflow-stubs/plugins/pypi/pypi_environment.pyi +2 -2
  115. metaflow-stubs/plugins/pypi/utils.pyi +2 -2
  116. metaflow-stubs/plugins/resources_decorator.pyi +2 -2
  117. metaflow-stubs/plugins/retry_decorator.pyi +2 -2
  118. metaflow-stubs/plugins/secrets/__init__.pyi +2 -2
  119. metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +2 -2
  120. metaflow-stubs/plugins/secrets/secrets_decorator.pyi +2 -2
  121. metaflow-stubs/plugins/storage_executor.pyi +2 -2
  122. metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +3 -3
  123. metaflow-stubs/plugins/timeout_decorator.pyi +3 -3
  124. metaflow-stubs/plugins/uv/__init__.pyi +2 -2
  125. metaflow-stubs/plugins/uv/uv_environment.pyi +3 -3
  126. metaflow-stubs/pylint_wrapper.pyi +2 -2
  127. metaflow-stubs/runner/__init__.pyi +2 -2
  128. metaflow-stubs/runner/deployer.pyi +9 -6
  129. metaflow-stubs/runner/deployer_impl.pyi +2 -2
  130. metaflow-stubs/runner/metaflow_runner.pyi +4 -4
  131. metaflow-stubs/runner/nbdeploy.pyi +2 -2
  132. metaflow-stubs/runner/nbrun.pyi +2 -2
  133. metaflow-stubs/runner/subprocess_manager.pyi +2 -2
  134. metaflow-stubs/runner/utils.pyi +3 -3
  135. metaflow-stubs/system/__init__.pyi +2 -2
  136. metaflow-stubs/system/system_logger.pyi +3 -3
  137. metaflow-stubs/system/system_monitor.pyi +2 -2
  138. metaflow-stubs/tagging_util.pyi +2 -2
  139. metaflow-stubs/tuple_util.pyi +2 -2
  140. metaflow-stubs/user_configs/__init__.pyi +2 -2
  141. metaflow-stubs/user_configs/config_decorators.pyi +7 -7
  142. metaflow-stubs/user_configs/config_options.pyi +3 -3
  143. metaflow-stubs/user_configs/config_parameters.pyi +4 -4
  144. metaflow-stubs/version.pyi +2 -2
  145. {metaflow_stubs-2.15.15.dist-info → metaflow_stubs-2.15.17.dist-info}/METADATA +2 -2
  146. metaflow_stubs-2.15.17.dist-info/RECORD +149 -0
  147. metaflow_stubs-2.15.15.dist-info/RECORD +0 -149
  148. {metaflow_stubs-2.15.15.dist-info → metaflow_stubs-2.15.17.dist-info}/WHEEL +0 -0
  149. {metaflow_stubs-2.15.15.dist-info → metaflow_stubs-2.15.17.dist-info}/top_level.txt +0 -0
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.15 #
4
- # Generated on 2025-05-30T13:33:59.754579 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:28.005883 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.753473 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:28.004997 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.750739 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:28.001926 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.735648 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:27.984469 #
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.cards.card_modules.basic
13
12
  import metaflow.plugins.cards.card_modules.card
13
+ import metaflow.plugins.cards.card_modules.basic
14
14
 
15
15
  from .card import MetaflowCard as MetaflowCard
16
16
  from .card import MetaflowCardComponent as MetaflowCardComponent
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.15 #
4
- # Generated on 2025-05-30T13:33:59.738449 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:27.987326 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.740216 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:27.989077 #
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.plugins.cards.card_modules.components
14
12
  import metaflow.plugins.cards.card_modules.card
13
+ import metaflow.plugins.cards.card_modules.components
14
+ import typing
15
15
 
16
16
  from .basic import LogComponent as LogComponent
17
17
  from .basic import ErrorComponent as ErrorComponent
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.15 #
4
- # Generated on 2025-05-30T13:33:59.765120 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:28.014268 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.765288 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:28.014446 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.736730 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:27.985583 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.754775 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:28.006144 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.752658 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:28.004169 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.751401 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:28.002889 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.727145 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:27.979680 #
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.exception
13
12
  import metaflow.decorators
13
+ import metaflow.exception
14
14
 
15
15
  from ..exception import MetaflowException as MetaflowException
16
16
  from ..exception import MetaflowExceptionWrapper as MetaflowExceptionWrapper
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.15 #
4
- # Generated on 2025-05-30T13:33:59.729979 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:27.978744 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.757574 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:28.007539 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.757789 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:28.007748 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.709041 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:27.957309 #
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 io
13
- import metaflow.exception
14
12
  import metaflow.plugins.datatools.s3.s3
13
+ import metaflow.exception
14
+ import io
15
15
 
16
16
  from ....flowspec import FlowSpec as FlowSpec
17
17
  from ....metaflow_current import current as current
@@ -161,7 +161,7 @@ class S3Object(object, metaclass=type):
161
161
  `S3Object`s are not instantiated directly, but they are returned
162
162
  by many methods of the `S3` client.
163
163
  """
164
- def __init__(self, prefix: str, url: str, path: str, size: typing.Optional[int] = None, content_type: typing.Optional[str] = None, metadata: typing.Optional[typing.Dict[str, str]] = None, range_info: typing.Optional[RangeInfo] = None, last_modified: typing.Optional[int] = None, encryption: typing.Optional[str] = None):
164
+ def __init__(self, prefix: str, url: str, path: str, size: typing.Optional[int] = None, content_type: typing.Optional[str] = None, metadata: typing.Optional[typing.Dict[str, str]] = None, range_info: typing.Optional[metaflow.plugins.datatools.s3.s3.RangeInfo] = None, last_modified: typing.Optional[int] = None, encryption: typing.Optional[str] = None):
165
165
  ...
166
166
  @property
167
167
  def exists(self) -> bool:
@@ -328,7 +328,7 @@ class S3Object(object, metaclass=type):
328
328
  """
329
329
  ...
330
330
  @property
331
- def range_info(self) -> typing.Optional[RangeInfo]:
331
+ def range_info(self) -> typing.Optional[metaflow.plugins.datatools.s3.s3.RangeInfo]:
332
332
  """
333
333
  If the object corresponds to a partially downloaded object, returns
334
334
  information of what was downloaded.
@@ -442,7 +442,7 @@ class S3(object, metaclass=type):
442
442
  Delete all temporary files downloaded in this context.
443
443
  """
444
444
  ...
445
- def list_paths(self, keys: typing.Optional[typing.Iterable[str]] = None) -> typing.List[S3Object]:
445
+ def list_paths(self, keys: typing.Optional[typing.Iterable[str]] = None) -> typing.List[metaflow.plugins.datatools.s3.s3.S3Object]:
446
446
  """
447
447
  List the next level of paths in S3.
448
448
 
@@ -479,7 +479,7 @@ class S3(object, metaclass=type):
479
479
  do not correspond to leaf objects.
480
480
  """
481
481
  ...
482
- def list_recursive(self, keys: typing.Optional[typing.Iterable[str]] = None) -> typing.List[S3Object]:
482
+ def list_recursive(self, keys: typing.Optional[typing.Iterable[str]] = None) -> typing.List[metaflow.plugins.datatools.s3.s3.S3Object]:
483
483
  """
484
484
  List all objects recursively under the given prefixes.
485
485
 
@@ -536,7 +536,7 @@ class S3(object, metaclass=type):
536
536
  will have `.downloaded == False`.
537
537
  """
538
538
  ...
539
- def info_many(self, keys: typing.Iterable[str], return_missing: bool = False) -> typing.List[S3Object]:
539
+ def info_many(self, keys: typing.Iterable[str], return_missing: bool = False) -> typing.List[metaflow.plugins.datatools.s3.s3.S3Object]:
540
540
  """
541
541
  Get metadata about many objects in S3 in parallel.
542
542
 
@@ -558,7 +558,7 @@ class S3(object, metaclass=type):
558
558
  objects will have `.downloaded == False`.
559
559
  """
560
560
  ...
561
- def get(self, key: typing.Union[str, S3GetObject, None] = None, return_missing: bool = False, return_info: bool = True) -> S3Object:
561
+ def get(self, key: typing.Union[str, metaflow.plugins.datatools.s3.s3.S3GetObject, None] = None, return_missing: bool = False, return_info: bool = True) -> S3Object:
562
562
  """
563
563
  Get a single object from S3.
564
564
 
@@ -581,7 +581,7 @@ class S3(object, metaclass=type):
581
581
  An S3Object corresponding to the object requested.
582
582
  """
583
583
  ...
584
- def get_many(self, keys: typing.Iterable[typing.Union[str, S3GetObject]], return_missing: bool = False, return_info: bool = True) -> typing.List[S3Object]:
584
+ def get_many(self, keys: typing.Iterable[typing.Union[str, metaflow.plugins.datatools.s3.s3.S3GetObject]], return_missing: bool = False, return_info: bool = True) -> typing.List[metaflow.plugins.datatools.s3.s3.S3Object]:
585
585
  """
586
586
  Get many objects from S3 in parallel.
587
587
 
@@ -603,7 +603,7 @@ class S3(object, metaclass=type):
603
603
  S3Objects corresponding to the objects requested.
604
604
  """
605
605
  ...
606
- def get_recursive(self, keys: typing.Iterable[str], return_info: bool = False) -> typing.List[S3Object]:
606
+ def get_recursive(self, keys: typing.Iterable[str], return_info: bool = False) -> typing.List[metaflow.plugins.datatools.s3.s3.S3Object]:
607
607
  """
608
608
  Get many objects from S3 recursively in parallel.
609
609
 
@@ -622,7 +622,7 @@ class S3(object, metaclass=type):
622
622
  S3Objects stored under the given prefixes.
623
623
  """
624
624
  ...
625
- def get_all(self, return_info: bool = False) -> typing.List[S3Object]:
625
+ def get_all(self, return_info: bool = False) -> typing.List[metaflow.plugins.datatools.s3.s3.S3Object]:
626
626
  """
627
627
  Get all objects under the prefix set in the `S3` constructor.
628
628
 
@@ -641,7 +641,7 @@ class S3(object, metaclass=type):
641
641
  S3Objects stored under the main prefix.
642
642
  """
643
643
  ...
644
- def put(self, key: typing.Union[str, S3PutObject], obj: typing.Union[io.RawIOBase, io.BufferedIOBase, str, bytes], overwrite: bool = True, content_type: typing.Optional[str] = None, metadata: typing.Optional[typing.Dict[str, str]] = None) -> str:
644
+ def put(self, key: typing.Union[str, metaflow.plugins.datatools.s3.s3.S3PutObject], obj: typing.Union[io.RawIOBase, io.BufferedIOBase, str, bytes], overwrite: bool = True, content_type: typing.Optional[str] = None, metadata: typing.Optional[typing.Dict[str, str]] = None) -> str:
645
645
  """
646
646
  Upload a single object to S3.
647
647
 
@@ -666,7 +666,7 @@ class S3(object, metaclass=type):
666
666
  URL of the object stored.
667
667
  """
668
668
  ...
669
- def put_many(self, key_objs: typing.List[typing.Union[typing.Tuple[str, typing.Union[io.RawIOBase, io.BufferedIOBase, str, bytes]], S3PutObject]], overwrite: bool = True) -> typing.List[typing.Tuple[str, str]]:
669
+ def put_many(self, key_objs: typing.List[typing.Union[typing.Tuple[str, typing.Union[io.RawIOBase, io.BufferedIOBase, str, bytes]], metaflow.plugins.datatools.s3.s3.S3PutObject]], overwrite: bool = True) -> typing.List[typing.Tuple[str, str]]:
670
670
  """
671
671
  Upload many objects to S3.
672
672
 
@@ -692,7 +692,7 @@ class S3(object, metaclass=type):
692
692
  List of `(key, url)` pairs corresponding to the objects uploaded.
693
693
  """
694
694
  ...
695
- def put_files(self, key_paths: typing.List[typing.Union[typing.Tuple[str, typing.Union[io.RawIOBase, io.BufferedIOBase, str, bytes]], S3PutObject]], overwrite: bool = True) -> typing.List[typing.Tuple[str, str]]:
695
+ def put_files(self, key_paths: typing.List[typing.Union[typing.Tuple[str, typing.Union[io.RawIOBase, io.BufferedIOBase, str, bytes]], metaflow.plugins.datatools.s3.s3.S3PutObject]], overwrite: bool = True) -> typing.List[typing.Tuple[str, str]]:
696
696
  """
697
697
  Upload many local files to S3.
698
698
 
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.15 #
4
- # Generated on 2025-05-30T13:33:59.775925 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:28.024557 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.737104 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:27.985946 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.733162 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:27.981945 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.733410 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:27.982210 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.730413 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:27.977695 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.731511 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:27.980181 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.729745 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:27.978955 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.756975 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:28.008119 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.732870 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:27.981054 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.764651 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:28.011945 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -11,8 +11,8 @@ import metaflow
11
11
  import typing
12
12
  if typing.TYPE_CHECKING:
13
13
  import abc
14
- import metaflow.exception
15
14
  import metaflow.plugins.secrets
15
+ import metaflow.exception
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.763585 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:28.010850 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.763450 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:28.010714 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.763808 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:28.011099 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.764333 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:28.011609 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.728047 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:27.975378 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.742852 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:27.991701 #
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.exception
13
13
 
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.15 #
4
- # Generated on 2025-05-30T13:33:59.746793 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:27.995640 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -106,9 +106,9 @@ class Kubernetes(object, metaclass=type):
106
106
  ...
107
107
  def launch_job(self, **kwargs):
108
108
  ...
109
- def create_jobset(self, flow_name, run_id, step_name, task_id, attempt, user, code_package_sha, code_package_url, code_package_ds, docker_image, docker_image_pull_policy, step_cli = None, service_account = None, secrets = None, node_selector = None, namespace = None, cpu = None, gpu = None, gpu_vendor = None, disk = None, memory = None, use_tmpfs = None, tmpfs_tempdir = None, tmpfs_size = None, tmpfs_path = None, run_time_limit = None, env = None, persistent_volume_claims = None, tolerations = None, labels = None, annotations = None, shared_memory = None, port = None, num_parallel = None, qos = None, security_context = None):
109
+ def create_jobset(self, flow_name, run_id, step_name, task_id, attempt, user, code_package_sha, code_package_url, code_package_ds, docker_image, docker_image_pull_policy, image_pull_secrets = None, step_cli = None, service_account = None, secrets = None, node_selector = None, namespace = None, cpu = None, gpu = None, gpu_vendor = None, disk = None, memory = None, use_tmpfs = None, tmpfs_tempdir = None, tmpfs_size = None, tmpfs_path = None, run_time_limit = None, env = None, persistent_volume_claims = None, tolerations = None, labels = None, annotations = None, shared_memory = None, port = None, num_parallel = None, qos = None, security_context = None):
110
110
  ...
111
- def create_job_object(self, flow_name, run_id, step_name, task_id, attempt, user, code_package_sha, code_package_url, code_package_ds, step_cli, docker_image, docker_image_pull_policy, service_account = None, secrets = None, node_selector = None, namespace = None, cpu = None, gpu = None, gpu_vendor = None, disk = None, memory = None, use_tmpfs = None, tmpfs_tempdir = None, tmpfs_size = None, tmpfs_path = None, run_time_limit = None, env = None, persistent_volume_claims = None, tolerations = None, labels = None, shared_memory = None, port = None, name_pattern = None, qos = None, annotations = None, security_context = None):
111
+ def create_job_object(self, flow_name, run_id, step_name, task_id, attempt, user, code_package_sha, code_package_url, code_package_ds, step_cli, docker_image, docker_image_pull_policy, image_pull_secrets = None, service_account = None, secrets = None, node_selector = None, namespace = None, cpu = None, gpu = None, gpu_vendor = None, disk = None, memory = None, use_tmpfs = None, tmpfs_tempdir = None, tmpfs_size = None, tmpfs_path = None, run_time_limit = None, env = None, persistent_volume_claims = None, tolerations = None, labels = None, shared_memory = None, port = None, name_pattern = None, qos = None, annotations = None, security_context = None):
112
112
  ...
113
113
  def create_k8sjob(self, job):
114
114
  ...
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.15.15 #
4
- # Generated on 2025-05-30T13:33:59.745788 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:27.994618 #
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.15.15 #
4
- # Generated on 2025-05-30T13:33:59.747687 #
3
+ # MF version: 2.15.17 #
4
+ # Generated on 2025-06-13T18:00:27.996498 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -39,6 +39,8 @@ KUBERNETES_GPU_VENDOR: str
39
39
 
40
40
  KUBERNETES_IMAGE_PULL_POLICY: None
41
41
 
42
+ KUBERNETES_IMAGE_PULL_SECRETS: str
43
+
42
44
  KUBERNETES_MEMORY: None
43
45
 
44
46
  KUBERNETES_LABELS: str
@@ -90,6 +92,10 @@ class KubernetesDecorator(metaflow.decorators.StepDecorator, metaclass=type):
90
92
  not, a default Docker image mapping to the current version of Python is used.
91
93
  image_pull_policy: str, default KUBERNETES_IMAGE_PULL_POLICY
92
94
  If given, the imagePullPolicy to be applied to the Docker image of the step.
95
+ image_pull_secrets: List[str], default []
96
+ The default is extracted from METAFLOW_KUBERNETES_IMAGE_PULL_SECRETS.
97
+ Kubernetes image pull secrets to use when pulling container images
98
+ in Kubernetes.
93
99
  service_account : str, default METAFLOW_KUBERNETES_SERVICE_ACCOUNT
94
100
  Kubernetes service account to use when launching pod in Kubernetes.
95
101
  secrets : List[str], optional, default None