ob-metaflow-stubs 6.0.3.114__py2.py3-none-any.whl → 6.0.3.115__py2.py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (140) hide show
  1. metaflow-stubs/__init__.pyi +498 -498
  2. metaflow-stubs/cards.pyi +5 -5
  3. metaflow-stubs/cli.pyi +3 -3
  4. metaflow-stubs/client/__init__.pyi +5 -5
  5. metaflow-stubs/client/core.pyi +11 -9
  6. metaflow-stubs/client/filecache.pyi +2 -2
  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 +5 -5
  11. metaflow-stubs/generated_for.txt +1 -1
  12. metaflow-stubs/includefile.pyi +5 -5
  13. metaflow-stubs/info_file.pyi +2 -2
  14. metaflow-stubs/metadata/metadata.pyi +6 -3
  15. metaflow-stubs/metadata/util.pyi +2 -2
  16. metaflow-stubs/metaflow_config.pyi +2 -2
  17. metaflow-stubs/metaflow_current.pyi +68 -68
  18. metaflow-stubs/mflog/mflog.pyi +2 -2
  19. metaflow-stubs/multicore_utils.pyi +2 -2
  20. metaflow-stubs/parameters.pyi +3 -3
  21. metaflow-stubs/plugins/__init__.pyi +2 -2
  22. metaflow-stubs/plugins/airflow/__init__.pyi +2 -2
  23. metaflow-stubs/plugins/airflow/airflow_utils.pyi +2 -2
  24. metaflow-stubs/plugins/airflow/exception.pyi +2 -2
  25. metaflow-stubs/plugins/airflow/sensors/__init__.pyi +2 -2
  26. metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi +3 -3
  27. metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi +4 -4
  28. metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi +4 -4
  29. metaflow-stubs/plugins/argo/__init__.pyi +2 -2
  30. metaflow-stubs/plugins/argo/argo_client.pyi +2 -2
  31. metaflow-stubs/plugins/argo/argo_events.pyi +2 -2
  32. metaflow-stubs/plugins/argo/argo_workflows.pyi +4 -4
  33. metaflow-stubs/plugins/argo/argo_workflows_cli.pyi +5 -5
  34. metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +7 -7
  35. metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +2 -2
  36. metaflow-stubs/plugins/aws/__init__.pyi +2 -2
  37. metaflow-stubs/plugins/aws/aws_client.pyi +2 -2
  38. metaflow-stubs/plugins/aws/aws_utils.pyi +2 -2
  39. metaflow-stubs/plugins/aws/batch/__init__.pyi +2 -2
  40. metaflow-stubs/plugins/aws/batch/batch.pyi +2 -2
  41. metaflow-stubs/plugins/aws/batch/batch_client.pyi +2 -2
  42. metaflow-stubs/plugins/aws/batch/batch_decorator.pyi +5 -5
  43. metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi +2 -2
  44. metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi +4 -4
  45. metaflow-stubs/plugins/aws/step_functions/__init__.pyi +2 -2
  46. metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi +2 -2
  47. metaflow-stubs/plugins/aws/step_functions/production_token.pyi +2 -2
  48. metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi +2 -2
  49. metaflow-stubs/plugins/aws/step_functions/step_functions.pyi +2 -2
  50. metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi +2 -2
  51. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi +2 -2
  52. metaflow-stubs/plugins/azure/__init__.pyi +2 -2
  53. metaflow-stubs/plugins/azure/azure_credential.pyi +2 -2
  54. metaflow-stubs/plugins/azure/azure_exceptions.pyi +2 -2
  55. metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi +4 -4
  56. metaflow-stubs/plugins/azure/azure_utils.pyi +2 -2
  57. metaflow-stubs/plugins/azure/blob_service_client_factory.pyi +2 -2
  58. metaflow-stubs/plugins/azure/includefile_support.pyi +2 -2
  59. metaflow-stubs/plugins/cards/__init__.pyi +2 -2
  60. metaflow-stubs/plugins/cards/card_cli.pyi +4 -4
  61. metaflow-stubs/plugins/cards/card_client.pyi +3 -3
  62. metaflow-stubs/plugins/cards/card_creator.pyi +2 -2
  63. metaflow-stubs/plugins/cards/card_datastore.pyi +2 -2
  64. metaflow-stubs/plugins/cards/card_decorator.pyi +3 -3
  65. metaflow-stubs/plugins/cards/card_modules/__init__.pyi +2 -2
  66. metaflow-stubs/plugins/cards/card_modules/basic.pyi +3 -3
  67. metaflow-stubs/plugins/cards/card_modules/card.pyi +2 -2
  68. metaflow-stubs/plugins/cards/card_modules/components.pyi +4 -4
  69. metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi +2 -2
  70. metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi +2 -2
  71. metaflow-stubs/plugins/cards/card_modules/test_cards.pyi +3 -3
  72. metaflow-stubs/plugins/cards/card_resolver.pyi +2 -2
  73. metaflow-stubs/plugins/cards/component_serializer.pyi +4 -4
  74. metaflow-stubs/plugins/cards/exception.pyi +2 -2
  75. metaflow-stubs/plugins/catch_decorator.pyi +4 -4
  76. metaflow-stubs/plugins/datatools/__init__.pyi +3 -3
  77. metaflow-stubs/plugins/datatools/local.pyi +2 -2
  78. metaflow-stubs/plugins/datatools/s3/__init__.pyi +3 -3
  79. metaflow-stubs/plugins/datatools/s3/s3.pyi +7 -7
  80. metaflow-stubs/plugins/datatools/s3/s3tail.pyi +2 -2
  81. metaflow-stubs/plugins/datatools/s3/s3util.pyi +2 -2
  82. metaflow-stubs/plugins/debug_logger.pyi +2 -2
  83. metaflow-stubs/plugins/debug_monitor.pyi +2 -2
  84. metaflow-stubs/plugins/environment_decorator.pyi +2 -2
  85. metaflow-stubs/plugins/events_decorator.pyi +3 -3
  86. metaflow-stubs/plugins/frameworks/__init__.pyi +2 -2
  87. metaflow-stubs/plugins/frameworks/pytorch.pyi +3 -3
  88. metaflow-stubs/plugins/gcp/__init__.pyi +2 -2
  89. metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi +4 -4
  90. metaflow-stubs/plugins/gcp/gs_exceptions.pyi +2 -2
  91. metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi +2 -2
  92. metaflow-stubs/plugins/gcp/gs_utils.pyi +2 -2
  93. metaflow-stubs/plugins/gcp/includefile_support.pyi +2 -2
  94. metaflow-stubs/plugins/kubernetes/__init__.pyi +2 -2
  95. metaflow-stubs/plugins/kubernetes/kube_utils.pyi +2 -2
  96. metaflow-stubs/plugins/kubernetes/kubernetes.pyi +3 -3
  97. metaflow-stubs/plugins/kubernetes/kubernetes_cli.pyi +3 -3
  98. metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +2 -2
  99. metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +5 -5
  100. metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi +2 -2
  101. metaflow-stubs/plugins/logs_cli.pyi +3 -3
  102. metaflow-stubs/plugins/package_cli.pyi +2 -2
  103. metaflow-stubs/plugins/parallel_decorator.pyi +3 -3
  104. metaflow-stubs/plugins/perimeters.pyi +2 -2
  105. metaflow-stubs/plugins/project_decorator.pyi +3 -3
  106. metaflow-stubs/plugins/pypi/__init__.pyi +2 -2
  107. metaflow-stubs/plugins/pypi/conda_decorator.pyi +2 -2
  108. metaflow-stubs/plugins/pypi/conda_environment.pyi +5 -5
  109. metaflow-stubs/plugins/pypi/pypi_decorator.pyi +2 -2
  110. metaflow-stubs/plugins/pypi/pypi_environment.pyi +3 -3
  111. metaflow-stubs/plugins/pypi/utils.pyi +2 -2
  112. metaflow-stubs/plugins/resources_decorator.pyi +3 -3
  113. metaflow-stubs/plugins/retry_decorator.pyi +2 -2
  114. metaflow-stubs/plugins/secrets/__init__.pyi +2 -2
  115. metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +3 -3
  116. metaflow-stubs/plugins/secrets/secrets_decorator.pyi +2 -2
  117. metaflow-stubs/plugins/storage_executor.pyi +2 -2
  118. metaflow-stubs/plugins/tag_cli.pyi +4 -4
  119. metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +2 -2
  120. metaflow-stubs/plugins/timeout_decorator.pyi +3 -3
  121. metaflow-stubs/procpoll.pyi +2 -2
  122. metaflow-stubs/profilers/__init__.pyi +2 -2
  123. metaflow-stubs/pylint_wrapper.pyi +2 -2
  124. metaflow-stubs/runner/__init__.pyi +2 -2
  125. metaflow-stubs/runner/deployer.pyi +4 -4
  126. metaflow-stubs/runner/metaflow_runner.pyi +5 -33
  127. metaflow-stubs/runner/nbdeploy.pyi +2 -2
  128. metaflow-stubs/runner/nbrun.pyi +2 -2
  129. metaflow-stubs/runner/subprocess_manager.pyi +2 -2
  130. metaflow-stubs/runner/utils.pyi +2 -2
  131. metaflow-stubs/system/__init__.pyi +4 -4
  132. metaflow-stubs/system/system_logger.pyi +3 -3
  133. metaflow-stubs/system/system_monitor.pyi +3 -3
  134. metaflow-stubs/tagging_util.pyi +2 -2
  135. metaflow-stubs/tuple_util.pyi +2 -2
  136. {ob_metaflow_stubs-6.0.3.114.dist-info → ob_metaflow_stubs-6.0.3.115.dist-info}/METADATA +1 -1
  137. ob_metaflow_stubs-6.0.3.115.dist-info/RECORD +140 -0
  138. ob_metaflow_stubs-6.0.3.114.dist-info/RECORD +0 -140
  139. {ob_metaflow_stubs-6.0.3.114.dist-info → ob_metaflow_stubs-6.0.3.115.dist-info}/WHEEL +0 -0
  140. {ob_metaflow_stubs-6.0.3.114.dist-info → ob_metaflow_stubs-6.0.3.115.dist-info}/top_level.txt +0 -0
metaflow-stubs/cards.pyi CHANGED
@@ -1,19 +1,19 @@
1
1
  ##################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.137003 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.745053 #
5
5
  ##################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
9
  import typing
10
10
  if typing.TYPE_CHECKING:
11
- import metaflow
12
- import metaflow.plugins.cards.card_modules.components
11
+ import typing
13
12
  import metaflow.plugins.cards.card_modules.basic
14
13
  import metaflow.plugins.cards.card_client
14
+ import metaflow
15
+ import metaflow.plugins.cards.card_modules.components
15
16
  import metaflow.plugins.cards.card_modules.card
16
- import typing
17
17
 
18
18
  def get_cards(task: typing.Union[str, "metaflow.Task"], id: typing.Optional[str] = None, type: typing.Optional[str] = None, follow_resumed: bool = True) -> metaflow.plugins.cards.card_client.CardContainer:
19
19
  """
metaflow-stubs/cli.pyi CHANGED
@@ -1,15 +1,15 @@
1
1
  ##################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.153873 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.761596 #
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.exception
12
11
  import metaflow.metaflow_current
12
+ import metaflow.exception
13
13
 
14
14
  def get_metadata() -> str:
15
15
  """
@@ -1,16 +1,16 @@
1
1
  ##################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.143901 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.751982 #
5
5
  ##################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
9
  import typing
10
10
  if typing.TYPE_CHECKING:
11
- import datetime
12
- import metaflow.events
13
11
  import typing
12
+ import metaflow.events
13
+ import datetime
14
14
  import metaflow.client.core
15
15
 
16
16
  def namespace(ns: typing.Optional[str]) -> typing.Optional[str]:
@@ -132,7 +132,7 @@ class Metaflow(object, metaclass=type):
132
132
  flows present in the current namespace will be returned. A `Flow` is present in a namespace
133
133
  if it has at least one run in the namespace.
134
134
  """
135
- def __init__(self):
135
+ def __init__(self, _current_metadata: typing.Optional[str] = None):
136
136
  ...
137
137
  @property
138
138
  def flows(self) -> typing.List[metaflow.client.core.Flow]:
@@ -1,21 +1,23 @@
1
1
  ##################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.132836 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.741608 #
5
5
  ##################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
9
  import typing
10
10
  if typing.TYPE_CHECKING:
11
- import tarfile
12
- import metaflow
11
+ import typing
12
+ import metaflow.metaflow_current
13
+ import datetime
13
14
  import metaflow.events
14
15
  import metaflow.exception
15
16
  import metaflow.client.core
16
- import metaflow.metaflow_current
17
- import datetime
18
- import typing
17
+ import tarfile
18
+ import metaflow
19
+
20
+ TYPE_CHECKING: bool
19
21
 
20
22
  current: metaflow.metaflow_current.Current
21
23
 
@@ -349,7 +351,7 @@ class MetaflowObject(object, metaclass=type):
349
351
  Pathspec of the original object this object was cloned from (in the case of a resume).
350
352
  None if not applicable.
351
353
  """
352
- def __init__(self, pathspec: typing.Optional[str] = None, attempt: typing.Optional[int] = None, _object: typing.Optional["MetaflowObject"] = None, _parent: typing.Optional["MetaflowObject"] = None, _namespace_check: bool = True, _current_namespace: typing.Optional[str] = None):
354
+ def __init__(self, pathspec: typing.Optional[str] = None, attempt: typing.Optional[int] = None, _object: typing.Optional["MetaflowObject"] = None, _parent: typing.Optional["MetaflowObject"] = None, _namespace_check: bool = True, _metaflow: typing.Optional["Metaflow"] = None, _current_namespace: typing.Optional[str] = None, _current_metadata: typing.Optional[str] = None):
353
355
  ...
354
356
  def __iter__(self) -> typing.Iterator["MetaflowObject"]:
355
357
  """
@@ -1593,7 +1595,7 @@ class Metaflow(object, metaclass=type):
1593
1595
  flows present in the current namespace will be returned. A `Flow` is present in a namespace
1594
1596
  if it has at least one run in the namespace.
1595
1597
  """
1596
- def __init__(self):
1598
+ def __init__(self, _current_metadata: typing.Optional[str] = None):
1597
1599
  ...
1598
1600
  @property
1599
1601
  def flows(self) -> typing.List[Flow]:
@@ -1,7 +1,7 @@
1
1
  ##################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.154756 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.762456 #
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.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.152712 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.760436 #
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.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.134915 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.745670 #
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.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.124998 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.732561 #
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.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.134316 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.742986 #
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.flowspec
11
+ import typing
12
12
  import metaflow.unbounded_foreach
13
- import metaflow.datastore.inputs
14
13
  import metaflow.parameters
15
14
  import metaflow.exception
16
- import typing
15
+ import metaflow.datastore.inputs
16
+ import metaflow.flowspec
17
17
 
18
18
  class DelayedEvaluationParameter(object, metaclass=type):
19
19
  """
@@ -1 +1 @@
1
- 2.12.26.1+obcheckpoint(0.1.1);ob(v1) 2024-10-23T20:57:01.109645
1
+ 2.12.27.1+obcheckpoint(0.1.1);ob(v1) 2024-10-25T21:39:54.716202
@@ -1,18 +1,18 @@
1
1
  ##################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.139910 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.748434 #
5
5
  ##################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
9
  import typing
10
10
  if typing.TYPE_CHECKING:
11
- import io
11
+ import typing
12
+ import metaflow._vendor.click.types
12
13
  import metaflow.parameters
13
14
  import metaflow.plugins.datatools.s3.s3
14
- import metaflow._vendor.click.types
15
- import typing
15
+ import io
16
16
 
17
17
  class MetaflowException(Exception, metaclass=type):
18
18
  def __init__(self, msg = "", lineno = None):
@@ -1,7 +1,7 @@
1
1
  ##################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.124316 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.731461 #
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.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.179096 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.786249 #
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.metadata.metadata
12
11
  import metaflow.exception
12
+ import metaflow.metadata.metadata
13
13
 
14
14
  class MetaflowInternalError(metaflow.exception.MetaflowException, metaclass=type):
15
15
  ...
@@ -99,6 +99,9 @@ class ObjectOrder(object, metaclass=type):
99
99
 
100
100
  class MetadataProvider(object, metaclass=MetadataProviderMeta):
101
101
  @classmethod
102
+ def metadata_str(cls):
103
+ ...
104
+ @classmethod
102
105
  def compute_info(cls, val):
103
106
  """
104
107
  Compute the new information for this provider
@@ -1,7 +1,7 @@
1
1
  ##################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.216115 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.833493 #
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.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.126055 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.734509 #
5
5
  ##################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,21 +1,21 @@
1
1
  ##################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.226367 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.836176 #
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_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator
12
- import metaflow
13
- import metaflow.events
11
+ import typing
14
12
  import metaflow.metaflow_current
15
- import metaflow.plugins.cards.component_serializer
13
+ import metaflow.events
14
+ import metaflow_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator
16
15
  import metaflow_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core
16
+ import metaflow.plugins.cards.component_serializer
17
+ import metaflow
17
18
  import metaflow_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator
18
- import typing
19
19
 
20
20
  TYPE_CHECKING: bool
21
21
 
@@ -288,6 +288,67 @@ class Current(object, metaclass=type):
288
288
  """
289
289
  ...
290
290
  @property
291
+ def model(self) -> "metaflow_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core.ModelSerializer":
292
+ """
293
+ (only in the presence of the @model decorator)
294
+
295
+ The object used for loading / saving models.
296
+ `current.model` exposes a `save` method to save models and a `load` method to load models.
297
+ `current.model.loaded` exposes the paths to the models loaded via the `load` argument in the @model decorator
298
+ or models loaded via `current.model.load`.
299
+
300
+ Usage (Saving a model):
301
+ -------
302
+
303
+ ```
304
+ @model
305
+ @step
306
+ def train(self):
307
+ # current.model.save returns a dictionary reference to the model saved
308
+ self.my_model = current.model.save(
309
+ path_to_my_model,
310
+ label="my_model",
311
+ metadata={
312
+ "epochs": 10,
313
+ "batch-size": 32,
314
+ "learning-rate": 0.001,
315
+ }
316
+ )
317
+ self.next(self.test)
318
+
319
+ @model(load="my_model")
320
+ @step
321
+ def test(self):
322
+ # `current.model.loaded` returns a dictionary of the loaded models
323
+ # where the key is the name of the artifact and the value is the path to the model
324
+ print(os.listdir(current.model.loaded["my_model"]))
325
+ self.next(self.end)
326
+ ```
327
+
328
+ Usage (Loading models):
329
+ -------
330
+
331
+ ```
332
+ @step
333
+ def train(self):
334
+ # current.model.load returns the path to the model loaded
335
+ checkpoint_path = current.model.load(
336
+ self.checkpoint_key,
337
+ )
338
+ model_path = current.model.load(
339
+ self.model,
340
+ )
341
+ self.next(self.test)
342
+ ```
343
+
344
+
345
+ Returns
346
+ -------
347
+ ModelSerializer
348
+ The object used for loading / saving models.
349
+ """
350
+ ...
351
+ @property
291
352
  def card(self) -> "metaflow.plugins.cards.component_serializer.CardComponentCollector":
292
353
  """
293
354
  (only in the presence of the @card decorator)
@@ -368,67 +429,6 @@ class Current(object, metaclass=type):
368
429
  """
369
430
  ...
370
431
  @property
371
- def model(self) -> "metaflow_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core.ModelSerializer":
372
- """
373
- (only in the presence of the @model decorator)
374
-
375
- The object used for loading / saving models.
376
- `current.model` exposes a `save` method to save models and a `load` method to load models.
377
- `current.model.loaded` exposes the paths to the models loaded via the `load` argument in the @model decorator
378
- or models loaded via `current.model.load`.
379
-
380
- Usage (Saving a model):
381
- -------
382
-
383
- ```
384
- @model
385
- @step
386
- def train(self):
387
- # current.model.save returns a dictionary reference to the model saved
388
- self.my_model = current.model.save(
389
- path_to_my_model,
390
- label="my_model",
391
- metadata={
392
- "epochs": 10,
393
- "batch-size": 32,
394
- "learning-rate": 0.001,
395
- }
396
- )
397
- self.next(self.test)
398
-
399
- @model(load="my_model")
400
- @step
401
- def test(self):
402
- # `current.model.loaded` returns a dictionary of the loaded models
403
- # where the key is the name of the artifact and the value is the path to the model
404
- print(os.listdir(current.model.loaded["my_model"]))
405
- self.next(self.end)
406
- ```
407
-
408
- Usage (Loading models):
409
- -------
410
-
411
- ```
412
- @step
413
- def train(self):
414
- # current.model.load returns the path to the model loaded
415
- checkpoint_path = current.model.load(
416
- self.checkpoint_key,
417
- )
418
- model_path = current.model.load(
419
- self.model,
420
- )
421
- self.next(self.test)
422
- ```
423
-
424
-
425
- Returns
426
- -------
427
- ModelSerializer
428
- The object used for loading / saving models.
429
- """
430
- ...
431
- @property
432
432
  def trigger(self) -> "metaflow.events.Trigger":
433
433
  """
434
434
  (only in the presence of the @trigger_on_finish, or @trigger decorators)
@@ -1,7 +1,7 @@
1
1
  ##################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.179460 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.786741 #
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.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.126552 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.735169 #
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.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.127789 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.736490 #
5
5
  ##################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
9
  import typing
10
10
  if typing.TYPE_CHECKING:
11
+ import typing
11
12
  import metaflow.parameters
12
13
  import metaflow.exception
13
14
  import metaflow._vendor.click.types
14
- import typing
15
15
 
16
16
  class ParameterFieldFailed(metaflow.exception.MetaflowException, metaclass=type):
17
17
  def __init__(self, name, field):
@@ -1,7 +1,7 @@
1
1
  ##################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.145685 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.753757 #
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.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.173472 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.780145 #
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.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.213283 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.823226 #
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.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.212332 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.822259 #
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.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.213580 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.823525 #
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.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.224209 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.834025 #
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.exception
12
11
  import metaflow.decorators
12
+ import metaflow.exception
13
13
 
14
14
  class AirflowException(metaflow.exception.MetaflowException, metaclass=type):
15
15
  def __init__(self, msg):
@@ -1,16 +1,16 @@
1
1
  ##################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.224629 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.834534 #
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.plugins.airflow.sensors.base_sensor
12
- import metaflow.exception
13
11
  import metaflow.decorators
12
+ import metaflow.exception
13
+ import metaflow.plugins.airflow.sensors.base_sensor
14
14
 
15
15
  class AirflowSensorDecorator(metaflow.decorators.FlowDecorator, metaclass=type):
16
16
  """
@@ -1,16 +1,16 @@
1
1
  ##################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.225008 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.834913 #
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.plugins.airflow.sensors.base_sensor
12
- import metaflow.exception
13
11
  import metaflow.decorators
12
+ import metaflow.exception
13
+ import metaflow.plugins.airflow.sensors.base_sensor
14
14
 
15
15
  class AirflowSensorDecorator(metaflow.decorators.FlowDecorator, metaclass=type):
16
16
  """
@@ -1,7 +1,7 @@
1
1
  ##################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.12.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.168142 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.779378 #
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.26.1+obcheckpoint(0.1.1);ob(v1) #
4
- # Generated on 2024-10-23T20:57:01.184751 #
3
+ # MF version: 2.12.27.1+obcheckpoint(0.1.1);ob(v1) #
4
+ # Generated on 2024-10-25T21:39:54.812794 #
5
5
  ##################################################################################
6
6
 
7
7
  from __future__ import annotations