ob-metaflow-stubs 6.0.3.128__py2.py3-none-any.whl → 6.0.3.129__py2.py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (191) hide show
  1. metaflow-stubs/__init__.pyi +320 -319
  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 +6 -6
  8. metaflow-stubs/client/filecache.pyi +2 -2
  9. metaflow-stubs/events.pyi +2 -2
  10. metaflow-stubs/exception.pyi +3 -3
  11. metaflow-stubs/flowspec.pyi +8 -8
  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 +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 +136 -136
  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/card_utils/__init__.pyi +2 -2
  26. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/async_cards.pyi +3 -3
  27. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/deco_injection_mixin.pyi +2 -2
  28. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/extra_components.pyi +4 -4
  29. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/__init__.pyi +2 -2
  30. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/__init__.pyi +2 -2
  31. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/checkpoint_lister.pyi +4 -4
  32. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/lineage_card.pyi +2 -2
  33. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/checkpoint_storage.pyi +4 -4
  34. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/constructors.pyi +2 -2
  35. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/core.pyi +4 -4
  36. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/decorator.pyi +5 -5
  37. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/exceptions.pyi +2 -2
  38. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/final_api.pyi +3 -3
  39. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/lineage.pyi +2 -2
  40. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/__init__.pyi +2 -2
  41. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/core.pyi +3 -3
  42. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/exceptions.pyi +2 -2
  43. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/task_utils.pyi +3 -3
  44. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/utils.pyi +2 -2
  45. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastructures.pyi +3 -3
  46. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/exceptions.pyi +2 -2
  47. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/__init__.pyi +2 -2
  48. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/decorator.pyi +2 -2
  49. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/__init__.pyi +2 -2
  50. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/core.pyi +3 -3
  51. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/exceptions.pyi +2 -2
  52. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/model_storage.pyi +5 -5
  53. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/__init__.pyi +2 -2
  54. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/flowspec_utils.pyi +2 -2
  55. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/general.pyi +2 -2
  56. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/identity_utils.pyi +2 -2
  57. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/__init__.pyi +2 -2
  58. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/base.pyi +2 -2
  59. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/tar.pyi +3 -3
  60. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/tar_utils.pyi +3 -3
  61. metaflow-stubs/mf_extensions/outerbounds/__init__.pyi +2 -2
  62. metaflow-stubs/mf_extensions/outerbounds/plugins/__init__.pyi +2 -2
  63. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/__init__.pyi +2 -2
  64. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/snowflake.pyi +2 -2
  65. metaflow-stubs/mf_extensions/outerbounds/profilers/__init__.pyi +2 -2
  66. metaflow-stubs/mf_extensions/outerbounds/profilers/gpu.pyi +2 -2
  67. metaflow-stubs/mf_extensions/outerbounds/remote_config.pyi +2 -2
  68. metaflow-stubs/mf_extensions/outerbounds/toplevel/__init__.pyi +2 -2
  69. metaflow-stubs/mf_extensions/outerbounds/toplevel/global_aliases_for_metaflow_package.pyi +2 -2
  70. metaflow-stubs/multicore_utils.pyi +2 -2
  71. metaflow-stubs/parameters.pyi +5 -5
  72. metaflow-stubs/plugins/__init__.pyi +13 -13
  73. metaflow-stubs/plugins/airflow/__init__.pyi +2 -2
  74. metaflow-stubs/plugins/airflow/airflow_utils.pyi +2 -2
  75. metaflow-stubs/plugins/airflow/exception.pyi +2 -2
  76. metaflow-stubs/plugins/airflow/sensors/__init__.pyi +2 -2
  77. metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi +2 -2
  78. metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi +2 -2
  79. metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi +2 -2
  80. metaflow-stubs/plugins/argo/__init__.pyi +2 -2
  81. metaflow-stubs/plugins/argo/argo_client.pyi +2 -2
  82. metaflow-stubs/plugins/argo/argo_events.pyi +2 -2
  83. metaflow-stubs/plugins/argo/argo_workflows.pyi +2 -2
  84. metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +3 -3
  85. metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +4 -4
  86. metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi +4 -4
  87. metaflow-stubs/plugins/aws/__init__.pyi +2 -2
  88. metaflow-stubs/plugins/aws/aws_client.pyi +2 -2
  89. metaflow-stubs/plugins/aws/aws_utils.pyi +2 -2
  90. metaflow-stubs/plugins/aws/batch/__init__.pyi +2 -2
  91. metaflow-stubs/plugins/aws/batch/batch.pyi +2 -2
  92. metaflow-stubs/plugins/aws/batch/batch_client.pyi +2 -2
  93. metaflow-stubs/plugins/aws/batch/batch_decorator.pyi +2 -2
  94. metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi +2 -2
  95. metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi +15 -13
  96. metaflow-stubs/plugins/aws/step_functions/__init__.pyi +2 -2
  97. metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi +2 -2
  98. metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi +2 -2
  99. metaflow-stubs/plugins/aws/step_functions/step_functions.pyi +2 -2
  100. metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi +2 -2
  101. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi +4 -4
  102. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi +3 -3
  103. metaflow-stubs/plugins/azure/__init__.pyi +2 -2
  104. metaflow-stubs/plugins/azure/azure_credential.pyi +2 -2
  105. metaflow-stubs/plugins/azure/azure_exceptions.pyi +2 -2
  106. metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi +3 -3
  107. metaflow-stubs/plugins/azure/azure_utils.pyi +2 -2
  108. metaflow-stubs/plugins/azure/blob_service_client_factory.pyi +2 -2
  109. metaflow-stubs/plugins/azure/includefile_support.pyi +2 -2
  110. metaflow-stubs/plugins/cards/__init__.pyi +2 -2
  111. metaflow-stubs/plugins/cards/card_client.pyi +2 -2
  112. metaflow-stubs/plugins/cards/card_creator.pyi +2 -2
  113. metaflow-stubs/plugins/cards/card_datastore.pyi +2 -2
  114. metaflow-stubs/plugins/cards/card_decorator.pyi +4 -2
  115. metaflow-stubs/plugins/cards/card_modules/__init__.pyi +2 -2
  116. metaflow-stubs/plugins/cards/card_modules/basic.pyi +3 -3
  117. metaflow-stubs/plugins/cards/card_modules/card.pyi +2 -2
  118. metaflow-stubs/plugins/cards/card_modules/components.pyi +4 -4
  119. metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi +2 -2
  120. metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi +2 -2
  121. metaflow-stubs/plugins/cards/card_modules/test_cards.pyi +2 -2
  122. metaflow-stubs/plugins/cards/card_resolver.pyi +2 -2
  123. metaflow-stubs/plugins/cards/component_serializer.pyi +2 -2
  124. metaflow-stubs/plugins/cards/exception.pyi +2 -2
  125. metaflow-stubs/plugins/catch_decorator.pyi +3 -3
  126. metaflow-stubs/plugins/datatools/__init__.pyi +2 -2
  127. metaflow-stubs/plugins/datatools/local.pyi +2 -2
  128. metaflow-stubs/plugins/datatools/s3/__init__.pyi +2 -2
  129. metaflow-stubs/plugins/datatools/s3/s3.pyi +4 -4
  130. metaflow-stubs/plugins/datatools/s3/s3tail.pyi +2 -2
  131. metaflow-stubs/plugins/datatools/s3/s3util.pyi +2 -2
  132. metaflow-stubs/plugins/debug_logger.pyi +2 -2
  133. metaflow-stubs/plugins/debug_monitor.pyi +2 -2
  134. metaflow-stubs/plugins/environment_decorator.pyi +2 -2
  135. metaflow-stubs/plugins/events_decorator.pyi +2 -2
  136. metaflow-stubs/plugins/frameworks/__init__.pyi +2 -2
  137. metaflow-stubs/plugins/frameworks/pytorch.pyi +2 -2
  138. metaflow-stubs/plugins/gcp/__init__.pyi +2 -2
  139. metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi +3 -3
  140. metaflow-stubs/plugins/gcp/gs_exceptions.pyi +2 -2
  141. metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi +2 -2
  142. metaflow-stubs/plugins/gcp/gs_utils.pyi +2 -2
  143. metaflow-stubs/plugins/gcp/includefile_support.pyi +2 -2
  144. metaflow-stubs/plugins/kubernetes/__init__.pyi +2 -2
  145. metaflow-stubs/plugins/kubernetes/kube_utils.pyi +2 -2
  146. metaflow-stubs/plugins/kubernetes/kubernetes.pyi +3 -3
  147. metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +2 -2
  148. metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +2 -2
  149. metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi +2 -2
  150. metaflow-stubs/plugins/parallel_decorator.pyi +2 -2
  151. metaflow-stubs/plugins/perimeters.pyi +2 -2
  152. metaflow-stubs/plugins/project_decorator.pyi +2 -2
  153. metaflow-stubs/plugins/pypi/__init__.pyi +2 -2
  154. metaflow-stubs/plugins/pypi/conda_decorator.pyi +2 -2
  155. metaflow-stubs/plugins/pypi/conda_environment.pyi +5 -5
  156. metaflow-stubs/plugins/pypi/pypi_decorator.pyi +2 -2
  157. metaflow-stubs/plugins/pypi/pypi_environment.pyi +2 -2
  158. metaflow-stubs/plugins/pypi/utils.pyi +2 -2
  159. metaflow-stubs/plugins/resources_decorator.pyi +2 -2
  160. metaflow-stubs/plugins/retry_decorator.pyi +2 -2
  161. metaflow-stubs/plugins/secrets/__init__.pyi +3 -3
  162. metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +3 -3
  163. metaflow-stubs/plugins/secrets/secrets_decorator.pyi +2 -2
  164. metaflow-stubs/plugins/snowflake/__init__.pyi +2 -2
  165. metaflow-stubs/plugins/storage_executor.pyi +2 -2
  166. metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +2 -2
  167. metaflow-stubs/plugins/timeout_decorator.pyi +3 -3
  168. metaflow-stubs/profilers/__init__.pyi +2 -2
  169. metaflow-stubs/pylint_wrapper.pyi +2 -2
  170. metaflow-stubs/runner/__init__.pyi +2 -2
  171. metaflow-stubs/runner/deployer.pyi +7 -7
  172. metaflow-stubs/runner/deployer_impl.pyi +2 -2
  173. metaflow-stubs/runner/metaflow_runner.pyi +2 -2
  174. metaflow-stubs/runner/nbdeploy.pyi +2 -2
  175. metaflow-stubs/runner/nbrun.pyi +2 -2
  176. metaflow-stubs/runner/subprocess_manager.pyi +2 -2
  177. metaflow-stubs/runner/utils.pyi +3 -3
  178. metaflow-stubs/system/__init__.pyi +2 -2
  179. metaflow-stubs/system/system_logger.pyi +3 -3
  180. metaflow-stubs/system/system_monitor.pyi +2 -2
  181. metaflow-stubs/tagging_util.pyi +2 -2
  182. metaflow-stubs/tuple_util.pyi +2 -2
  183. metaflow-stubs/user_configs/__init__.pyi +2 -2
  184. metaflow-stubs/user_configs/config_decorators.pyi +6 -6
  185. metaflow-stubs/user_configs/config_options.pyi +10 -5
  186. metaflow-stubs/user_configs/config_parameters.pyi +10 -10
  187. {ob_metaflow_stubs-6.0.3.128.dist-info → ob_metaflow_stubs-6.0.3.129.dist-info}/METADATA +1 -1
  188. ob_metaflow_stubs-6.0.3.129.dist-info/RECORD +191 -0
  189. ob_metaflow_stubs-6.0.3.128.dist-info/RECORD +0 -191
  190. {ob_metaflow_stubs-6.0.3.128.dist-info → ob_metaflow_stubs-6.0.3.129.dist-info}/WHEEL +0 -0
  191. {ob_metaflow_stubs-6.0.3.128.dist-info → ob_metaflow_stubs-6.0.3.129.dist-info}/top_level.txt +0 -0
metaflow-stubs/cards.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:34.004313 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.641958 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/cli.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:34.017658 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.660191 #
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.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:34.016490 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.658856 #
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.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:34.041799 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.685938 #
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.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:34.007201 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.645630 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,19 +1,19 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:34.011423 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.653546 #
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 datetime
13
- import typing
12
+ import tarfile
14
13
  import metaflow.client.core
15
14
  import metaflow.events
16
- import tarfile
15
+ import datetime
16
+ import typing
17
17
 
18
18
  from ..metaflow_current import current as current
19
19
  from ..events import Trigger as Trigger
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:34.036870 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.680994 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
metaflow-stubs/events.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:34.004081 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.642544 #
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.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:33.996282 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.634367 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -32,7 +32,7 @@ class MetaflowExceptionWrapper(Exception, metaclass=type):
32
32
  ...
33
33
 
34
34
  class MetaflowException(Exception, metaclass=type):
35
- def __init__(self, msg = '', lineno = None):
35
+ def __init__(self, msg = '', lineno = None, source_file = None):
36
36
  ...
37
37
  def __str__(self):
38
38
  ...
@@ -1,19 +1,19 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:34.000902 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.639033 #
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.datastore.inputs
12
+ import metaflow.exception
13
+ import metaflow.unbounded_foreach
13
14
  import metaflow.flowspec
14
15
  import typing
15
- import metaflow.unbounded_foreach
16
- import metaflow.exception
16
+ import metaflow.datastore.inputs
17
17
 
18
18
  from . import parameters as parameters
19
19
  from .parameters import DelayedEvaluationParameter as DelayedEvaluationParameter
@@ -22,6 +22,7 @@ from .exception import MetaflowException as MetaflowException
22
22
  from .exception import MissingInMergeArtifactsException as MissingInMergeArtifactsException
23
23
  from .exception import MetaflowInternalError as MetaflowInternalError
24
24
  from .exception import UnhandledInMergeArtifactsException as UnhandledInMergeArtifactsException
25
+ from .user_configs.config_parameters import ConfigValue as ConfigValue
25
26
  from .user_configs.config_decorators import CustomFlowDecorator as CustomFlowDecorator
26
27
  from .user_configs.config_decorators import CustomStepDecorator as CustomStepDecorator
27
28
  from .user_configs.config_decorators import MutableFlow as MutableFlow
@@ -51,8 +52,7 @@ class ParallelUBF(metaflow.unbounded_foreach.UnboundedForeachInput, metaclass=ty
51
52
  ...
52
53
 
53
54
  class FlowSpecMeta(type, metaclass=type):
54
- @staticmethod
55
- def __new__(cls, name, bases, dct):
55
+ def __init__(cls, name, bases, attrs):
56
56
  ...
57
57
  ...
58
58
 
@@ -1 +1 @@
1
- 2.12.39.1+obcheckpoint(0.1.6);ob(v1) 2024-12-18T07:29:33.993102
1
+ 2.13.0.1+obcheckpoint(0.1.6);ob(v1) 2024-12-20T17:11:09.630991
@@ -1,17 +1,17 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:34.006898 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.645258 #
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._vendor.click.types
12
13
  import metaflow.parameters
13
14
  import typing
14
- import metaflow._vendor.click.types
15
15
 
16
16
  from ._vendor import click as click
17
17
  from .exception import MetaflowException as MetaflowException
@@ -81,7 +81,7 @@ class IncludeFile(metaflow.parameters.Parameter, metaclass=type):
81
81
  """
82
82
  def __init__(self, name: str, required: typing.Optional[bool] = None, is_text: typing.Optional[bool] = None, encoding: typing.Optional[str] = None, help: typing.Optional[str] = None, **kwargs: typing.Dict[str, str]):
83
83
  ...
84
- def init(self):
84
+ def init(self, ignore_errors = False):
85
85
  ...
86
86
  def load_parameter(self, v):
87
87
  ...
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:33.995436 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.633483 #
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.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:33.999812 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.637929 #
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.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:34.021259 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.663977 #
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.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:34.020763 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.663423 #
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.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:34.020996 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.663703 #
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.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:33.997364 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.635436 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -289,6 +289,8 @@ MAX_CPU_PER_TASK: None
289
289
 
290
290
  MAX_ATTEMPTS: int
291
291
 
292
+ CLICK_API_PROCESS_CONFIG: bool
293
+
292
294
  def get_pinned_conda_libs(python_version, datastore_type):
293
295
  ...
294
296
 
@@ -1,21 +1,21 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:34.133820 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.780732 #
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.metaflow_current
12
11
  import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core
13
- import typing
14
- import metaflow.plugins.cards.component_serializer
15
12
  import metaflow
16
- import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator
17
- import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator
18
13
  import metaflow.events
14
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator
15
+ import metaflow.metaflow_current
16
+ import metaflow.plugins.cards.component_serializer
17
+ import typing
18
+ import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator
19
19
 
20
20
 
21
21
  TYPE_CHECKING: bool
@@ -228,6 +228,122 @@ class Current(object, metaclass=type):
228
228
  def graph(self):
229
229
  ...
230
230
  @property
231
+ def card(self) -> "metaflow.plugins.cards.component_serializer.CardComponentCollector":
232
+ """
233
+ (only in the presence of the @card decorator)
234
+
235
+ The `@card` decorator makes the cards available through the `current.card`
236
+ object. If multiple `@card` decorators are present, you can add an `ID` to
237
+ distinguish between them using `@card(id=ID)` as the decorator. You will then
238
+ be able to access that specific card using `current.card[ID].
239
+
240
+ Methods available are `append` and `extend`
241
+
242
+ Returns
243
+ -------
244
+ CardComponentCollector
245
+ The or one of the cards attached to this step.
246
+ """
247
+ ...
248
+ @property
249
+ def huggingface_hub(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator.HuggingfaceRegistry":
250
+ """
251
+ (only in the presence of the @huggingface_hub decorator)
252
+
253
+ The `@huggingface_hub` injects a `huggingface_hub` object into the `current` object. This object provides syntactic sugar
254
+ over [huggingface_hub](https://github.com/huggingface/huggingface_hub)'s
255
+ [snapshot_download](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/file_download#huggingface_hub.snapshot_download) function.
256
+ The `current.huggingface_hub.snapshot_download` function downloads objects from huggingface hub and saves them to the Metaflow's datastore under the
257
+ `<repo_type>/<repo_id>` name. The `repo_type` is by default `model` and can be overriden by passing the `repo_type` parameter to the `snapshot_download` function.
258
+
259
+
260
+ Usage:
261
+ ------
262
+
263
+ **Usage: creating references of models from huggingface that may be loaded in downstream steps**
264
+ ```python
265
+ @huggingface_hub
266
+ @step
267
+ def pull_model_from_huggingface(self):
268
+ # `current.huggingface_hub.snapshot_download` downloads the model from the Hugging Face Hub
269
+ # and saves it in the backend storage based on the model's `repo_id`. If there exists a model
270
+ # with the same `repo_id` in the backend storage, it will not download the model again. The return
271
+ # value of the function is a reference to the model in the backend storage.
272
+ # This reference can be used to load the model in the subsequent steps via `@model(load=["llama_model"])`
273
+
274
+ self.model_id = "mistralai/Mistral-7B-Instruct-v0.1"
275
+ self.llama_model = current.huggingface_hub.snapshot_download(
276
+ repo_id=self.model_id,
277
+ allow_patterns=["*.safetensors", "*.json", "tokenizer.*"],
278
+ )
279
+ self.next(self.train)
280
+ ```
281
+
282
+ **Usage: loading models directly from huggingface hub or from cache (from metaflow's datastore)**
283
+ ```python
284
+ @huggingface_hub(load=["mistralai/Mistral-7B-Instruct-v0.1"])
285
+ @step
286
+ def pull_model_from_huggingface(self):
287
+ path_to_model = current.huggingface_hub.loaded["mistralai/Mistral-7B-Instruct-v0.1"]
288
+ ```
289
+
290
+ ```python
291
+ @huggingface_hub(load=[("mistralai/Mistral-7B-Instruct-v0.1", "/my-directory"), ("myorg/mistral-lora, "/my-lora-directory")])
292
+ @step
293
+ def finetune_model(self):
294
+ path_to_model = current.huggingface_hub.loaded["mistralai/Mistral-7B-Instruct-v0.1"]
295
+ # path_to_model will be /my-directory
296
+ ```
297
+
298
+ ```python
299
+ # Takes all the arguments passed to `snapshot_download`
300
+ # except for `local_dir`
301
+ @huggingface_hub(load=[
302
+ {
303
+ "repo_id": "mistralai/Mistral-7B-Instruct-v0.1",
304
+ },
305
+ {
306
+ "repo_id": "myorg/mistral-lora",
307
+ "repo_type": "model",
308
+ },
309
+ ])
310
+ @step
311
+ def finetune_model(self):
312
+ path_to_model = current.huggingface_hub.loaded["mistralai/Mistral-7B-Instruct-v0.1"]
313
+ # path_to_model will be /my-directory
314
+ ```
315
+ """
316
+ ...
317
+ @property
318
+ def parallel(self) -> "metaflow.metaflow_current.Parallel":
319
+ """
320
+ (only in the presence of the @parallel decorator)
321
+
322
+ Returns a namedtuple with relevant information about the parallel task.
323
+
324
+ Returns
325
+ -------
326
+ Parallel
327
+ `namedtuple` with the following fields:
328
+ - main_ip (`str`)
329
+ The IP address of the control task.
330
+ - num_nodes (`int`)
331
+ The total number of tasks created by @parallel
332
+ - node_index (`int`)
333
+ The index of the current task in all the @parallel tasks.
334
+ - control_task_id (`Optional[str]`)
335
+ The task ID of the control task. Available to all tasks.
336
+ """
337
+ ...
338
+ @property
339
+ def is_parallel(self) -> bool:
340
+ """
341
+ (only in the presence of the @parallel decorator)
342
+
343
+ True if the current step is a @parallel step.
344
+ """
345
+ ...
346
+ @property
231
347
  def model(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core.ModelSerializer":
232
348
  """
233
349
  (only in the presence of the @model decorator)
@@ -289,35 +405,6 @@ class Current(object, metaclass=type):
289
405
  """
290
406
  ...
291
407
  @property
292
- def parallel(self) -> "metaflow.metaflow_current.Parallel":
293
- """
294
- (only in the presence of the @parallel decorator)
295
-
296
- Returns a namedtuple with relevant information about the parallel task.
297
-
298
- Returns
299
- -------
300
- Parallel
301
- `namedtuple` with the following fields:
302
- - main_ip (`str`)
303
- The IP address of the control task.
304
- - num_nodes (`int`)
305
- The total number of tasks created by @parallel
306
- - node_index (`int`)
307
- The index of the current task in all the @parallel tasks.
308
- - control_task_id (`Optional[str]`)
309
- The task ID of the control task. Available to all tasks.
310
- """
311
- ...
312
- @property
313
- def is_parallel(self) -> bool:
314
- """
315
- (only in the presence of the @parallel decorator)
316
-
317
- True if the current step is a @parallel step.
318
- """
319
- ...
320
- @property
321
408
  def checkpoint(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator.CurrentCheckpointer":
322
409
  """
323
410
  (only in the presence of the @checkpoint decorator)
@@ -379,106 +466,6 @@ class Current(object, metaclass=type):
379
466
  """
380
467
  ...
381
468
  @property
382
- def huggingface_hub(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator.HuggingfaceRegistry":
383
- """
384
- (only in the presence of the @huggingface_hub decorator)
385
-
386
- The `@huggingface_hub` injects a `huggingface_hub` object into the `current` object. This object provides syntactic sugar
387
- over [huggingface_hub](https://github.com/huggingface/huggingface_hub)'s
388
- [snapshot_download](https://huggingface.co/docs/huggingface_hub/main/en/package_reference/file_download#huggingface_hub.snapshot_download) function.
389
- The `current.huggingface_hub.snapshot_download` function downloads objects from huggingface hub and saves them to the Metaflow's datastore under the
390
- `<repo_type>/<repo_id>` name. The `repo_type` is by default `model` and can be overriden by passing the `repo_type` parameter to the `snapshot_download` function.
391
-
392
-
393
- Usage:
394
- ------
395
-
396
- **Usage: creating references of models from huggingface that may be loaded in downstream steps**
397
- ```python
398
- @huggingface_hub
399
- @step
400
- def pull_model_from_huggingface(self):
401
- # `current.huggingface_hub.snapshot_download` downloads the model from the Hugging Face Hub
402
- # and saves it in the backend storage based on the model's `repo_id`. If there exists a model
403
- # with the same `repo_id` in the backend storage, it will not download the model again. The return
404
- # value of the function is a reference to the model in the backend storage.
405
- # This reference can be used to load the model in the subsequent steps via `@model(load=["llama_model"])`
406
-
407
- self.model_id = "mistralai/Mistral-7B-Instruct-v0.1"
408
- self.llama_model = current.huggingface_hub.snapshot_download(
409
- repo_id=self.model_id,
410
- allow_patterns=["*.safetensors", "*.json", "tokenizer.*"],
411
- )
412
- self.next(self.train)
413
- ```
414
-
415
- **Usage: loading models directly from huggingface hub or from cache (from metaflow's datastore)**
416
- ```python
417
- @huggingface_hub(load=["mistralai/Mistral-7B-Instruct-v0.1"])
418
- @step
419
- def pull_model_from_huggingface(self):
420
- path_to_model = current.huggingface_hub.loaded["mistralai/Mistral-7B-Instruct-v0.1"]
421
- ```
422
-
423
- ```python
424
- @huggingface_hub(load=[("mistralai/Mistral-7B-Instruct-v0.1", "/my-directory"), ("myorg/mistral-lora, "/my-lora-directory")])
425
- @step
426
- def finetune_model(self):
427
- path_to_model = current.huggingface_hub.loaded["mistralai/Mistral-7B-Instruct-v0.1"]
428
- # path_to_model will be /my-directory
429
- ```
430
-
431
- ```python
432
- # Takes all the arguments passed to `snapshot_download`
433
- # except for `local_dir`
434
- @huggingface_hub(load=[
435
- {
436
- "repo_id": "mistralai/Mistral-7B-Instruct-v0.1",
437
- },
438
- {
439
- "repo_id": "myorg/mistral-lora",
440
- "repo_type": "model",
441
- },
442
- ])
443
- @step
444
- def finetune_model(self):
445
- path_to_model = current.huggingface_hub.loaded["mistralai/Mistral-7B-Instruct-v0.1"]
446
- # path_to_model will be /my-directory
447
- ```
448
- """
449
- ...
450
- @property
451
- def card(self) -> "metaflow.plugins.cards.component_serializer.CardComponentCollector":
452
- """
453
- (only in the presence of the @card decorator)
454
-
455
- The `@card` decorator makes the cards available through the `current.card`
456
- object. If multiple `@card` decorators are present, you can add an `ID` to
457
- distinguish between them using `@card(id=ID)` as the decorator. You will then
458
- be able to access that specific card using `current.card[ID].
459
-
460
- Methods available are `append` and `extend`
461
-
462
- Returns
463
- -------
464
- CardComponentCollector
465
- The or one of the cards attached to this step.
466
- """
467
- ...
468
- @property
469
- def trigger(self) -> "metaflow.events.Trigger":
470
- """
471
- (only in the presence of the @trigger, or @trigger_on_finish decorators)
472
-
473
- Returns `Trigger` if the current run is triggered by an event
474
-
475
- Returns
476
- -------
477
- Trigger
478
- `Trigger` if triggered by an event
479
- """
480
- ...
481
- @property
482
469
  def project_name(self) -> str:
483
470
  """
484
471
  (only in the presence of the @project decorator)
@@ -545,6 +532,19 @@ class Current(object, metaclass=type):
545
532
  True if the flow is deployed with `--production`.
546
533
  """
547
534
  ...
535
+ @property
536
+ def trigger(self) -> "metaflow.events.Trigger":
537
+ """
538
+ (only in the presence of the @trigger_on_finish, or @trigger decorators)
539
+
540
+ Returns `Trigger` if the current run is triggered by an event
541
+
542
+ Returns
543
+ -------
544
+ Trigger
545
+ `Trigger` if triggered by an event
546
+ """
547
+ ...
548
548
  ...
549
549
 
550
550
  current: Current
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:34.005648 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.644023 #
5
5
  ######################################################################################################
6
6
 
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:34.014218 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.656538 #
5
5
  ######################################################################################################
6
6
 
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:34.014289 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.656603 #
5
5
  ######################################################################################################
6
6
 
@@ -1,6 +1,6 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.39.1+obcheckpoint(0.1.6);ob(v1) #
4
- # Generated on 2024-12-18T07:29:34.014349 #
3
+ # MF version: 2.13.0.1+obcheckpoint(0.1.6);ob(v1) #
4
+ # Generated on 2024-12-20T17:11:09.656664 #
5
5
  ######################################################################################################
6
6