metaflow-stubs 2.18.13__py2.py3-none-any.whl → 2.19.1__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.

Potentially problematic release.


This version of metaflow-stubs might be problematic. Click here for more details.

Files changed (168) hide show
  1. metaflow-stubs/__init__.pyi +651 -650
  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 +3 -2
  7. metaflow-stubs/client/core.pyi +63 -32
  8. metaflow-stubs/client/filecache.pyi +20 -4
  9. metaflow-stubs/events.pyi +7 -8
  10. metaflow-stubs/exception.pyi +2 -2
  11. metaflow-stubs/flowspec.pyi +7 -8
  12. metaflow-stubs/generated_for.txt +1 -1
  13. metaflow-stubs/includefile.pyi +5 -5
  14. metaflow-stubs/meta_files.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 +16 -2
  20. metaflow-stubs/metaflow_current.pyi +14 -16
  21. metaflow-stubs/metaflow_git.pyi +3 -5
  22. metaflow-stubs/multicore_utils.pyi +4 -4
  23. metaflow-stubs/packaging_sys/__init__.pyi +39 -40
  24. metaflow-stubs/packaging_sys/backend.pyi +13 -13
  25. metaflow-stubs/packaging_sys/distribution_support.pyi +6 -7
  26. metaflow-stubs/packaging_sys/tar_backend.pyi +13 -13
  27. metaflow-stubs/packaging_sys/utils.pyi +3 -6
  28. metaflow-stubs/packaging_sys/v1.pyi +5 -5
  29. metaflow-stubs/parameters.pyi +6 -9
  30. metaflow-stubs/plugins/__init__.pyi +13 -13
  31. metaflow-stubs/plugins/airflow/__init__.pyi +2 -2
  32. metaflow-stubs/plugins/airflow/airflow_decorator.pyi +2 -2
  33. metaflow-stubs/plugins/airflow/airflow_utils.pyi +2 -2
  34. metaflow-stubs/plugins/airflow/exception.pyi +2 -2
  35. metaflow-stubs/plugins/airflow/sensors/__init__.pyi +2 -2
  36. metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi +2 -2
  37. metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi +2 -2
  38. metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi +2 -2
  39. metaflow-stubs/plugins/argo/__init__.pyi +2 -2
  40. metaflow-stubs/plugins/argo/argo_client.pyi +2 -2
  41. metaflow-stubs/plugins/argo/argo_events.pyi +2 -2
  42. metaflow-stubs/plugins/argo/argo_workflows.pyi +3 -3
  43. metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +3 -3
  44. metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +5 -3
  45. metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi +11 -9
  46. metaflow-stubs/plugins/argo/exit_hooks.pyi +4 -7
  47. metaflow-stubs/plugins/aws/__init__.pyi +3 -3
  48. metaflow-stubs/plugins/aws/aws_client.pyi +2 -2
  49. metaflow-stubs/plugins/aws/aws_utils.pyi +2 -2
  50. metaflow-stubs/plugins/aws/batch/__init__.pyi +2 -2
  51. metaflow-stubs/plugins/aws/batch/batch.pyi +2 -2
  52. metaflow-stubs/plugins/aws/batch/batch_client.pyi +2 -2
  53. metaflow-stubs/plugins/aws/batch/batch_decorator.pyi +2 -2
  54. metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi +2 -2
  55. metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi +4 -4
  56. metaflow-stubs/plugins/aws/step_functions/__init__.pyi +2 -2
  57. metaflow-stubs/plugins/aws/step_functions/dynamo_db_client.pyi +2 -2
  58. metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi +2 -2
  59. metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi +2 -2
  60. metaflow-stubs/plugins/aws/step_functions/step_functions.pyi +2 -2
  61. metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi +2 -2
  62. metaflow-stubs/plugins/aws/step_functions/step_functions_decorator.pyi +2 -2
  63. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi +6 -4
  64. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi +9 -7
  65. metaflow-stubs/plugins/azure/__init__.pyi +2 -2
  66. metaflow-stubs/plugins/azure/azure_credential.pyi +2 -2
  67. metaflow-stubs/plugins/azure/azure_exceptions.pyi +2 -2
  68. metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi +4 -4
  69. metaflow-stubs/plugins/azure/azure_utils.pyi +2 -2
  70. metaflow-stubs/plugins/azure/blob_service_client_factory.pyi +2 -2
  71. metaflow-stubs/plugins/azure/includefile_support.pyi +2 -2
  72. metaflow-stubs/plugins/cards/__init__.pyi +2 -2
  73. metaflow-stubs/plugins/cards/card_client.pyi +5 -6
  74. metaflow-stubs/plugins/cards/card_creator.pyi +2 -2
  75. metaflow-stubs/plugins/cards/card_datastore.pyi +4 -2
  76. metaflow-stubs/plugins/cards/card_decorator.pyi +3 -3
  77. metaflow-stubs/plugins/cards/card_modules/__init__.pyi +2 -2
  78. metaflow-stubs/plugins/cards/card_modules/basic.pyi +2 -2
  79. metaflow-stubs/plugins/cards/card_modules/card.pyi +2 -2
  80. metaflow-stubs/plugins/cards/card_modules/components.pyi +14 -14
  81. metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi +2 -2
  82. metaflow-stubs/plugins/cards/card_modules/json_viewer.pyi +5 -5
  83. metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi +2 -2
  84. metaflow-stubs/plugins/cards/card_modules/test_cards.pyi +2 -2
  85. metaflow-stubs/plugins/cards/card_resolver.pyi +2 -2
  86. metaflow-stubs/plugins/cards/component_serializer.pyi +2 -2
  87. metaflow-stubs/plugins/cards/exception.pyi +2 -2
  88. metaflow-stubs/plugins/catch_decorator.pyi +3 -3
  89. metaflow-stubs/plugins/datatools/__init__.pyi +2 -2
  90. metaflow-stubs/plugins/datatools/local.pyi +2 -2
  91. metaflow-stubs/plugins/datatools/s3/__init__.pyi +2 -2
  92. metaflow-stubs/plugins/datatools/s3/s3.pyi +27 -35
  93. metaflow-stubs/plugins/datatools/s3/s3tail.pyi +2 -2
  94. metaflow-stubs/plugins/datatools/s3/s3util.pyi +2 -2
  95. metaflow-stubs/plugins/debug_logger.pyi +2 -2
  96. metaflow-stubs/plugins/debug_monitor.pyi +2 -2
  97. metaflow-stubs/plugins/environment_decorator.pyi +2 -2
  98. metaflow-stubs/plugins/events_decorator.pyi +2 -2
  99. metaflow-stubs/plugins/exit_hook/__init__.pyi +2 -2
  100. metaflow-stubs/plugins/exit_hook/exit_hook_decorator.pyi +2 -2
  101. metaflow-stubs/plugins/frameworks/__init__.pyi +2 -2
  102. metaflow-stubs/plugins/frameworks/pytorch.pyi +2 -2
  103. metaflow-stubs/plugins/gcp/__init__.pyi +2 -2
  104. metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi +4 -4
  105. metaflow-stubs/plugins/gcp/gs_exceptions.pyi +2 -2
  106. metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi +2 -2
  107. metaflow-stubs/plugins/gcp/gs_utils.pyi +2 -2
  108. metaflow-stubs/plugins/gcp/includefile_support.pyi +2 -2
  109. metaflow-stubs/plugins/kubernetes/__init__.pyi +2 -2
  110. metaflow-stubs/plugins/kubernetes/kube_utils.pyi +3 -3
  111. metaflow-stubs/plugins/kubernetes/kubernetes.pyi +2 -2
  112. metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +2 -2
  113. metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +2 -2
  114. metaflow-stubs/plugins/kubernetes/kubernetes_job.pyi +2 -2
  115. metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi +2 -2
  116. metaflow-stubs/plugins/kubernetes/spot_monitor_sidecar.pyi +2 -2
  117. metaflow-stubs/plugins/parallel_decorator.pyi +2 -2
  118. metaflow-stubs/plugins/parsers.pyi +2 -2
  119. metaflow-stubs/plugins/project_decorator.pyi +2 -2
  120. metaflow-stubs/plugins/pypi/__init__.pyi +3 -3
  121. metaflow-stubs/plugins/pypi/conda_decorator.pyi +2 -2
  122. metaflow-stubs/plugins/pypi/conda_environment.pyi +5 -5
  123. metaflow-stubs/plugins/pypi/parsers.pyi +2 -2
  124. metaflow-stubs/plugins/pypi/pypi_decorator.pyi +2 -2
  125. metaflow-stubs/plugins/pypi/pypi_environment.pyi +2 -2
  126. metaflow-stubs/plugins/pypi/utils.pyi +2 -2
  127. metaflow-stubs/plugins/resources_decorator.pyi +2 -2
  128. metaflow-stubs/plugins/retry_decorator.pyi +2 -2
  129. metaflow-stubs/plugins/secrets/__init__.pyi +2 -2
  130. metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +3 -3
  131. metaflow-stubs/plugins/secrets/secrets_decorator.pyi +2 -2
  132. metaflow-stubs/plugins/secrets/secrets_func.pyi +3 -5
  133. metaflow-stubs/plugins/secrets/secrets_spec.pyi +2 -2
  134. metaflow-stubs/plugins/secrets/utils.pyi +2 -2
  135. metaflow-stubs/plugins/storage_executor.pyi +2 -2
  136. metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +2 -2
  137. metaflow-stubs/plugins/timeout_decorator.pyi +3 -3
  138. metaflow-stubs/plugins/uv/__init__.pyi +2 -2
  139. metaflow-stubs/plugins/uv/uv_environment.pyi +3 -3
  140. metaflow-stubs/pylint_wrapper.pyi +2 -2
  141. metaflow-stubs/runner/__init__.pyi +2 -2
  142. metaflow-stubs/runner/deployer.pyi +23 -22
  143. metaflow-stubs/runner/deployer_impl.pyi +6 -4
  144. metaflow-stubs/runner/metaflow_runner.pyi +134 -21
  145. metaflow-stubs/runner/nbdeploy.pyi +3 -3
  146. metaflow-stubs/runner/nbrun.pyi +3 -3
  147. metaflow-stubs/runner/subprocess_manager.pyi +8 -8
  148. metaflow-stubs/runner/utils.pyi +5 -6
  149. metaflow-stubs/system/__init__.pyi +2 -2
  150. metaflow-stubs/system/system_logger.pyi +4 -5
  151. metaflow-stubs/system/system_monitor.pyi +3 -3
  152. metaflow-stubs/tagging_util.pyi +2 -2
  153. metaflow-stubs/tuple_util.pyi +2 -2
  154. metaflow-stubs/user_configs/__init__.pyi +2 -2
  155. metaflow-stubs/user_configs/config_options.pyi +5 -6
  156. metaflow-stubs/user_configs/config_parameters.pyi +8 -8
  157. metaflow-stubs/user_decorators/__init__.pyi +2 -2
  158. metaflow-stubs/user_decorators/common.pyi +6 -6
  159. metaflow-stubs/user_decorators/mutable_flow.pyi +8 -9
  160. metaflow-stubs/user_decorators/mutable_step.pyi +6 -9
  161. metaflow-stubs/user_decorators/user_flow_decorator.pyi +7 -8
  162. metaflow-stubs/user_decorators/user_step_decorator.pyi +24 -17
  163. metaflow-stubs/version.pyi +2 -2
  164. {metaflow_stubs-2.18.13.dist-info → metaflow_stubs-2.19.1.dist-info}/METADATA +2 -2
  165. metaflow_stubs-2.19.1.dist-info/RECORD +168 -0
  166. metaflow_stubs-2.18.13.dist-info/RECORD +0 -168
  167. {metaflow_stubs-2.18.13.dist-info → metaflow_stubs-2.19.1.dist-info}/WHEEL +0 -0
  168. {metaflow_stubs-2.18.13.dist-info → metaflow_stubs-2.19.1.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.18.13 #
4
- # Generated on 2025-10-20T17:35:52.545763 #
3
+ # MF version: 2.19.1 #
4
+ # Generated on 2025-10-28T01:39:06.625363 #
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.18.13 #
4
- # Generated on 2025-10-20T17:35:52.555140 #
3
+ # MF version: 2.19.1 #
4
+ # Generated on 2025-10-28T01:39:06.636140 #
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.18.13 #
4
- # Generated on 2025-10-20T17:35:52.554084 #
3
+ # MF version: 2.19.1 #
4
+ # Generated on 2025-10-28T01:39:06.634847 #
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.18.13 #
4
- # Generated on 2025-10-20T17:35:52.587564 #
3
+ # MF version: 2.19.1 #
4
+ # Generated on 2025-10-28T01:39:06.665751 #
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.18.13 #
4
- # Generated on 2025-10-20T17:35:52.546029 #
3
+ # MF version: 2.19.1 #
4
+ # Generated on 2025-10-28T01:39:06.625637 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -15,6 +15,7 @@ from .core import default_namespace as default_namespace
15
15
  from .core import metadata as metadata
16
16
  from .core import get_metadata as get_metadata
17
17
  from .core import default_metadata as default_metadata
18
+ from .core import inspect_spin as inspect_spin
18
19
  from .core import Metaflow as Metaflow
19
20
  from .core import Flow as Flow
20
21
  from .core import Run as Run
@@ -1,20 +1,19 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.13 #
4
- # Generated on 2025-10-20T17:35:52.551194 #
3
+ # MF version: 2.19.1 #
4
+ # Generated on 2025-10-28T01:39:06.631506 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import metaflow
10
9
  import typing
10
+ import metaflow
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.events
13
- import datetime
14
- import typing
15
- import metaflow.client.core
16
12
  import tarfile
13
+ import typing
17
14
  import tempfile
15
+ import datetime
16
+ import metaflow.client.core
18
17
 
19
18
  from ..metaflow_current import current as current
20
19
  from ..events import Trigger as Trigger
@@ -126,7 +125,7 @@ def default_metadata() -> str:
126
125
  """
127
126
  ...
128
127
 
129
- def namespace(ns: typing.Optional[str]) -> typing.Optional[str]:
128
+ def namespace(ns: str | None) -> str | None:
130
129
  """
131
130
  Switch namespace to the one provided.
132
131
 
@@ -146,7 +145,7 @@ def namespace(ns: typing.Optional[str]) -> typing.Optional[str]:
146
145
  """
147
146
  ...
148
147
 
149
- def get_namespace() -> typing.Optional[str]:
148
+ def get_namespace() -> str | None:
150
149
  """
151
150
  Return the current namespace that is currently being used to filter objects.
152
151
 
@@ -171,6 +170,18 @@ def default_namespace() -> str:
171
170
  """
172
171
  ...
173
172
 
173
+ def inspect_spin(datastore_root: str = '.'):
174
+ """
175
+ Set metadata provider to spin metadata so that users can inspect spin
176
+ steps, tasks, and artifacts.
177
+
178
+ Parameters
179
+ ----------
180
+ datastore_root : str, default "."
181
+ The root path to the spin datastore.
182
+ """
183
+ ...
184
+
174
185
  class MetaflowObject(object, metaclass=type):
175
186
  """
176
187
  Base class for all Metaflow objects.
@@ -211,7 +222,7 @@ class MetaflowObject(object, metaclass=type):
211
222
  Pathspec of the original object this object was cloned from (in the case of a resume).
212
223
  None if not applicable.
213
224
  """
214
- 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):
225
+ def __init__(self, pathspec: str | None = None, attempt: int | None = None, _object: ForwardRef('MetaflowObject') | None = None, _parent: ForwardRef('MetaflowObject') | None = None, _namespace_check: bool = True, _metaflow: ForwardRef('Metaflow') | None = None, _current_namespace: str | None = None, _current_metadata: str | None = None):
215
226
  ...
216
227
  def __iter__(self) -> typing.Iterator["MetaflowObject"]:
217
228
  """
@@ -344,7 +355,7 @@ class MetaflowObject(object, metaclass=type):
344
355
  """
345
356
  ...
346
357
  @property
347
- def origin_pathspec(self) -> typing.Optional[str]:
358
+ def origin_pathspec(self) -> str | None:
348
359
  """
349
360
  The pathspec of the object from which the current object was cloned.
350
361
 
@@ -354,7 +365,7 @@ class MetaflowObject(object, metaclass=type):
354
365
  """
355
366
  ...
356
367
  @property
357
- def parent(self) -> typing.Optional["MetaflowObject"]:
368
+ def parent(self) -> ForwardRef('MetaflowObject') | None:
358
369
  """
359
370
  Returns the parent object of this object or None if none exists.
360
371
 
@@ -672,7 +683,27 @@ class Task(MetaflowObject, metaclass=type):
672
683
  environment_info : Dict[str, str]
673
684
  Information about the execution environment.
674
685
  """
675
- def __init__(self, *args, **kwargs):
686
+ @property
687
+ def parent_task_pathspecs(self) -> typing.Iterator[str]:
688
+ """
689
+ Yields pathspecs of all parent tasks of the current task.
690
+
691
+ Yields
692
+ ------
693
+ str
694
+ Pathspec of the parent task of the current task
695
+ """
696
+ ...
697
+ @property
698
+ def child_task_pathspecs(self) -> typing.Iterator[str]:
699
+ """
700
+ Yields pathspecs of all child tasks of the current task.
701
+
702
+ Yields
703
+ ------
704
+ str
705
+ Pathspec of the child task of the current task
706
+ """
676
707
  ...
677
708
  @property
678
709
  def parent_tasks(self) -> typing.Iterator["Task"]:
@@ -688,7 +719,7 @@ class Task(MetaflowObject, metaclass=type):
688
719
  @property
689
720
  def child_tasks(self) -> typing.Iterator["Task"]:
690
721
  """
691
- Yield all child tasks of the current task if one exists.
722
+ Yields all child tasks of the current task if one exists.
692
723
 
693
724
  Yields
694
725
  ------
@@ -728,7 +759,7 @@ class Task(MetaflowObject, metaclass=type):
728
759
  """
729
760
  ...
730
761
  @property
731
- def index(self) -> typing.Optional[int]:
762
+ def index(self) -> int | None:
732
763
  """
733
764
  Returns the index of the innermost foreach loop if this task is run inside at least
734
765
  one foreach.
@@ -805,7 +836,7 @@ class Task(MetaflowObject, metaclass=type):
805
836
  """
806
837
  ...
807
838
  @property
808
- def exception(self) -> typing.Optional[typing.Any]:
839
+ def exception(self) -> typing.Any | None:
809
840
  """
810
841
  Returns the exception that caused the task to fail, if any.
811
842
 
@@ -820,7 +851,7 @@ class Task(MetaflowObject, metaclass=type):
820
851
  """
821
852
  ...
822
853
  @property
823
- def finished_at(self) -> typing.Optional[datetime.datetime]:
854
+ def finished_at(self) -> datetime.datetime | None:
824
855
  """
825
856
  Returns the datetime object of when the task finished (successfully or not).
826
857
 
@@ -834,7 +865,7 @@ class Task(MetaflowObject, metaclass=type):
834
865
  """
835
866
  ...
836
867
  @property
837
- def runtime_name(self) -> typing.Optional[str]:
868
+ def runtime_name(self) -> str | None:
838
869
  """
839
870
  Returns the name of the runtime this task executed on.
840
871
 
@@ -926,7 +957,7 @@ class Task(MetaflowObject, metaclass=type):
926
957
  """
927
958
  ...
928
959
  @property
929
- def code(self) -> typing.Optional[metaflow.client.core.MetaflowCode]:
960
+ def code(self) -> metaflow.client.core.MetaflowCode | None:
930
961
  """
931
962
  Returns the MetaflowCode object for this task, if present.
932
963
 
@@ -953,7 +984,7 @@ class Task(MetaflowObject, metaclass=type):
953
984
  Dictionary describing the environment
954
985
  """
955
986
  ...
956
- def loglines(self, stream: str, as_unicode: bool = True, meta_dict: typing.Optional[typing.Dict[str, typing.Any]] = None) -> typing.Iterator[typing.Tuple[datetime.datetime, str]]:
987
+ def loglines(self, stream: str, as_unicode: bool = True, meta_dict: typing.Dict[str, typing.Any] | None = None) -> typing.Iterator[typing.Tuple[datetime.datetime, str]]:
957
988
  """
958
989
  Return an iterator over (utc_timestamp, logline) tuples.
959
990
 
@@ -1026,7 +1057,7 @@ class Step(MetaflowObject, metaclass=type):
1026
1057
  Information about the execution environment.
1027
1058
  """
1028
1059
  @property
1029
- def task(self) -> typing.Optional[metaflow.client.core.Task]:
1060
+ def task(self) -> metaflow.client.core.Task | None:
1030
1061
  """
1031
1062
  Returns a Task object belonging to this step.
1032
1063
 
@@ -1060,7 +1091,7 @@ class Step(MetaflowObject, metaclass=type):
1060
1091
  """
1061
1092
  ...
1062
1093
  @property
1063
- def control_task(self) -> typing.Optional[metaflow.client.core.Task]:
1094
+ def control_task(self) -> metaflow.client.core.Task | None:
1064
1095
  """
1065
1096
  [Unpublished API - use with caution!]
1066
1097
 
@@ -1127,7 +1158,7 @@ class Step(MetaflowObject, metaclass=type):
1127
1158
  def __setstate__(self, state):
1128
1159
  ...
1129
1160
  @property
1130
- def finished_at(self) -> typing.Optional[datetime.datetime]:
1161
+ def finished_at(self) -> datetime.datetime | None:
1131
1162
  """
1132
1163
  Returns the datetime object of when the step finished (successfully or not).
1133
1164
 
@@ -1141,7 +1172,7 @@ class Step(MetaflowObject, metaclass=type):
1141
1172
  """
1142
1173
  ...
1143
1174
  @property
1144
- def environment_info(self) -> typing.Optional[typing.Dict[str, typing.Any]]:
1175
+ def environment_info(self) -> typing.Dict[str, typing.Any] | None:
1145
1176
  """
1146
1177
  Returns information about the environment that was used to execute this step. As an
1147
1178
  example, if the Conda environment is selected, this will return information about the
@@ -1223,7 +1254,7 @@ class Run(MetaflowObject, metaclass=type):
1223
1254
  """
1224
1255
  ...
1225
1256
  @property
1226
- def code(self) -> typing.Optional[metaflow.client.core.MetaflowCode]:
1257
+ def code(self) -> metaflow.client.core.MetaflowCode | None:
1227
1258
  """
1228
1259
  Returns the MetaflowCode object for this run, if present.
1229
1260
  Code is packed if atleast one `Step` runs remotely, else None is returned.
@@ -1235,7 +1266,7 @@ class Run(MetaflowObject, metaclass=type):
1235
1266
  """
1236
1267
  ...
1237
1268
  @property
1238
- def data(self) -> typing.Optional[metaflow.client.core.MetaflowData]:
1269
+ def data(self) -> metaflow.client.core.MetaflowData | None:
1239
1270
  """
1240
1271
  Returns a container of data artifacts produced by this run.
1241
1272
 
@@ -1279,7 +1310,7 @@ class Run(MetaflowObject, metaclass=type):
1279
1310
  """
1280
1311
  ...
1281
1312
  @property
1282
- def finished_at(self) -> typing.Optional[datetime.datetime]:
1313
+ def finished_at(self) -> datetime.datetime | None:
1283
1314
  """
1284
1315
  Returns the datetime object of when the run finished (successfully or not).
1285
1316
 
@@ -1293,7 +1324,7 @@ class Run(MetaflowObject, metaclass=type):
1293
1324
  """
1294
1325
  ...
1295
1326
  @property
1296
- def end_task(self) -> typing.Optional[metaflow.client.core.Task]:
1327
+ def end_task(self) -> metaflow.client.core.Task | None:
1297
1328
  """
1298
1329
  Returns the Task corresponding to the 'end' step.
1299
1330
 
@@ -1418,7 +1449,7 @@ class Run(MetaflowObject, metaclass=type):
1418
1449
  def __setstate__(self, state):
1419
1450
  ...
1420
1451
  @property
1421
- def trigger(self) -> typing.Optional[metaflow.events.Trigger]:
1452
+ def trigger(self) -> metaflow.events.Trigger | None:
1422
1453
  """
1423
1454
  Returns a container of events that triggered this run.
1424
1455
 
@@ -1447,7 +1478,7 @@ class Flow(MetaflowObject, metaclass=type):
1447
1478
  def __init__(self, *args, **kwargs):
1448
1479
  ...
1449
1480
  @property
1450
- def latest_run(self) -> typing.Optional[metaflow.client.core.Run]:
1481
+ def latest_run(self) -> metaflow.client.core.Run | None:
1451
1482
  """
1452
1483
  Returns the latest run (either in progress or completed) of this flow.
1453
1484
 
@@ -1461,7 +1492,7 @@ class Flow(MetaflowObject, metaclass=type):
1461
1492
  """
1462
1493
  ...
1463
1494
  @property
1464
- def latest_successful_run(self) -> typing.Optional[metaflow.client.core.Run]:
1495
+ def latest_successful_run(self) -> metaflow.client.core.Run | None:
1465
1496
  """
1466
1497
  Returns the latest successful run of this flow.
1467
1498
 
@@ -1543,7 +1574,7 @@ class Metaflow(object, metaclass=type):
1543
1574
  flows present in the current namespace will be returned. A `Flow` is present in a namespace
1544
1575
  if it has at least one run in the namespace.
1545
1576
  """
1546
- def __init__(self, _current_metadata: typing.Optional[str] = None):
1577
+ def __init__(self, _current_metadata: str | None = None):
1547
1578
  ...
1548
1579
  @property
1549
1580
  def flows(self) -> typing.List[metaflow.client.core.Flow]:
@@ -1,16 +1,19 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.13 #
4
- # Generated on 2025-10-20T17:35:52.586257 #
3
+ # MF version: 2.19.1 #
4
+ # Generated on 2025-10-28T01:39:06.665373 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
9
  import metaflow
10
+ import abc
10
11
  import typing
11
12
  if typing.TYPE_CHECKING:
12
13
  import metaflow.datastore.content_addressed_store
13
14
  import metaflow.exception
15
+ import abc
16
+ import metaflow.datastore.flow_datastore
14
17
 
15
18
  from ..exception import MetaflowException as MetaflowException
16
19
 
@@ -20,8 +23,6 @@ CLIENT_CACHE_MAX_SIZE: int
20
23
 
21
24
  CLIENT_CACHE_MAX_FLOWDATASTORE_COUNT: int
22
25
 
23
- CLIENT_CACHE_MAX_TASKDATASTORE_COUNT: int
24
-
25
26
  DATASTORES: list
26
27
 
27
28
  NEW_FILE_QUARANTINE: int
@@ -70,6 +71,21 @@ class FileCache(object, metaclass=type):
70
71
  ...
71
72
  def read_file(self, path):
72
73
  ...
74
+ @staticmethod
75
+ def flow_ds_id(ds_type, ds_root, flow_name):
76
+ ...
77
+ @staticmethod
78
+ def task_ds_id(ds_type, ds_root, flow_name, run_id, step_name, task_id, attempt):
79
+ ...
80
+ ...
81
+
82
+ class TaskMetadataCache(metaflow.datastore.flow_datastore.MetadataCache, metaclass=abc.ABCMeta):
83
+ def __init__(self, filecache, ds_type, ds_root, flow_name):
84
+ ...
85
+ def load_metadata(self, run_id, step_name, task_id, attempt):
86
+ ...
87
+ def store_metadata(self, run_id, step_name, task_id, attempt, metadata_dict):
88
+ ...
73
89
  ...
74
90
 
75
91
  class FileBlobCache(metaflow.datastore.content_addressed_store.BlobCache, metaclass=type):
metaflow-stubs/events.pyi CHANGED
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.13 #
4
- # Generated on 2025-10-20T17:35:52.539201 #
3
+ # MF version: 2.19.1 #
4
+ # Generated on 2025-10-28T01:39:06.618925 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,7 +9,6 @@ from __future__ import annotations
9
9
  import typing
10
10
  if typing.TYPE_CHECKING:
11
11
  import metaflow
12
- import metaflow.events
13
12
 
14
13
 
15
14
  TYPE_CHECKING: bool
@@ -58,7 +57,7 @@ class Trigger(object, metaclass=type):
58
57
  def from_runs(cls, run_objs: typing.List["metaflow.Run"]):
59
58
  ...
60
59
  @property
61
- def event(self) -> typing.Optional[metaflow.events.MetaflowEvent]:
60
+ def event(self) -> metaflow.events.MetaflowEvent | None:
62
61
  """
63
62
  The `MetaflowEvent` object corresponding to the triggering event.
64
63
 
@@ -71,7 +70,7 @@ class Trigger(object, metaclass=type):
71
70
  """
72
71
  ...
73
72
  @property
74
- def events(self) -> typing.Optional[typing.List[metaflow.events.MetaflowEvent]]:
73
+ def events(self) -> typing.List[metaflow.events.MetaflowEvent] | None:
75
74
  """
76
75
  The list of `MetaflowEvent` objects correspondings to all the triggering events.
77
76
 
@@ -82,7 +81,7 @@ class Trigger(object, metaclass=type):
82
81
  """
83
82
  ...
84
83
  @property
85
- def run(self) -> typing.Optional["metaflow.Run"]:
84
+ def run(self) -> ForwardRef('metaflow.Run') | None:
86
85
  """
87
86
  The corresponding `Run` object if the triggering event is a Metaflow run.
88
87
 
@@ -96,7 +95,7 @@ class Trigger(object, metaclass=type):
96
95
  """
97
96
  ...
98
97
  @property
99
- def runs(self) -> typing.Optional[typing.List["metaflow.Run"]]:
98
+ def runs(self) -> typing.List[ForwardRef('metaflow.Run')] | None:
100
99
  """
101
100
  The list of `Run` objects in the triggering events.
102
101
  Returns `None` if none of the triggering events are `Run` objects.
@@ -107,7 +106,7 @@ class Trigger(object, metaclass=type):
107
106
  List of runs that triggered this run, if applicable.
108
107
  """
109
108
  ...
110
- def __getitem__(self, key: str) -> typing.Union["metaflow.Run", metaflow.events.MetaflowEvent]:
109
+ def __getitem__(self, key: str) -> ForwardRef('metaflow.Run') | metaflow.events.MetaflowEvent:
111
110
  """
112
111
  If triggering events are runs, `key` corresponds to the flow name of the triggering run.
113
112
  Otherwise, `key` corresponds to the event name and a `MetaflowEvent` object is returned.
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.13 #
4
- # Generated on 2025-10-20T17:35:52.527498 #
3
+ # MF version: 2.19.1 #
4
+ # Generated on 2025-10-28T01:39:06.605923 #
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.18.13 #
4
- # Generated on 2025-10-20T17:35:52.533234 #
3
+ # MF version: 2.19.1 #
4
+ # Generated on 2025-10-28T01:39:06.611778 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,11 +9,10 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import typing
13
12
  import metaflow.unbounded_foreach
14
13
  import metaflow.exception
15
- import metaflow.datastore.inputs
16
14
  import metaflow.flowspec
15
+ import metaflow.datastore.inputs
17
16
 
18
17
  from . import parameters as parameters
19
18
  from .parameters import DelayedEvaluationParameter as DelayedEvaluationParameter
@@ -108,7 +107,7 @@ class FlowSpec(object, metaclass=FlowSpecMeta):
108
107
  """
109
108
  ...
110
109
  @property
111
- def index(self) -> typing.Optional[int]:
110
+ def index(self) -> int | None:
112
111
  """
113
112
  The index of this foreach branch.
114
113
 
@@ -126,7 +125,7 @@ class FlowSpec(object, metaclass=FlowSpecMeta):
126
125
  """
127
126
  ...
128
127
  @property
129
- def input(self) -> typing.Optional[typing.Any]:
128
+ def input(self) -> typing.Any | None:
130
129
  """
131
130
  The value of the foreach artifact in this foreach branch.
132
131
 
@@ -143,7 +142,7 @@ class FlowSpec(object, metaclass=FlowSpecMeta):
143
142
  Input passed to the foreach task.
144
143
  """
145
144
  ...
146
- def foreach_stack(self) -> typing.Optional[typing.List[typing.Tuple[int, int, typing.Any]]]:
145
+ def foreach_stack(self) -> typing.List[typing.Tuple[int, int, typing.Any]] | None:
147
146
  """
148
147
  Returns the current stack of foreach indexes and values for the current step.
149
148
 
@@ -191,7 +190,7 @@ class FlowSpec(object, metaclass=FlowSpecMeta):
191
190
  An array describing the current stack of foreach steps.
192
191
  """
193
192
  ...
194
- def merge_artifacts(self, inputs: metaflow.datastore.inputs.Inputs, exclude: typing.Optional[typing.List[str]] = None, include: typing.Optional[typing.List[str]] = None):
193
+ def merge_artifacts(self, inputs: metaflow.datastore.inputs.Inputs, exclude: typing.List[str] | None = None, include: typing.List[str] | None = None):
195
194
  """
196
195
  Helper function for merging artifacts in a join step.
197
196
 
@@ -1 +1 @@
1
- 2.18.13 2025-10-20T17:35:52.521947
1
+ 2.19.1 2025-10-28T01:39:06.600277
@@ -1,17 +1,17 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.13 #
4
- # Generated on 2025-10-20T17:35:52.544960 #
3
+ # MF version: 2.19.1 #
4
+ # Generated on 2025-10-28T01:39:06.624542 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
- import metaflow
10
9
  import typing
10
+ import metaflow
11
11
  if typing.TYPE_CHECKING:
12
+ import typing
12
13
  import metaflow._vendor.click.types
13
14
  import metaflow.parameters
14
- import typing
15
15
 
16
16
  from ._vendor import click as click
17
17
  from ._vendor import yaml as yaml
@@ -86,7 +86,7 @@ class IncludeFile(metaflow.parameters.Parameter, metaclass=type):
86
86
  be able to parse the file contents. If the name starts with a ".", it is assumed
87
87
  to be relative to "metaflow".
88
88
  """
89
- 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, parser: typing.Union[str, typing.Callable[[str], typing.Any], None] = None, **kwargs: typing.Dict[str, str]):
89
+ def __init__(self, name: str, required: bool | None = None, is_text: bool | None = None, encoding: str | None = None, help: str | None = None, parser: str | typing.Callable[[str], typing.Any] | None = None, **kwargs: typing.Dict[str, str]):
90
90
  ...
91
91
  def init(self, ignore_errors = False):
92
92
  ...
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
- # MF version: 2.18.13 #
4
- # Generated on 2025-10-20T17:35:52.522769 #
3
+ # MF version: 2.19.1 #
4
+ # Generated on 2025-10-28T01:39:06.601038 #
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.18.13 #
4
- # Generated on 2025-10-20T17:35:52.532049 #
3
+ # MF version: 2.19.1 #
4
+ # Generated on 2025-10-28T01:39:06.610499 #
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.18.13 #
4
- # Generated on 2025-10-20T17:35:52.567216 #
3
+ # MF version: 2.19.1 #
4
+ # Generated on 2025-10-28T01:39:06.645661 #
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.18.13 #
4
- # Generated on 2025-10-20T17:35:52.566674 #
3
+ # MF version: 2.19.1 #
4
+ # Generated on 2025-10-28T01:39:06.645093 #
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.18.13 #
4
- # Generated on 2025-10-20T17:35:52.566912 #
3
+ # MF version: 2.19.1 #
4
+ # Generated on 2025-10-28T01:39:06.645348 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations