ob-metaflow-stubs 6.0.3.123__py2.py3-none-any.whl → 6.0.3.126__py2.py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (181) hide show
  1. metaflow-stubs/__init__.pyi +253 -248
  2. metaflow-stubs/cards.pyi +2 -2
  3. metaflow-stubs/cli.pyi +9 -21
  4. metaflow-stubs/{plugins/package_cli.pyi → cli_components/__init__.pyi} +3 -3
  5. metaflow-stubs/cli_components/utils.pyi +35 -0
  6. metaflow-stubs/client/__init__.pyi +2 -2
  7. metaflow-stubs/client/core.pyi +4 -4
  8. metaflow-stubs/client/filecache.pyi +2 -2
  9. metaflow-stubs/events.pyi +3 -3
  10. metaflow-stubs/exception.pyi +2 -2
  11. metaflow-stubs/flowspec.pyi +10 -3
  12. metaflow-stubs/generated_for.txt +1 -1
  13. metaflow-stubs/includefile.pyi +18 -11
  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 +4 -2
  20. metaflow-stubs/metaflow_current.pyi +58 -58
  21. metaflow-stubs/mf_extensions/__init__.pyi +2 -2
  22. metaflow-stubs/mf_extensions/obcheckpoint/__init__.pyi +2 -2
  23. metaflow-stubs/mf_extensions/obcheckpoint/plugins/__init__.pyi +2 -2
  24. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/__init__.pyi +2 -2
  25. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/__init__.pyi +2 -2
  26. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/checkpoint_storage.pyi +4 -4
  27. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/constructors.pyi +2 -2
  28. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/core.pyi +3 -3
  29. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/exceptions.pyi +2 -2
  30. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/final_api.pyi +2 -2
  31. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/__init__.pyi +2 -2
  32. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/core.pyi +2 -2
  33. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/exceptions.pyi +2 -2
  34. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/task_utils.pyi +3 -3
  35. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/utils.pyi +2 -2
  36. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastructures.pyi +2 -2
  37. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/exceptions.pyi +2 -2
  38. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/__init__.pyi +2 -2
  39. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/flowspec_utils.pyi +2 -2
  40. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/general.pyi +2 -2
  41. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/identity_utils.pyi +3 -3
  42. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/tar_utils.pyi +2 -2
  43. metaflow-stubs/mf_extensions/outerbounds/__init__.pyi +2 -2
  44. metaflow-stubs/{mflog → mf_extensions/outerbounds/plugins}/__init__.pyi +2 -2
  45. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/__init__.pyi +6 -0
  46. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/snowflake.pyi +92 -0
  47. metaflow-stubs/mf_extensions/outerbounds/profilers/__init__.pyi +2 -2
  48. metaflow-stubs/mf_extensions/outerbounds/profilers/gpu.pyi +2 -2
  49. metaflow-stubs/mf_extensions/outerbounds/remote_config.pyi +2 -2
  50. metaflow-stubs/mf_extensions/outerbounds/toplevel/__init__.pyi +2 -2
  51. metaflow-stubs/mf_extensions/outerbounds/toplevel/global_aliases_for_metaflow_package.pyi +3 -2
  52. metaflow-stubs/multicore_utils.pyi +2 -2
  53. metaflow-stubs/parameters.pyi +33 -16
  54. metaflow-stubs/plugins/__init__.pyi +16 -15
  55. metaflow-stubs/plugins/airflow/__init__.pyi +2 -2
  56. metaflow-stubs/plugins/airflow/airflow_utils.pyi +2 -2
  57. metaflow-stubs/plugins/airflow/exception.pyi +2 -2
  58. metaflow-stubs/plugins/airflow/sensors/__init__.pyi +2 -2
  59. metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi +2 -2
  60. metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi +2 -2
  61. metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi +2 -2
  62. metaflow-stubs/plugins/argo/__init__.pyi +2 -5
  63. metaflow-stubs/plugins/argo/argo_client.pyi +2 -2
  64. metaflow-stubs/plugins/argo/argo_events.pyi +2 -2
  65. metaflow-stubs/plugins/argo/argo_workflows.pyi +4 -3
  66. metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +3 -3
  67. metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +3 -3
  68. metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi +31 -3
  69. metaflow-stubs/plugins/aws/__init__.pyi +2 -2
  70. metaflow-stubs/plugins/aws/aws_client.pyi +2 -2
  71. metaflow-stubs/plugins/aws/aws_utils.pyi +2 -2
  72. metaflow-stubs/plugins/aws/batch/__init__.pyi +2 -2
  73. metaflow-stubs/plugins/aws/batch/batch.pyi +2 -2
  74. metaflow-stubs/plugins/aws/batch/batch_client.pyi +2 -2
  75. metaflow-stubs/plugins/aws/batch/batch_decorator.pyi +3 -3
  76. metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi +2 -2
  77. metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi +5 -5
  78. metaflow-stubs/plugins/aws/step_functions/__init__.pyi +2 -3
  79. metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi +2 -2
  80. metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi +2 -2
  81. metaflow-stubs/plugins/aws/step_functions/step_functions.pyi +3 -2
  82. metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi +2 -2
  83. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi +3 -3
  84. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi +3 -3
  85. metaflow-stubs/plugins/azure/__init__.pyi +2 -2
  86. metaflow-stubs/plugins/azure/azure_credential.pyi +2 -2
  87. metaflow-stubs/plugins/azure/azure_exceptions.pyi +2 -2
  88. metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi +5 -5
  89. metaflow-stubs/plugins/azure/azure_utils.pyi +2 -2
  90. metaflow-stubs/plugins/azure/blob_service_client_factory.pyi +2 -2
  91. metaflow-stubs/plugins/azure/includefile_support.pyi +2 -2
  92. metaflow-stubs/plugins/cards/__init__.pyi +2 -3
  93. metaflow-stubs/plugins/cards/card_client.pyi +3 -3
  94. metaflow-stubs/plugins/cards/card_creator.pyi +2 -2
  95. metaflow-stubs/plugins/cards/card_datastore.pyi +2 -2
  96. metaflow-stubs/plugins/cards/card_decorator.pyi +2 -2
  97. metaflow-stubs/plugins/cards/card_modules/__init__.pyi +2 -2
  98. metaflow-stubs/plugins/cards/card_modules/basic.pyi +3 -3
  99. metaflow-stubs/plugins/cards/card_modules/card.pyi +2 -2
  100. metaflow-stubs/plugins/cards/card_modules/components.pyi +4 -4
  101. metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi +2 -2
  102. metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi +2 -2
  103. metaflow-stubs/plugins/cards/card_modules/test_cards.pyi +2 -2
  104. metaflow-stubs/plugins/cards/card_resolver.pyi +2 -2
  105. metaflow-stubs/plugins/cards/component_serializer.pyi +2 -2
  106. metaflow-stubs/plugins/cards/exception.pyi +2 -2
  107. metaflow-stubs/plugins/catch_decorator.pyi +2 -2
  108. metaflow-stubs/plugins/datatools/__init__.pyi +2 -2
  109. metaflow-stubs/plugins/datatools/local.pyi +2 -2
  110. metaflow-stubs/plugins/datatools/s3/__init__.pyi +2 -2
  111. metaflow-stubs/plugins/datatools/s3/s3.pyi +3 -3
  112. metaflow-stubs/plugins/datatools/s3/s3tail.pyi +2 -2
  113. metaflow-stubs/plugins/datatools/s3/s3util.pyi +2 -2
  114. metaflow-stubs/plugins/debug_logger.pyi +2 -2
  115. metaflow-stubs/plugins/debug_monitor.pyi +2 -2
  116. metaflow-stubs/plugins/environment_decorator.pyi +2 -2
  117. metaflow-stubs/plugins/events_decorator.pyi +2 -2
  118. metaflow-stubs/plugins/frameworks/__init__.pyi +2 -2
  119. metaflow-stubs/plugins/frameworks/pytorch.pyi +2 -2
  120. metaflow-stubs/plugins/gcp/__init__.pyi +2 -2
  121. metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi +5 -5
  122. metaflow-stubs/plugins/gcp/gs_exceptions.pyi +2 -2
  123. metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi +2 -2
  124. metaflow-stubs/plugins/gcp/gs_utils.pyi +2 -2
  125. metaflow-stubs/plugins/gcp/includefile_support.pyi +2 -2
  126. metaflow-stubs/plugins/kubernetes/__init__.pyi +2 -5
  127. metaflow-stubs/plugins/kubernetes/kube_utils.pyi +2 -2
  128. metaflow-stubs/plugins/kubernetes/kubernetes.pyi +2 -2
  129. metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +2 -2
  130. metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +3 -3
  131. metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi +2 -2
  132. metaflow-stubs/plugins/parallel_decorator.pyi +2 -2
  133. metaflow-stubs/plugins/perimeters.pyi +2 -2
  134. metaflow-stubs/plugins/project_decorator.pyi +2 -2
  135. metaflow-stubs/plugins/pypi/__init__.pyi +2 -2
  136. metaflow-stubs/plugins/pypi/conda_decorator.pyi +6 -2
  137. metaflow-stubs/plugins/pypi/conda_environment.pyi +5 -5
  138. metaflow-stubs/plugins/pypi/pypi_decorator.pyi +2 -2
  139. metaflow-stubs/plugins/pypi/pypi_environment.pyi +2 -2
  140. metaflow-stubs/plugins/pypi/utils.pyi +2 -2
  141. metaflow-stubs/plugins/resources_decorator.pyi +2 -2
  142. metaflow-stubs/plugins/retry_decorator.pyi +2 -2
  143. metaflow-stubs/plugins/secrets/__init__.pyi +3 -3
  144. metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +4 -4
  145. metaflow-stubs/plugins/secrets/secrets_decorator.pyi +2 -2
  146. metaflow-stubs/plugins/snowflake/__init__.pyi +14 -0
  147. metaflow-stubs/plugins/storage_executor.pyi +2 -2
  148. metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +3 -3
  149. metaflow-stubs/plugins/timeout_decorator.pyi +3 -3
  150. metaflow-stubs/profilers/__init__.pyi +2 -2
  151. metaflow-stubs/pylint_wrapper.pyi +2 -2
  152. metaflow-stubs/runner/__init__.pyi +2 -2
  153. metaflow-stubs/runner/deployer.pyi +30 -30
  154. metaflow-stubs/runner/deployer_impl.pyi +3 -3
  155. metaflow-stubs/runner/metaflow_runner.pyi +4 -4
  156. metaflow-stubs/runner/nbdeploy.pyi +4 -2
  157. metaflow-stubs/runner/nbrun.pyi +3 -3
  158. metaflow-stubs/runner/subprocess_manager.pyi +5 -2
  159. metaflow-stubs/runner/utils.pyi +8 -8
  160. metaflow-stubs/system/__init__.pyi +2 -2
  161. metaflow-stubs/system/system_logger.pyi +2 -2
  162. metaflow-stubs/system/system_monitor.pyi +2 -2
  163. metaflow-stubs/tagging_util.pyi +2 -2
  164. metaflow-stubs/tuple_util.pyi +2 -2
  165. metaflow-stubs/{plugins/aws/step_functions/production_token.pyi → user_configs/__init__.pyi} +5 -11
  166. metaflow-stubs/user_configs/config_decorators.pyi +253 -0
  167. metaflow-stubs/user_configs/config_options.pyi +82 -0
  168. metaflow-stubs/user_configs/config_parameters.pyi +217 -0
  169. {ob_metaflow_stubs-6.0.3.123.dist-info → ob_metaflow_stubs-6.0.3.126.dist-info}/METADATA +1 -1
  170. ob_metaflow_stubs-6.0.3.126.dist-info/RECORD +173 -0
  171. metaflow-stubs/clone_util.pyi +0 -14
  172. metaflow-stubs/mflog/mflog.pyi +0 -69
  173. metaflow-stubs/plugins/argo/argo_workflows_cli.pyi +0 -103
  174. metaflow-stubs/plugins/cards/card_cli.pyi +0 -146
  175. metaflow-stubs/plugins/kubernetes/kubernetes_cli.pyi +0 -33
  176. metaflow-stubs/plugins/logs_cli.pyi +0 -46
  177. metaflow-stubs/plugins/tag_cli.pyi +0 -19
  178. metaflow-stubs/procpoll.pyi +0 -53
  179. ob_metaflow_stubs-6.0.3.123.dist-info/RECORD +0 -174
  180. {ob_metaflow_stubs-6.0.3.123.dist-info → ob_metaflow_stubs-6.0.3.126.dist-info}/WHEEL +0 -0
  181. {ob_metaflow_stubs-6.0.3.123.dist-info → ob_metaflow_stubs-6.0.3.126.dist-info}/top_level.txt +0 -0
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.640308 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.826477 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -10,8 +10,8 @@ import typing
10
10
  if typing.TYPE_CHECKING:
11
11
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.checkpoint_storage
12
12
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
13
- import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.core
14
13
  import metaflow
14
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.core
15
15
 
16
16
  from ......metaflow_current import current as current
17
17
  from ......exception import MetaflowException as MetaflowException
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.610062 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.803826 #
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.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.586643 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.780827 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.610669 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.804429 #
5
5
  ######################################################################################################
6
6
 
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.641506 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.827688 #
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.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.661920 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.880630 #
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.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.610720 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.804479 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,8 +9,8 @@ from __future__ import annotations
9
9
  import typing
10
10
  import metaflow
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow
13
12
  import metaflow.exception
13
+ import metaflow
14
14
 
15
15
  from ......exception import MetaflowException as MetaflowException
16
16
  from .core import resolve_root as resolve_root
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.662105 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.880816 #
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.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.587565 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.781899 #
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.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.610383 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.804146 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.650198 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.863855 #
5
5
  ######################################################################################################
6
6
 
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.650255 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.863908 #
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.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.652030 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.865939 #
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.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.661647 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.880354 #
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
13
12
  import metaflow.exception
13
+ import metaflow
14
14
 
15
15
  from ......exception import MetaflowException as MetaflowException
16
16
 
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.651658 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.865543 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.577996 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.772056 #
5
5
  ######################################################################################################
6
6
 
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.611991 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.782580 #
5
5
  ######################################################################################################
6
6
 
@@ -0,0 +1,6 @@
1
+ ######################################################################################################
2
+ # Auto-generated Metaflow stub file #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.782644 #
5
+ ######################################################################################################
6
+
@@ -0,0 +1,92 @@
1
+ ######################################################################################################
2
+ # Auto-generated Metaflow stub file #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.782690 #
5
+ ######################################################################################################
6
+
7
+ from __future__ import annotations
8
+
9
+ import typing
10
+ if typing.TYPE_CHECKING:
11
+ import metaflow.mf_extensions.outerbounds.plugins.snowflake.snowflake
12
+
13
+ from ...remote_config import init_config as init_config
14
+
15
+ SERVICE_URL: None
16
+
17
+ class OuterboundsSnowflakeConnectorException(Exception, metaclass=type):
18
+ ...
19
+
20
+ class OuterboundsSnowflakeIntegrationSpecApiResponse(object, metaclass=type):
21
+ def __init__(self, response):
22
+ ...
23
+ @property
24
+ def account(self):
25
+ ...
26
+ @property
27
+ def user(self):
28
+ ...
29
+ @property
30
+ def default_role(self):
31
+ ...
32
+ @property
33
+ def warehouse(self):
34
+ ...
35
+ @property
36
+ def database(self):
37
+ ...
38
+ ...
39
+
40
+ def get_snowflake_token(user: str = '', role: str = '', integration: str = '') -> str:
41
+ """
42
+ Uses the Outerbounds source token to request for a snowflake compatible OIDC
43
+ token. This token can then be used to connect to snowflake.
44
+ user: str
45
+ The user the token will be minted for
46
+ role: str
47
+ The role to which the token will be scoped to
48
+ integration: str
49
+ The name of the snowflake integration to use. If not set, an existing integration will be used provided that only one exists per perimeter. If integration is not set and more than one exists, then we raise an exception.
50
+ """
51
+ ...
52
+
53
+ def connect(user: str = '', role: str = '', integration: str = '', **kwargs):
54
+ """
55
+ Connect to snowflake using the token minted by Outerbounds
56
+ user: str
57
+ The user name used to authenticate with snowflake
58
+ role: str
59
+ The role to request when connect with snowflake
60
+ integration: str
61
+ The name of the snowflake integration to use. If not set, an existing integration will be used provided that only one exists in the current perimeter. If integration is not set and more than one exists in the current perimeter, then we raise an exception.
62
+ kwargs: dict
63
+ Additional arguments to pass to the python snowflake connector
64
+ """
65
+ ...
66
+
67
+ class Snowflake(object, metaclass=type):
68
+ def __init__(self, user: str = '', role: str = '', integration: str = '', **kwargs):
69
+ ...
70
+ def __enter__(self):
71
+ ...
72
+ def __exit__(self, exception_type, exception_value, traceback):
73
+ ...
74
+ def close(self):
75
+ ...
76
+ ...
77
+
78
+ class SnowflakeIntegrationProvisioner(object, metaclass=type):
79
+ def __init__(self, integration_name: str):
80
+ ...
81
+ def get_snowflake_integration_spec(self) -> OuterboundsSnowflakeIntegrationSpecApiResponse:
82
+ ...
83
+ def get_integration_name(self) -> str:
84
+ ...
85
+ def get_perimeter(self) -> str:
86
+ ...
87
+ def get_snowflake_token_url(self) -> str:
88
+ ...
89
+ def get_service_auth_header(self) -> str:
90
+ ...
91
+ ...
92
+
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.590512 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.785119 #
5
5
  ######################################################################################################
6
6
 
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.590568 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.785192 #
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.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.641892 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.804800 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.578054 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.772115 #
5
5
  ######################################################################################################
6
6
 
@@ -1,13 +1,14 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.578100 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.772161 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
9
 
10
10
  from .... import profilers as profilers
11
+ from ..plugins.snowflake.snowflake import Snowflake as Snowflake
11
12
 
12
13
  def get_aws_client(module, with_error = False, role_arn = None, session_vars = None, client_params = None):
13
14
  ...
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.571899 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.764026 #
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.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.573142 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.765312 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,23 +9,25 @@ from __future__ import annotations
9
9
  import typing
10
10
  import metaflow
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow._vendor.click.types
13
12
  import typing
14
13
  import metaflow.parameters
14
+ import metaflow._vendor.click.types
15
15
 
16
16
  from ._vendor import click as click
17
17
  from .exception import ParameterFieldFailed as ParameterFieldFailed
18
18
  from .exception import ParameterFieldTypeMismatch as ParameterFieldTypeMismatch
19
19
  from .exception import MetaflowException as MetaflowException
20
20
 
21
+ TYPE_CHECKING: bool
22
+
21
23
  class ParameterContext(tuple, metaclass=type):
22
24
  """
23
- ParameterContext(flow_name, user_name, parameter_name, logger, ds_type)
25
+ ParameterContext(flow_name, user_name, parameter_name, logger, ds_type, configs)
24
26
  """
25
27
  @staticmethod
26
- def __new__(_cls, flow_name: str, user_name: str, parameter_name: str, logger: typing.Callable[..., None], ds_type: str):
28
+ def __new__(_cls, flow_name: str, user_name: str, parameter_name: str, logger: typing.Callable[..., None], ds_type: str, configs: typing.Optional["ConfigValue"]):
27
29
  """
28
- Create new instance of ParameterContext(flow_name, user_name, parameter_name, logger, ds_type)
30
+ Create new instance of ParameterContext(flow_name, user_name, parameter_name, logger, ds_type, configs)
29
31
  """
30
32
  ...
31
33
  def __repr__(self):
@@ -38,7 +40,7 @@ class ParameterContext(tuple, metaclass=type):
38
40
  Return self as a plain tuple. Used by copy and pickle.
39
41
  """
40
42
  ...
41
- def __init__(self, flow_name: str, user_name: str, parameter_name: str, logger: typing.Callable[..., None], ds_type: str):
43
+ def __init__(self, flow_name: str, user_name: str, parameter_name: str, logger: typing.Callable[..., None], ds_type: str, configs: typing.Optional["ConfigValue"]):
42
44
  ...
43
45
  ...
44
46
 
@@ -52,6 +54,13 @@ def flow_context(flow_cls):
52
54
 
53
55
  context_proto: None
54
56
 
57
+ def replace_flow_context(flow_cls):
58
+ """
59
+ Replace the current flow context with a new flow class. This is used
60
+ when we change the current flow class after having run user configuration functions
61
+ """
62
+ ...
63
+
55
64
  class JSONTypeClass(metaflow._vendor.click.types.ParamType, metaclass=type):
56
65
  def convert(self, value, param, ctx):
57
66
  ...
@@ -86,7 +95,7 @@ class DeployTimeField(object, metaclass=type):
86
95
  def deploy_time_eval(value):
87
96
  ...
88
97
 
89
- def set_parameter_context(flow_name, echo, datastore):
98
+ def set_parameter_context(flow_name, echo, datastore, configs):
90
99
  ...
91
100
 
92
101
  class DelayedEvaluationParameter(object, metaclass=type):
@@ -135,7 +144,11 @@ class Parameter(object, metaclass=type):
135
144
  ----------
136
145
  name : str
137
146
  User-visible parameter name.
138
- default : str or float or int or bool or `JSONType` or a function.
147
+ default : Union[str, float, int, bool, Dict[str, Any],
148
+ Callable[
149
+ [ParameterContext], Union[str, float, int, bool, Dict[str, Any]]
150
+ ],
151
+ ], optional, default None
139
152
  Default value for the parameter. Use a special `JSONType` class to
140
153
  indicate that the value must be a valid JSON object. A function
141
154
  implies that the parameter corresponds to a *deploy-time parameter*.
@@ -144,15 +157,19 @@ class Parameter(object, metaclass=type):
144
157
  If `default` is not specified, define the parameter type. Specify
145
158
  one of `str`, `float`, `int`, `bool`, or `JSONType`. If None, defaults
146
159
  to the type of `default` or `str` if none specified.
147
- help : str, optional
160
+ help : str, optional, default None
148
161
  Help text to show in `run --help`.
149
- required : bool, default False
150
- Require that the user specified a value for the parameter.
151
- `required=True` implies that the `default` is not used.
152
- show_default : bool, default True
153
- If True, show the default value in the help text.
162
+ required : bool, optional, default None
163
+ Require that the user specified a value for the parameter. Note that if
164
+ a default is provide, the required flag is ignored.
165
+ A value of None is equivalent to False.
166
+ show_default : bool, optional, default None
167
+ If True, show the default value in the help text. A value of None is equivalent
168
+ to True.
154
169
  """
155
- def __init__(self, name: str, default: typing.Union[str, float, int, bool, typing.Dict[str, typing.Any], typing.Callable[[], typing.Union[str, float, int, bool, typing.Dict[str, typing.Any]]], None] = None, type: typing.Union[typing.Type[str], typing.Type[float], typing.Type[int], typing.Type[bool], JSONTypeClass, None] = None, help: typing.Optional[str] = None, required: bool = False, show_default: bool = True, **kwargs: typing.Dict[str, typing.Any]):
170
+ def __init__(self, name: str, default: typing.Union[str, float, int, bool, typing.Dict[str, typing.Any], typing.Callable[[ParameterContext], typing.Union[str, float, int, bool, typing.Dict[str, typing.Any]]], None] = None, type: typing.Union[typing.Type[str], typing.Type[float], typing.Type[int], typing.Type[bool], JSONTypeClass, None] = None, help: typing.Optional[str] = None, required: typing.Optional[bool] = None, show_default: typing.Optional[bool] = None, **kwargs: typing.Dict[str, typing.Any]):
171
+ ...
172
+ def init(self):
156
173
  ...
157
174
  def __repr__(self):
158
175
  ...
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.577538 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.771637 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,25 +9,25 @@ from __future__ import annotations
9
9
 
10
10
  from . import test_unbounded_foreach_decorator as test_unbounded_foreach_decorator
11
11
  from .test_unbounded_foreach_decorator import InternalTestUnboundedForeachInput as InternalTestUnboundedForeachInput
12
- from . import kubernetes as kubernetes
13
- from . import resources_decorator as resources_decorator
14
- from . import timeout_decorator as timeout_decorator
12
+ from . import pypi as pypi
15
13
  from . import aws as aws
14
+ from . import frameworks as frameworks
15
+ from . import parallel_decorator as parallel_decorator
16
+ from . import secrets as secrets
17
+ from . import argo as argo
18
+ from . import timeout_decorator as timeout_decorator
16
19
  from . import datatools as datatools
17
20
  from . import gcp as gcp
18
21
  from . import storage_executor as storage_executor
19
22
  from . import cards as cards
20
- from . import pypi as pypi
21
- from . import environment_decorator as environment_decorator
22
- from . import argo as argo
23
- from . import parallel_decorator as parallel_decorator
24
23
  from . import catch_decorator as catch_decorator
25
- from . import secrets as secrets
24
+ from . import resources_decorator as resources_decorator
25
+ from . import kubernetes as kubernetes
26
+ from . import environment_decorator as environment_decorator
26
27
  from . import retry_decorator as retry_decorator
27
- from . import frameworks as frameworks
28
- from . import events_decorator as events_decorator
29
28
  from . import airflow as airflow
30
29
  from . import project_decorator as project_decorator
30
+ from . import events_decorator as events_decorator
31
31
  from . import azure as azure
32
32
  from . import debug_logger as debug_logger
33
33
  from . import debug_monitor as debug_monitor
@@ -45,9 +45,7 @@ from .cards.card_modules.test_cards import TestPathSpecCard as TestPathSpecCard
45
45
  from .cards.card_modules.test_cards import TestTimeoutCard as TestTimeoutCard
46
46
  from .cards.card_modules.test_cards import TestRefreshCard as TestRefreshCard
47
47
  from .cards.card_modules.test_cards import TestRefreshComponentCard as TestRefreshComponentCard
48
- from . import logs_cli as logs_cli
49
- from . import tag_cli as tag_cli
50
- from . import package_cli as package_cli
48
+ from . import snowflake as snowflake
51
49
  from . import perimeters as perimeters
52
50
 
53
51
  CLIS_DESC: list
@@ -85,6 +83,9 @@ DEPLOYER_IMPL_PROVIDERS_DESC: list
85
83
  def get_plugin_cli():
86
84
  ...
87
85
 
86
+ def get_plugin_cli_path():
87
+ ...
88
+
88
89
  STEP_DECORATORS: list
89
90
 
90
91
  FLOW_DECORATORS: list
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.605859 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.799972 #
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.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.637282 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.823334 #
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.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.636322 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.822451 #
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.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.637536 #
3
+ # MF version: 2.12.39.1+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-17T01:10:55.823586 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations