ob-metaflow-stubs 6.0.3.170__py2.py3-none-any.whl → 6.0.3.172__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 (209) hide show
  1. metaflow-stubs/__init__.pyi +645 -641
  2. metaflow-stubs/cards.pyi +1 -1
  3. metaflow-stubs/cli.pyi +1 -1
  4. metaflow-stubs/cli_components/__init__.pyi +1 -1
  5. metaflow-stubs/cli_components/utils.pyi +1 -1
  6. metaflow-stubs/client/__init__.pyi +1 -1
  7. metaflow-stubs/client/core.pyi +4 -4
  8. metaflow-stubs/client/filecache.pyi +2 -2
  9. metaflow-stubs/events.pyi +1 -1
  10. metaflow-stubs/exception.pyi +1 -1
  11. metaflow-stubs/flowspec.pyi +3 -3
  12. metaflow-stubs/generated_for.txt +1 -1
  13. metaflow-stubs/includefile.pyi +2 -2
  14. metaflow-stubs/info_file.pyi +1 -1
  15. metaflow-stubs/metadata_provider/__init__.pyi +1 -1
  16. metaflow-stubs/metadata_provider/heartbeat.pyi +1 -1
  17. metaflow-stubs/metadata_provider/metadata.pyi +1 -1
  18. metaflow-stubs/metadata_provider/util.pyi +1 -1
  19. metaflow-stubs/metaflow_config.pyi +1 -1
  20. metaflow-stubs/metaflow_current.pyi +125 -125
  21. metaflow-stubs/metaflow_git.pyi +1 -1
  22. metaflow-stubs/mf_extensions/__init__.pyi +1 -1
  23. metaflow-stubs/mf_extensions/obcheckpoint/__init__.pyi +1 -1
  24. metaflow-stubs/mf_extensions/obcheckpoint/plugins/__init__.pyi +1 -1
  25. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/__init__.pyi +1 -1
  26. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/__init__.pyi +1 -1
  27. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/async_cards.pyi +1 -1
  28. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/deco_injection_mixin.pyi +1 -1
  29. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/extra_components.pyi +2 -2
  30. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/__init__.pyi +1 -1
  31. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/__init__.pyi +1 -1
  32. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/checkpoint_lister.pyi +3 -3
  33. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/lineage_card.pyi +1 -1
  34. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/checkpoint_storage.pyi +3 -3
  35. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/constructors.pyi +1 -1
  36. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/core.pyi +3 -3
  37. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/decorator.pyi +2 -2
  38. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/exceptions.pyi +1 -1
  39. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/final_api.pyi +1 -1
  40. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/lineage.pyi +1 -1
  41. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/__init__.pyi +1 -1
  42. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/context.pyi +3 -3
  43. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/core.pyi +2 -2
  44. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/decorator.pyi +1 -1
  45. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/exceptions.pyi +1 -1
  46. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/task_utils.pyi +2 -2
  47. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/utils.pyi +1 -1
  48. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastructures.pyi +1 -1
  49. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/exceptions.pyi +1 -1
  50. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/__init__.pyi +1 -1
  51. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/decorator.pyi +2 -2
  52. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/__init__.pyi +1 -1
  53. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/core.pyi +2 -2
  54. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/exceptions.pyi +1 -1
  55. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/model_storage.pyi +2 -2
  56. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/__init__.pyi +1 -1
  57. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/flowspec_utils.pyi +1 -1
  58. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/general.pyi +1 -1
  59. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/identity_utils.pyi +2 -2
  60. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/__init__.pyi +1 -1
  61. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/base.pyi +1 -1
  62. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/tar.pyi +1 -1
  63. metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/tar_utils.pyi +1 -1
  64. metaflow-stubs/mf_extensions/outerbounds/__init__.pyi +1 -1
  65. metaflow-stubs/mf_extensions/outerbounds/plugins/__init__.pyi +1 -1
  66. metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/__init__.pyi +1 -1
  67. metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/injector.pyi +1 -1
  68. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/__init__.pyi +1 -1
  69. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/coreweave.pyi +1 -1
  70. metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/nebius.pyi +1 -1
  71. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/__init__.pyi +1 -1
  72. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/constants.pyi +12 -0
  73. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/exceptions.pyi +30 -0
  74. metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/ollama.pyi +15 -7
  75. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/__init__.pyi +1 -1
  76. metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/snowflake.pyi +1 -1
  77. metaflow-stubs/mf_extensions/outerbounds/profilers/__init__.pyi +1 -1
  78. metaflow-stubs/mf_extensions/outerbounds/profilers/gpu.pyi +1 -1
  79. metaflow-stubs/mf_extensions/outerbounds/remote_config.pyi +1 -1
  80. metaflow-stubs/mf_extensions/outerbounds/toplevel/__init__.pyi +1 -1
  81. metaflow-stubs/mf_extensions/outerbounds/toplevel/global_aliases_for_metaflow_package.pyi +1 -1
  82. metaflow-stubs/multicore_utils.pyi +1 -1
  83. metaflow-stubs/parameters.pyi +2 -2
  84. metaflow-stubs/plugins/__init__.pyi +12 -11
  85. metaflow-stubs/plugins/airflow/__init__.pyi +1 -1
  86. metaflow-stubs/plugins/airflow/airflow_utils.pyi +1 -1
  87. metaflow-stubs/plugins/airflow/exception.pyi +1 -1
  88. metaflow-stubs/plugins/airflow/sensors/__init__.pyi +1 -1
  89. metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi +1 -1
  90. metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi +1 -1
  91. metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi +1 -1
  92. metaflow-stubs/plugins/argo/__init__.pyi +1 -1
  93. metaflow-stubs/plugins/argo/argo_client.pyi +1 -1
  94. metaflow-stubs/plugins/argo/argo_events.pyi +1 -1
  95. metaflow-stubs/plugins/argo/argo_workflows.pyi +1 -1
  96. metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +2 -2
  97. metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +1 -1
  98. metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi +1 -1
  99. metaflow-stubs/plugins/aws/__init__.pyi +1 -1
  100. metaflow-stubs/plugins/aws/aws_client.pyi +1 -1
  101. metaflow-stubs/plugins/aws/aws_utils.pyi +1 -1
  102. metaflow-stubs/plugins/aws/batch/__init__.pyi +1 -1
  103. metaflow-stubs/plugins/aws/batch/batch.pyi +1 -1
  104. metaflow-stubs/plugins/aws/batch/batch_client.pyi +1 -1
  105. metaflow-stubs/plugins/aws/batch/batch_decorator.pyi +1 -1
  106. metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi +1 -1
  107. metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi +1 -1
  108. metaflow-stubs/plugins/aws/step_functions/__init__.pyi +1 -1
  109. metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi +1 -1
  110. metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi +1 -1
  111. metaflow-stubs/plugins/aws/step_functions/step_functions.pyi +1 -1
  112. metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi +1 -1
  113. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi +1 -1
  114. metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi +1 -1
  115. metaflow-stubs/plugins/azure/__init__.pyi +1 -1
  116. metaflow-stubs/plugins/azure/azure_credential.pyi +1 -1
  117. metaflow-stubs/plugins/azure/azure_exceptions.pyi +1 -1
  118. metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi +1 -1
  119. metaflow-stubs/plugins/azure/azure_utils.pyi +1 -1
  120. metaflow-stubs/plugins/azure/blob_service_client_factory.pyi +1 -1
  121. metaflow-stubs/plugins/azure/includefile_support.pyi +1 -1
  122. metaflow-stubs/plugins/cards/__init__.pyi +1 -1
  123. metaflow-stubs/plugins/cards/card_client.pyi +2 -2
  124. metaflow-stubs/plugins/cards/card_creator.pyi +1 -1
  125. metaflow-stubs/plugins/cards/card_datastore.pyi +1 -1
  126. metaflow-stubs/plugins/cards/card_decorator.pyi +1 -1
  127. metaflow-stubs/plugins/cards/card_modules/__init__.pyi +1 -1
  128. metaflow-stubs/plugins/cards/card_modules/basic.pyi +1 -1
  129. metaflow-stubs/plugins/cards/card_modules/card.pyi +1 -1
  130. metaflow-stubs/plugins/cards/card_modules/components.pyi +3 -3
  131. metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi +1 -1
  132. metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi +1 -1
  133. metaflow-stubs/plugins/cards/card_modules/test_cards.pyi +1 -1
  134. metaflow-stubs/plugins/cards/card_resolver.pyi +1 -1
  135. metaflow-stubs/plugins/cards/component_serializer.pyi +1 -1
  136. metaflow-stubs/plugins/cards/exception.pyi +1 -1
  137. metaflow-stubs/plugins/catch_decorator.pyi +1 -1
  138. metaflow-stubs/plugins/datatools/__init__.pyi +1 -1
  139. metaflow-stubs/plugins/datatools/local.pyi +1 -1
  140. metaflow-stubs/plugins/datatools/s3/__init__.pyi +1 -1
  141. metaflow-stubs/plugins/datatools/s3/s3.pyi +2 -2
  142. metaflow-stubs/plugins/datatools/s3/s3tail.pyi +1 -1
  143. metaflow-stubs/plugins/datatools/s3/s3util.pyi +1 -1
  144. metaflow-stubs/plugins/debug_logger.pyi +1 -1
  145. metaflow-stubs/plugins/debug_monitor.pyi +1 -1
  146. metaflow-stubs/plugins/environment_decorator.pyi +1 -1
  147. metaflow-stubs/plugins/events_decorator.pyi +1 -1
  148. metaflow-stubs/plugins/frameworks/__init__.pyi +1 -1
  149. metaflow-stubs/plugins/frameworks/pytorch.pyi +1 -1
  150. metaflow-stubs/plugins/gcp/__init__.pyi +1 -1
  151. metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi +1 -1
  152. metaflow-stubs/plugins/gcp/gs_exceptions.pyi +1 -1
  153. metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi +1 -1
  154. metaflow-stubs/plugins/gcp/gs_utils.pyi +1 -1
  155. metaflow-stubs/plugins/gcp/includefile_support.pyi +1 -1
  156. metaflow-stubs/plugins/kubernetes/__init__.pyi +1 -1
  157. metaflow-stubs/plugins/kubernetes/kube_utils.pyi +1 -1
  158. metaflow-stubs/plugins/kubernetes/kubernetes.pyi +1 -1
  159. metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +1 -1
  160. metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +1 -1
  161. metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi +1 -1
  162. metaflow-stubs/plugins/kubernetes/spot_monitor_sidecar.pyi +1 -1
  163. metaflow-stubs/plugins/ollama/__init__.pyi +18 -10
  164. metaflow-stubs/plugins/parallel_decorator.pyi +1 -1
  165. metaflow-stubs/plugins/perimeters.pyi +1 -1
  166. metaflow-stubs/plugins/project_decorator.pyi +1 -1
  167. metaflow-stubs/plugins/pypi/__init__.pyi +1 -1
  168. metaflow-stubs/plugins/pypi/conda_decorator.pyi +1 -1
  169. metaflow-stubs/plugins/pypi/conda_environment.pyi +3 -3
  170. metaflow-stubs/plugins/pypi/parsers.pyi +1 -1
  171. metaflow-stubs/plugins/pypi/pypi_decorator.pyi +1 -1
  172. metaflow-stubs/plugins/pypi/pypi_environment.pyi +1 -1
  173. metaflow-stubs/plugins/pypi/utils.pyi +1 -1
  174. metaflow-stubs/plugins/resources_decorator.pyi +1 -1
  175. metaflow-stubs/plugins/retry_decorator.pyi +1 -1
  176. metaflow-stubs/plugins/secrets/__init__.pyi +1 -1
  177. metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +1 -1
  178. metaflow-stubs/plugins/secrets/secrets_decorator.pyi +1 -1
  179. metaflow-stubs/plugins/snowflake/__init__.pyi +1 -1
  180. metaflow-stubs/plugins/storage_executor.pyi +1 -1
  181. metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +2 -2
  182. metaflow-stubs/plugins/timeout_decorator.pyi +1 -1
  183. metaflow-stubs/plugins/torchtune/__init__.pyi +39 -0
  184. metaflow-stubs/plugins/uv/__init__.pyi +1 -1
  185. metaflow-stubs/plugins/uv/uv_environment.pyi +2 -2
  186. metaflow-stubs/profilers/__init__.pyi +1 -1
  187. metaflow-stubs/pylint_wrapper.pyi +1 -1
  188. metaflow-stubs/runner/__init__.pyi +1 -1
  189. metaflow-stubs/runner/deployer.pyi +5 -5
  190. metaflow-stubs/runner/deployer_impl.pyi +2 -2
  191. metaflow-stubs/runner/metaflow_runner.pyi +1 -1
  192. metaflow-stubs/runner/nbdeploy.pyi +1 -1
  193. metaflow-stubs/runner/nbrun.pyi +1 -1
  194. metaflow-stubs/runner/subprocess_manager.pyi +1 -1
  195. metaflow-stubs/runner/utils.pyi +1 -1
  196. metaflow-stubs/system/__init__.pyi +1 -1
  197. metaflow-stubs/system/system_logger.pyi +2 -2
  198. metaflow-stubs/system/system_monitor.pyi +1 -1
  199. metaflow-stubs/tagging_util.pyi +1 -1
  200. metaflow-stubs/tuple_util.pyi +1 -1
  201. metaflow-stubs/user_configs/__init__.pyi +1 -1
  202. metaflow-stubs/user_configs/config_decorators.pyi +6 -6
  203. metaflow-stubs/user_configs/config_options.pyi +1 -1
  204. metaflow-stubs/user_configs/config_parameters.pyi +4 -4
  205. {ob_metaflow_stubs-6.0.3.170.dist-info → ob_metaflow_stubs-6.0.3.172.dist-info}/METADATA +1 -1
  206. ob_metaflow_stubs-6.0.3.172.dist-info/RECORD +209 -0
  207. ob_metaflow_stubs-6.0.3.170.dist-info/RECORD +0 -206
  208. {ob_metaflow_stubs-6.0.3.170.dist-info → ob_metaflow_stubs-6.0.3.172.dist-info}/WHEEL +0 -0
  209. {ob_metaflow_stubs-6.0.3.170.dist-info → ob_metaflow_stubs-6.0.3.172.dist-info}/top_level.txt +0 -0
@@ -1,15 +1,15 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.207385 #
4
+ # Generated on 2025-05-27T03:43:23.541120 #
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.cards.card_client
12
11
  import metaflow
12
+ import metaflow.plugins.cards.card_client
13
13
 
14
14
  from .card_resolver import resolve_paths_from_task as resolve_paths_from_task
15
15
  from .card_resolver import resumed_info as resumed_info
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.236688 #
4
+ # Generated on 2025-05-27T03:43:23.573593 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.228432 #
4
+ # Generated on 2025-05-27T03:43:23.563989 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.237211 #
4
+ # Generated on 2025-05-27T03:43:23.574061 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.235317 #
4
+ # Generated on 2025-05-27T03:43:23.572095 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.211219 #
4
+ # Generated on 2025-05-27T03:43:23.544954 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.207978 #
4
+ # Generated on 2025-05-27T03:43:23.541703 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.209465 #
4
+ # Generated on 2025-05-27T03:43:23.543209 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,9 +9,9 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.plugins.cards.card_modules.components
13
- import metaflow.plugins.cards.card_modules.card
14
12
  import typing
13
+ import metaflow.plugins.cards.card_modules.card
14
+ import metaflow.plugins.cards.card_modules.components
15
15
 
16
16
  from .basic import LogComponent as LogComponent
17
17
  from .basic import ErrorComponent as ErrorComponent
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.230183 #
4
+ # Generated on 2025-05-27T03:43:23.564945 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.230346 #
4
+ # Generated on 2025-05-27T03:43:23.565105 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.221212 #
4
+ # Generated on 2025-05-27T03:43:23.556412 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.227754 #
4
+ # Generated on 2025-05-27T03:43:23.563122 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.236424 #
4
+ # Generated on 2025-05-27T03:43:23.573323 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.229801 #
4
+ # Generated on 2025-05-27T03:43:23.564571 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.217330 #
4
+ # Generated on 2025-05-27T03:43:23.552331 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.214337 #
4
+ # Generated on 2025-05-27T03:43:23.548839 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.230843 #
4
+ # Generated on 2025-05-27T03:43:23.567606 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.231138 #
4
+ # Generated on 2025-05-27T03:43:23.567905 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.233770 #
4
+ # Generated on 2025-05-27T03:43:23.570551 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,9 +9,9 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.exception
13
12
  import io
14
13
  import metaflow.plugins.datatools.s3.s3
14
+ import metaflow.exception
15
15
 
16
16
  from ....flowspec import FlowSpec as FlowSpec
17
17
  from ....metaflow_current import current as current
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.256622 #
4
+ # Generated on 2025-05-27T03:43:23.593032 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.256324 #
4
+ # Generated on 2025-05-27T03:43:23.592713 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.220079 #
4
+ # Generated on 2025-05-27T03:43:23.555303 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.220297 #
4
+ # Generated on 2025-05-27T03:43:23.555525 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.215566 #
4
+ # Generated on 2025-05-27T03:43:23.553480 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.218727 #
4
+ # Generated on 2025-05-27T03:43:23.554548 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.218122 #
4
+ # Generated on 2025-05-27T03:43:23.552941 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.242540 #
4
+ # Generated on 2025-05-27T03:43:23.578808 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.214571 #
4
+ # Generated on 2025-05-27T03:43:23.549103 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.235100 #
4
+ # Generated on 2025-05-27T03:43:23.571873 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.234163 #
4
+ # Generated on 2025-05-27T03:43:23.570943 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.234030 #
4
+ # Generated on 2025-05-27T03:43:23.570812 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.234373 #
4
+ # Generated on 2025-05-27T03:43:23.571171 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.234804 #
4
+ # Generated on 2025-05-27T03:43:23.571585 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.218256 #
4
+ # Generated on 2025-05-27T03:43:23.551231 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.243765 #
4
+ # Generated on 2025-05-27T03:43:23.576300 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.243368 #
4
+ # Generated on 2025-05-27T03:43:23.575942 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.176533 #
4
+ # Generated on 2025-05-27T03:43:23.508038 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.244472 #
4
+ # Generated on 2025-05-27T03:43:23.576999 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.308021 #
4
+ # Generated on 2025-05-27T03:43:23.645480 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.244716 #
4
+ # Generated on 2025-05-27T03:43:23.577245 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.177451 #
4
+ # Generated on 2025-05-27T03:43:23.509007 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -9,10 +9,12 @@ from __future__ import annotations
9
9
  import metaflow
10
10
  import typing
11
11
  if typing.TYPE_CHECKING:
12
- import metaflow.mf_extensions.outerbounds.plugins.card_utilities.injector
13
12
  import metaflow.decorators
13
+ import metaflow.mf_extensions.outerbounds.plugins.card_utilities.injector
14
14
 
15
15
  from ...metaflow_current import current as current
16
+ from ...mf_extensions.outerbounds.plugins.ollama import constants as constants
17
+ from ...mf_extensions.outerbounds.plugins.ollama import exceptions as exceptions
16
18
  from ...mf_extensions.outerbounds.plugins.ollama import ollama as ollama
17
19
  from ...mf_extensions.outerbounds.plugins.ollama.ollama import OllamaManager as OllamaManager
18
20
  from ...mf_extensions.outerbounds.plugins.card_utilities.injector import CardDecoratorInjector as CardDecoratorInjector
@@ -22,10 +24,10 @@ class OllamaDecorator(metaflow.decorators.StepDecorator, metaflow.mf_extensions.
22
24
  This decorator is used to run Ollama APIs as Metaflow task sidecars.
23
25
 
24
26
  User code call
25
- -----------
27
+ --------------
26
28
  @ollama(
27
- models=['meta/llama3-8b-instruct', 'meta/llama3-70b-instruct'],
28
- backend='local'
29
+ models=[...],
30
+ ...
29
31
  )
30
32
 
31
33
  Valid backend options
@@ -35,18 +37,24 @@ class OllamaDecorator(metaflow.decorators.StepDecorator, metaflow.mf_extensions.
35
37
  - (TODO) 'remote': Spin up separate instance to serve Ollama models.
36
38
 
37
39
  Valid model options
38
- ----------------
39
- - 'llama3.2'
40
- - 'llama3.3'
41
- - any model here https://ollama.com/search
40
+ -------------------
41
+ Any model here https://ollama.com/search, e.g. 'llama3.2', 'llama3.3'
42
42
 
43
43
  Parameters
44
44
  ----------
45
- models: list[Ollama]
45
+ models: list[str]
46
46
  List of Ollama containers running models in sidecars.
47
47
  backend: str
48
48
  Determines where and how to run the Ollama process.
49
+ force_pull: bool
50
+ Whether to run `ollama pull` no matter what, or first check the remote cache in Metaflow datastore for this model key.
51
+ skip_push_check: bool
52
+ Whether to skip the check that populates/overwrites remote cache on terminating an ollama model.
53
+ debug: bool
54
+ Whether to turn on verbose debugging logs.
49
55
  """
56
+ def step_init(self, flow, graph, step_name, decorators, environment, flow_datastore, logger):
57
+ ...
50
58
  def task_decorate(self, step_func, flow, graph, retry_count, max_user_code_retries, ubf_context):
51
59
  ...
52
60
  ...
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.215994 #
4
+ # Generated on 2025-05-27T03:43:23.552748 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.177770 #
4
+ # Generated on 2025-05-27T03:43:23.510981 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.219338 #
4
+ # Generated on 2025-05-27T03:43:23.553830 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.216805 #
4
+ # Generated on 2025-05-27T03:43:23.548301 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.239611 #
4
+ # Generated on 2025-05-27T03:43:23.565690 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
@@ -1,18 +1,18 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.240889 #
4
+ # Generated on 2025-05-27T03:43:23.566982 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations
8
8
 
9
+ import io
9
10
  import metaflow
10
11
  import abc
11
- import io
12
12
  import typing
13
13
  if typing.TYPE_CHECKING:
14
- import metaflow.exception
15
14
  import metaflow.metaflow_environment
15
+ import metaflow.exception
16
16
  import io
17
17
  import abc
18
18
 
@@ -1,7 +1,7 @@
1
1
  ######################################################################################################
2
2
  # Auto-generated Metaflow stub file #
3
3
  # MF version: 2.15.14.1+obcheckpoint(0.2.1);ob(v1) #
4
- # Generated on 2025-05-22T07:40:40.190046 #
4
+ # Generated on 2025-05-27T03:43:23.523347 #
5
5
  ######################################################################################################
6
6
 
7
7
  from __future__ import annotations