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

Sign up to get free protection for your applications and to get access to all the features.
Files changed (178) hide show
  1. metaflow-stubs/__init__.pyi +211 -210
  2. metaflow-stubs/cards.pyi +2 -2
  3. metaflow-stubs/cli.pyi +2 -2
  4. metaflow-stubs/client/__init__.pyi +2 -2
  5. metaflow-stubs/client/core.pyi +5 -5
  6. metaflow-stubs/client/filecache.pyi +3 -3
  7. metaflow-stubs/clone_util.pyi +2 -2
  8. metaflow-stubs/events.pyi +2 -2
  9. metaflow-stubs/exception.pyi +2 -2
  10. metaflow-stubs/flowspec.pyi +6 -6
  11. metaflow-stubs/generated_for.txt +1 -1
  12. metaflow-stubs/includefile.pyi +4 -4
  13. metaflow-stubs/info_file.pyi +2 -2
  14. metaflow-stubs/metadata_provider/__init__.pyi +2 -2
  15. metaflow-stubs/metadata_provider/heartbeat.pyi +2 -2
  16. metaflow-stubs/metadata_provider/metadata.pyi +2 -2
  17. metaflow-stubs/metadata_provider/util.pyi +2 -2
  18. metaflow-stubs/metaflow_config.pyi +2 -2
  19. metaflow-stubs/metaflow_current.pyi +135 -135
  20. metaflow-stubs/mf_extensions/__init__.pyi +2 -2
  21. metaflow-stubs/mf_extensions/obcheckpoint/__init__.pyi +2 -2
  22. metaflow-stubs/mf_extensions/obcheckpoint/plugins/__init__.pyi +2 -2
  23. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/__init__.pyi +2 -2
  24. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/__init__.pyi +2 -2
  25. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/checkpoint_storage.pyi +4 -4
  26. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/constructors.pyi +2 -2
  27. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/core.pyi +4 -4
  28. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/exceptions.pyi +2 -2
  29. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/final_api.pyi +3 -3
  30. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/__init__.pyi +2 -2
  31. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/core.pyi +3 -3
  32. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/exceptions.pyi +2 -2
  33. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/task_utils.pyi +4 -4
  34. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/utils.pyi +2 -2
  35. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastructures.pyi +3 -3
  36. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/exceptions.pyi +2 -2
  37. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/__init__.pyi +2 -2
  38. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/flowspec_utils.pyi +2 -2
  39. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/general.pyi +2 -2
  40. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/identity_utils.pyi +3 -3
  41. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/tar_utils.pyi +3 -3
  42. metaflow-stubs/mf_extensions/outerbounds/__init__.pyi +2 -2
  43. metaflow-stubs/mf_extensions/outerbounds/plugins/__init__.pyi +6 -0
  44. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/__init__.pyi +6 -0
  45. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/snowflake.pyi +92 -0
  46. metaflow-stubs/mf_extensions/outerbounds/profilers/__init__.pyi +2 -2
  47. metaflow-stubs/mf_extensions/outerbounds/profilers/gpu.pyi +2 -2
  48. metaflow-stubs/mf_extensions/outerbounds/remote_config.pyi +2 -2
  49. metaflow-stubs/mf_extensions/outerbounds/toplevel/__init__.pyi +2 -2
  50. metaflow-stubs/mf_extensions/outerbounds/toplevel/global_aliases_for_metaflow_package.pyi +3 -2
  51. metaflow-stubs/mflog/__init__.pyi +2 -2
  52. metaflow-stubs/mflog/mflog.pyi +2 -2
  53. metaflow-stubs/multicore_utils.pyi +2 -2
  54. metaflow-stubs/parameters.pyi +4 -4
  55. metaflow-stubs/plugins/__init__.pyi +11 -10
  56. metaflow-stubs/plugins/airflow/__init__.pyi +2 -2
  57. metaflow-stubs/plugins/airflow/airflow_utils.pyi +2 -2
  58. metaflow-stubs/plugins/airflow/exception.pyi +2 -2
  59. metaflow-stubs/plugins/airflow/sensors/__init__.pyi +2 -2
  60. metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi +2 -2
  61. metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi +2 -2
  62. metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi +2 -2
  63. metaflow-stubs/plugins/argo/__init__.pyi +2 -2
  64. metaflow-stubs/plugins/argo/argo_client.pyi +2 -2
  65. metaflow-stubs/plugins/argo/argo_events.pyi +2 -2
  66. metaflow-stubs/plugins/argo/argo_workflows.pyi +3 -3
  67. metaflow-stubs/plugins/argo/argo_workflows_cli.pyi +2 -2
  68. metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +3 -3
  69. metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +4 -4
  70. metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi +4 -4
  71. metaflow-stubs/plugins/aws/__init__.pyi +2 -2
  72. metaflow-stubs/plugins/aws/aws_client.pyi +2 -2
  73. metaflow-stubs/plugins/aws/aws_utils.pyi +2 -2
  74. metaflow-stubs/plugins/aws/batch/__init__.pyi +2 -2
  75. metaflow-stubs/plugins/aws/batch/batch.pyi +2 -2
  76. metaflow-stubs/plugins/aws/batch/batch_client.pyi +2 -2
  77. metaflow-stubs/plugins/aws/batch/batch_decorator.pyi +2 -2
  78. metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi +2 -2
  79. metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi +3 -3
  80. metaflow-stubs/plugins/aws/step_functions/__init__.pyi +2 -2
  81. metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi +2 -2
  82. metaflow-stubs/plugins/aws/step_functions/production_token.pyi +2 -2
  83. metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi +2 -2
  84. metaflow-stubs/plugins/aws/step_functions/step_functions.pyi +2 -2
  85. metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi +2 -2
  86. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi +4 -4
  87. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi +4 -4
  88. metaflow-stubs/plugins/azure/__init__.pyi +2 -2
  89. metaflow-stubs/plugins/azure/azure_credential.pyi +2 -2
  90. metaflow-stubs/plugins/azure/azure_exceptions.pyi +2 -2
  91. metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi +3 -3
  92. metaflow-stubs/plugins/azure/azure_utils.pyi +2 -2
  93. metaflow-stubs/plugins/azure/blob_service_client_factory.pyi +2 -2
  94. metaflow-stubs/plugins/azure/includefile_support.pyi +2 -2
  95. metaflow-stubs/plugins/cards/__init__.pyi +2 -2
  96. metaflow-stubs/plugins/cards/card_cli.pyi +2 -2
  97. metaflow-stubs/plugins/cards/card_client.pyi +2 -2
  98. metaflow-stubs/plugins/cards/card_creator.pyi +2 -2
  99. metaflow-stubs/plugins/cards/card_datastore.pyi +2 -2
  100. metaflow-stubs/plugins/cards/card_decorator.pyi +2 -2
  101. metaflow-stubs/plugins/cards/card_modules/__init__.pyi +2 -2
  102. metaflow-stubs/plugins/cards/card_modules/basic.pyi +3 -3
  103. metaflow-stubs/plugins/cards/card_modules/card.pyi +2 -2
  104. metaflow-stubs/plugins/cards/card_modules/components.pyi +4 -4
  105. metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi +2 -2
  106. metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi +2 -2
  107. metaflow-stubs/plugins/cards/card_modules/test_cards.pyi +2 -2
  108. metaflow-stubs/plugins/cards/card_resolver.pyi +2 -2
  109. metaflow-stubs/plugins/cards/component_serializer.pyi +2 -2
  110. metaflow-stubs/plugins/cards/exception.pyi +2 -2
  111. metaflow-stubs/plugins/catch_decorator.pyi +3 -3
  112. metaflow-stubs/plugins/datatools/__init__.pyi +2 -2
  113. metaflow-stubs/plugins/datatools/local.pyi +2 -2
  114. metaflow-stubs/plugins/datatools/s3/__init__.pyi +2 -2
  115. metaflow-stubs/plugins/datatools/s3/s3.pyi +5 -5
  116. metaflow-stubs/plugins/datatools/s3/s3tail.pyi +2 -2
  117. metaflow-stubs/plugins/datatools/s3/s3util.pyi +2 -2
  118. metaflow-stubs/plugins/debug_logger.pyi +2 -2
  119. metaflow-stubs/plugins/debug_monitor.pyi +2 -2
  120. metaflow-stubs/plugins/environment_decorator.pyi +2 -2
  121. metaflow-stubs/plugins/events_decorator.pyi +2 -2
  122. metaflow-stubs/plugins/frameworks/__init__.pyi +2 -2
  123. metaflow-stubs/plugins/frameworks/pytorch.pyi +2 -2
  124. metaflow-stubs/plugins/gcp/__init__.pyi +2 -2
  125. metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi +3 -3
  126. metaflow-stubs/plugins/gcp/gs_exceptions.pyi +2 -2
  127. metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi +2 -2
  128. metaflow-stubs/plugins/gcp/gs_utils.pyi +2 -2
  129. metaflow-stubs/plugins/gcp/includefile_support.pyi +2 -2
  130. metaflow-stubs/plugins/kubernetes/__init__.pyi +2 -2
  131. metaflow-stubs/plugins/kubernetes/kube_utils.pyi +2 -2
  132. metaflow-stubs/plugins/kubernetes/kubernetes.pyi +3 -3
  133. metaflow-stubs/plugins/kubernetes/kubernetes_cli.pyi +2 -2
  134. metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +2 -2
  135. metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +2 -2
  136. metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi +2 -2
  137. metaflow-stubs/plugins/logs_cli.pyi +2 -2
  138. metaflow-stubs/plugins/package_cli.pyi +2 -2
  139. metaflow-stubs/plugins/parallel_decorator.pyi +2 -2
  140. metaflow-stubs/plugins/perimeters.pyi +2 -2
  141. metaflow-stubs/plugins/project_decorator.pyi +2 -2
  142. metaflow-stubs/plugins/pypi/__init__.pyi +3 -3
  143. metaflow-stubs/plugins/pypi/conda_decorator.pyi +2 -2
  144. metaflow-stubs/plugins/pypi/conda_environment.pyi +5 -5
  145. metaflow-stubs/plugins/pypi/pypi_decorator.pyi +2 -2
  146. metaflow-stubs/plugins/pypi/pypi_environment.pyi +2 -2
  147. metaflow-stubs/plugins/pypi/utils.pyi +2 -2
  148. metaflow-stubs/plugins/resources_decorator.pyi +2 -2
  149. metaflow-stubs/plugins/retry_decorator.pyi +2 -2
  150. metaflow-stubs/plugins/secrets/__init__.pyi +2 -2
  151. metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +2 -2
  152. metaflow-stubs/plugins/secrets/secrets_decorator.pyi +2 -2
  153. metaflow-stubs/plugins/snowflake/__init__.pyi +14 -0
  154. metaflow-stubs/plugins/storage_executor.pyi +2 -2
  155. metaflow-stubs/plugins/tag_cli.pyi +2 -2
  156. metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +3 -3
  157. metaflow-stubs/plugins/timeout_decorator.pyi +3 -3
  158. metaflow-stubs/procpoll.pyi +2 -2
  159. metaflow-stubs/profilers/__init__.pyi +2 -2
  160. metaflow-stubs/pylint_wrapper.pyi +2 -2
  161. metaflow-stubs/runner/__init__.pyi +2 -2
  162. metaflow-stubs/runner/deployer.pyi +5 -5
  163. metaflow-stubs/runner/deployer_impl.pyi +3 -3
  164. metaflow-stubs/runner/metaflow_runner.pyi +3 -3
  165. metaflow-stubs/runner/nbdeploy.pyi +2 -2
  166. metaflow-stubs/runner/nbrun.pyi +2 -2
  167. metaflow-stubs/runner/subprocess_manager.pyi +2 -2
  168. metaflow-stubs/runner/utils.pyi +3 -3
  169. metaflow-stubs/system/__init__.pyi +2 -2
  170. metaflow-stubs/system/system_logger.pyi +3 -3
  171. metaflow-stubs/system/system_monitor.pyi +2 -2
  172. metaflow-stubs/tagging_util.pyi +2 -2
  173. metaflow-stubs/tuple_util.pyi +2 -2
  174. {ob_metaflow_stubs-6.0.3.123.dist-info → ob_metaflow_stubs-6.0.3.125.dist-info}/METADATA +1 -1
  175. ob_metaflow_stubs-6.0.3.125.dist-info/RECORD +178 -0
  176. ob_metaflow_stubs-6.0.3.123.dist-info/RECORD +0 -174
  177. {ob_metaflow_stubs-6.0.3.123.dist-info → ob_metaflow_stubs-6.0.3.125.dist-info}/WHEEL +0 -0
  178. {ob_metaflow_stubs-6.0.3.123.dist-info → ob_metaflow_stubs-6.0.3.125.dist-info}/top_level.txt +0 -0
@@ -1,17 +1,17 @@
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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.164021 #
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.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.checkpoint_storage
12
- import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
13
11
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.core
14
12
  import metaflow
13
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
14
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.checkpoint_storage
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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.132990 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,15 +1,15 @@
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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.108101 #
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.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
12
11
  import metaflow
12
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
13
13
 
14
14
  from ..datastructures import CheckpointArtifact as CheckpointArtifact
15
15
  from .constructors import load_checkpoint as load_checkpoint
@@ -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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.133601 #
5
5
  ######################################################################################################
6
6
 
@@ -1,15 +1,15 @@
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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.165227 #
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.datastore.datastore_storage
12
11
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastore.core
12
+ import metaflow.datastore.datastore_storage
13
13
 
14
14
  from ......exception import MetaflowException as MetaflowException
15
15
  from ..utils.tar_utils import create_tarball_on_disk as create_tarball_on_disk
@@ -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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.185709 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,16 +1,16 @@
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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.133656 #
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 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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.185901 #
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.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.587565 #
3
+ # MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.109531 #
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.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
13
13
 
@@ -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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.133313 #
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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.173610 #
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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.173666 #
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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.175565 #
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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.185439 #
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,13 +1,13 @@
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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.175161 #
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,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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.098926 #
5
5
  ######################################################################################################
6
6
 
@@ -0,0 +1,6 @@
1
+ ######################################################################################################
2
+ # Auto-generated Metaflow stub file #
3
+ # MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.110253 #
5
+ ######################################################################################################
6
+
@@ -0,0 +1,6 @@
1
+ ######################################################################################################
2
+ # Auto-generated Metaflow stub file #
3
+ # MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.110319 #
5
+ ######################################################################################################
6
+
@@ -0,0 +1,92 @@
1
+ ######################################################################################################
2
+ # Auto-generated Metaflow stub file #
3
+ # MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.110409 #
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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.113425 #
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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.113477 #
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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.133968 #
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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.098985 #
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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.099032 #
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,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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.135381 #
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.612044 #
3
+ # MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.135438 #
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.571899 #
3
+ # MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.093965 #
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.12.36.1+obcheckpoint(0.1.4);ob(v1) #
4
- # Generated on 2024-12-09T17:36:48.573142 #
3
+ # MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.095124 #
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._vendor.click.types
13
- import typing
14
13
  import metaflow.parameters
14
+ import typing
15
15
 
16
16
  from ._vendor import click as click
17
17
  from .exception import ParameterFieldFailed as ParameterFieldFailed
@@ -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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.098502 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,24 +9,24 @@ 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
12
+ from . import frameworks as frameworks
13
+ from . import parallel_decorator as parallel_decorator
14
+ from . import environment_decorator as environment_decorator
13
15
  from . import resources_decorator as resources_decorator
14
- from . import timeout_decorator as timeout_decorator
15
- from . import aws as aws
16
+ from . import catch_decorator as catch_decorator
16
17
  from . import datatools as datatools
17
18
  from . import gcp as gcp
18
19
  from . import storage_executor as storage_executor
19
20
  from . import cards as cards
20
21
  from . import pypi as pypi
21
- from . import environment_decorator as environment_decorator
22
+ from . import aws as aws
23
+ from . import timeout_decorator as timeout_decorator
22
24
  from . import argo as argo
23
- from . import parallel_decorator as parallel_decorator
24
- from . import catch_decorator as catch_decorator
25
+ from . import kubernetes as kubernetes
25
26
  from . import secrets as secrets
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
29
+ from . import events_decorator as events_decorator
30
30
  from . import project_decorator as project_decorator
31
31
  from . import azure as azure
32
32
  from . import debug_logger as debug_logger
@@ -48,6 +48,7 @@ from .cards.card_modules.test_cards import TestRefreshComponentCard as TestRefre
48
48
  from . import logs_cli as logs_cli
49
49
  from . import tag_cli as tag_cli
50
50
  from . import package_cli as package_cli
51
+ from . import snowflake as snowflake
51
52
  from . import perimeters as perimeters
52
53
 
53
54
  CLIS_DESC: 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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.128439 #
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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.160922 #
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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.160043 #
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.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.161171 #
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.649102 #
3
+ # MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.172529 #
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.649358 #
3
+ # MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.172792 #
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.649610 #
3
+ # MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.173019 #
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.603523 #
3
+ # MF version: 2.12.36.3+obcheckpoint(0.1.4);ob(v1) #
4
+ # Generated on 2024-12-10T22:06:22.127453 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations