ob-metaflow-stubs 6.0.3.165__py2.py3-none-any.whl → 6.0.3.166__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.
- metaflow-stubs/__init__.pyi +722 -722
- metaflow-stubs/cards.pyi +2 -2
- metaflow-stubs/cli.pyi +2 -2
- metaflow-stubs/cli_components/__init__.pyi +2 -2
- metaflow-stubs/cli_components/utils.pyi +2 -2
- metaflow-stubs/client/__init__.pyi +2 -2
- metaflow-stubs/client/core.pyi +5 -5
- metaflow-stubs/client/filecache.pyi +2 -2
- metaflow-stubs/events.pyi +2 -2
- metaflow-stubs/exception.pyi +2 -2
- metaflow-stubs/flowspec.pyi +5 -5
- metaflow-stubs/generated_for.txt +1 -1
- metaflow-stubs/includefile.pyi +4 -4
- metaflow-stubs/info_file.pyi +2 -2
- metaflow-stubs/metadata_provider/__init__.pyi +2 -2
- metaflow-stubs/metadata_provider/heartbeat.pyi +2 -2
- metaflow-stubs/metadata_provider/metadata.pyi +3 -3
- metaflow-stubs/metadata_provider/util.pyi +2 -2
- metaflow-stubs/metaflow_config.pyi +4 -2
- metaflow-stubs/metaflow_current.pyi +127 -127
- metaflow-stubs/metaflow_git.pyi +2 -2
- metaflow-stubs/mf_extensions/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/async_cards.pyi +3 -3
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/deco_injection_mixin.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/extra_components.pyi +4 -4
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/checkpoint_lister.pyi +5 -5
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/lineage_card.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/checkpoint_storage.pyi +5 -5
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/constructors.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/core.pyi +4 -4
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/decorator.pyi +5 -5
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/exceptions.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/final_api.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/lineage.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/context.pyi +3 -3
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/core.pyi +3 -3
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/decorator.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/exceptions.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/task_utils.pyi +4 -4
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/utils.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastructures.pyi +3 -3
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/exceptions.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/decorator.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/core.pyi +3 -3
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/exceptions.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/model_storage.pyi +4 -4
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/flowspec_utils.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/general.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/identity_utils.pyi +3 -3
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/base.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/tar.pyi +3 -3
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/tar_utils.pyi +3 -3
- metaflow-stubs/mf_extensions/outerbounds/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/plugins/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/injector.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/coreweave.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/nebius.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/ollama.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/snowflake.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/profilers/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/profilers/gpu.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/remote_config.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/toplevel/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/toplevel/global_aliases_for_metaflow_package.pyi +2 -2
- metaflow-stubs/multicore_utils.pyi +2 -2
- metaflow-stubs/parameters.pyi +4 -4
- metaflow-stubs/plugins/__init__.pyi +11 -11
- metaflow-stubs/plugins/airflow/__init__.pyi +2 -2
- metaflow-stubs/plugins/airflow/airflow_utils.pyi +2 -2
- metaflow-stubs/plugins/airflow/exception.pyi +2 -2
- metaflow-stubs/plugins/airflow/sensors/__init__.pyi +2 -2
- metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi +2 -2
- metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi +2 -2
- metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi +2 -2
- metaflow-stubs/plugins/argo/__init__.pyi +2 -2
- metaflow-stubs/plugins/argo/argo_client.pyi +2 -2
- metaflow-stubs/plugins/argo/argo_events.pyi +2 -2
- metaflow-stubs/plugins/argo/argo_workflows.pyi +4 -4
- metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +4 -4
- metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +3 -3
- metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi +3 -3
- metaflow-stubs/plugins/aws/__init__.pyi +2 -2
- metaflow-stubs/plugins/aws/aws_client.pyi +2 -2
- metaflow-stubs/plugins/aws/aws_utils.pyi +2 -2
- metaflow-stubs/plugins/aws/batch/__init__.pyi +2 -2
- metaflow-stubs/plugins/aws/batch/batch.pyi +2 -2
- metaflow-stubs/plugins/aws/batch/batch_client.pyi +2 -2
- metaflow-stubs/plugins/aws/batch/batch_decorator.pyi +2 -2
- metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi +2 -2
- metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi +4 -4
- metaflow-stubs/plugins/aws/step_functions/__init__.pyi +2 -2
- metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi +2 -2
- metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi +2 -2
- metaflow-stubs/plugins/aws/step_functions/step_functions.pyi +2 -2
- metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi +2 -2
- metaflow-stubs/plugins/aws/step_functions/step_functions_deployer.pyi +4 -4
- metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi +3 -3
- metaflow-stubs/plugins/azure/__init__.pyi +2 -2
- metaflow-stubs/plugins/azure/azure_credential.pyi +2 -2
- metaflow-stubs/plugins/azure/azure_exceptions.pyi +2 -2
- metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi +4 -4
- metaflow-stubs/plugins/azure/azure_utils.pyi +2 -2
- metaflow-stubs/plugins/azure/blob_service_client_factory.pyi +2 -2
- metaflow-stubs/plugins/azure/includefile_support.pyi +2 -2
- metaflow-stubs/plugins/cards/__init__.pyi +2 -2
- metaflow-stubs/plugins/cards/card_client.pyi +2 -2
- metaflow-stubs/plugins/cards/card_creator.pyi +2 -2
- metaflow-stubs/plugins/cards/card_datastore.pyi +2 -2
- metaflow-stubs/plugins/cards/card_decorator.pyi +2 -2
- metaflow-stubs/plugins/cards/card_modules/__init__.pyi +2 -2
- metaflow-stubs/plugins/cards/card_modules/basic.pyi +3 -3
- metaflow-stubs/plugins/cards/card_modules/card.pyi +2 -2
- metaflow-stubs/plugins/cards/card_modules/components.pyi +4 -4
- metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi +2 -2
- metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi +2 -2
- metaflow-stubs/plugins/cards/card_modules/test_cards.pyi +2 -2
- metaflow-stubs/plugins/cards/card_resolver.pyi +2 -2
- metaflow-stubs/plugins/cards/component_serializer.pyi +2 -2
- metaflow-stubs/plugins/cards/exception.pyi +2 -2
- metaflow-stubs/plugins/catch_decorator.pyi +3 -3
- metaflow-stubs/plugins/datatools/__init__.pyi +2 -2
- metaflow-stubs/plugins/datatools/local.pyi +2 -2
- metaflow-stubs/plugins/datatools/s3/__init__.pyi +2 -2
- metaflow-stubs/plugins/datatools/s3/s3.pyi +4 -4
- metaflow-stubs/plugins/datatools/s3/s3tail.pyi +2 -2
- metaflow-stubs/plugins/datatools/s3/s3util.pyi +2 -2
- metaflow-stubs/plugins/debug_logger.pyi +2 -2
- metaflow-stubs/plugins/debug_monitor.pyi +2 -2
- metaflow-stubs/plugins/environment_decorator.pyi +2 -2
- metaflow-stubs/plugins/events_decorator.pyi +2 -2
- metaflow-stubs/plugins/frameworks/__init__.pyi +2 -2
- metaflow-stubs/plugins/frameworks/pytorch.pyi +2 -2
- metaflow-stubs/plugins/gcp/__init__.pyi +2 -2
- metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi +4 -4
- metaflow-stubs/plugins/gcp/gs_exceptions.pyi +2 -2
- metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi +2 -2
- metaflow-stubs/plugins/gcp/gs_utils.pyi +2 -2
- metaflow-stubs/plugins/gcp/includefile_support.pyi +2 -2
- metaflow-stubs/plugins/kubernetes/__init__.pyi +2 -2
- metaflow-stubs/plugins/kubernetes/kube_utils.pyi +3 -3
- metaflow-stubs/plugins/kubernetes/kubernetes.pyi +2 -2
- metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +2 -2
- metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +2 -2
- metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi +2 -2
- metaflow-stubs/plugins/kubernetes/spot_monitor_sidecar.pyi +2 -2
- metaflow-stubs/plugins/ollama/__init__.pyi +3 -3
- metaflow-stubs/plugins/parallel_decorator.pyi +2 -2
- metaflow-stubs/plugins/perimeters.pyi +2 -2
- metaflow-stubs/plugins/project_decorator.pyi +2 -2
- metaflow-stubs/plugins/pypi/__init__.pyi +3 -3
- metaflow-stubs/plugins/pypi/conda_decorator.pyi +2 -2
- metaflow-stubs/plugins/pypi/conda_environment.pyi +5 -5
- metaflow-stubs/plugins/pypi/parsers.pyi +2 -2
- metaflow-stubs/plugins/pypi/pypi_decorator.pyi +2 -2
- metaflow-stubs/plugins/pypi/pypi_environment.pyi +2 -2
- metaflow-stubs/plugins/pypi/utils.pyi +2 -2
- metaflow-stubs/plugins/resources_decorator.pyi +2 -2
- metaflow-stubs/plugins/retry_decorator.pyi +2 -2
- metaflow-stubs/plugins/secrets/__init__.pyi +3 -3
- metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +4 -4
- metaflow-stubs/plugins/secrets/secrets_decorator.pyi +2 -2
- metaflow-stubs/plugins/snowflake/__init__.pyi +2 -2
- metaflow-stubs/plugins/storage_executor.pyi +2 -2
- metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +3 -3
- metaflow-stubs/plugins/timeout_decorator.pyi +3 -3
- metaflow-stubs/plugins/uv/__init__.pyi +2 -2
- metaflow-stubs/plugins/uv/uv_environment.pyi +2 -2
- metaflow-stubs/profilers/__init__.pyi +2 -2
- metaflow-stubs/pylint_wrapper.pyi +2 -2
- metaflow-stubs/runner/__init__.pyi +2 -2
- metaflow-stubs/runner/deployer.pyi +31 -31
- metaflow-stubs/runner/deployer_impl.pyi +3 -3
- metaflow-stubs/runner/metaflow_runner.pyi +4 -4
- metaflow-stubs/runner/nbdeploy.pyi +2 -2
- metaflow-stubs/runner/nbrun.pyi +2 -2
- metaflow-stubs/runner/subprocess_manager.pyi +2 -2
- metaflow-stubs/runner/utils.pyi +3 -3
- metaflow-stubs/system/__init__.pyi +2 -2
- metaflow-stubs/system/system_logger.pyi +3 -3
- metaflow-stubs/system/system_monitor.pyi +2 -2
- metaflow-stubs/tagging_util.pyi +2 -2
- metaflow-stubs/tuple_util.pyi +2 -2
- metaflow-stubs/user_configs/__init__.pyi +2 -2
- metaflow-stubs/user_configs/config_decorators.pyi +6 -6
- metaflow-stubs/user_configs/config_options.pyi +3 -3
- metaflow-stubs/user_configs/config_parameters.pyi +6 -6
- {ob_metaflow_stubs-6.0.3.165.dist-info → ob_metaflow_stubs-6.0.3.166.dist-info}/METADATA +1 -1
- ob_metaflow_stubs-6.0.3.166.dist-info/RECORD +206 -0
- ob_metaflow_stubs-6.0.3.165.dist-info/RECORD +0 -206
- {ob_metaflow_stubs-6.0.3.165.dist-info → ob_metaflow_stubs-6.0.3.166.dist-info}/WHEEL +0 -0
- {ob_metaflow_stubs-6.0.3.165.dist-info → ob_metaflow_stubs-6.0.3.166.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.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:53.986458 #
|
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.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:54.002583 #
|
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.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:54.001407 #
|
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.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:54.028337 #
|
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.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:53.990554 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
metaflow-stubs/client/core.pyi
CHANGED
@@ -1,20 +1,20 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:53.994915 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
8
8
|
|
9
|
-
import typing
|
10
9
|
import metaflow
|
10
|
+
import typing
|
11
11
|
if typing.TYPE_CHECKING:
|
12
|
+
import metaflow.client.core
|
12
13
|
import datetime
|
13
|
-
import tempfile
|
14
14
|
import tarfile
|
15
|
-
import metaflow.client.core
|
16
15
|
import typing
|
17
16
|
import metaflow.events
|
17
|
+
import tempfile
|
18
18
|
|
19
19
|
from ..metaflow_current import current as current
|
20
20
|
from ..events import Trigger as Trigger
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:54.023430 #
|
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.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:53.987405 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
metaflow-stubs/exception.pyi
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:53.979014 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
metaflow-stubs/flowspec.pyi
CHANGED
@@ -1,19 +1,19 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:53.983659 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
8
8
|
|
9
|
-
import typing
|
10
9
|
import metaflow
|
10
|
+
import typing
|
11
11
|
if typing.TYPE_CHECKING:
|
12
|
+
import metaflow.exception
|
12
13
|
import metaflow.unbounded_foreach
|
13
|
-
import metaflow.datastore.inputs
|
14
14
|
import typing
|
15
15
|
import metaflow.flowspec
|
16
|
-
import metaflow.
|
16
|
+
import metaflow.datastore.inputs
|
17
17
|
|
18
18
|
from . import parameters as parameters
|
19
19
|
from .parameters import DelayedEvaluationParameter as DelayedEvaluationParameter
|
metaflow-stubs/generated_for.txt
CHANGED
@@ -1 +1 @@
|
|
1
|
-
2.15.11.
|
1
|
+
2.15.11.2+obcheckpoint(0.2.1);ob(v1) 2025-05-09T17:10:53.975413
|
metaflow-stubs/includefile.pyi
CHANGED
@@ -1,17 +1,17 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:53.989870 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
8
8
|
|
9
|
-
import typing
|
10
9
|
import metaflow
|
10
|
+
import typing
|
11
11
|
if typing.TYPE_CHECKING:
|
12
12
|
import metaflow._vendor.click.types
|
13
|
-
import typing
|
14
13
|
import metaflow.parameters
|
14
|
+
import typing
|
15
15
|
|
16
16
|
from ._vendor import click as click
|
17
17
|
from .exception import MetaflowException as MetaflowException
|
metaflow-stubs/info_file.pyi
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:53.978277 #
|
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.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:53.982549 #
|
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.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:54.006859 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,13 +1,13 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:54.006370 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
8
8
|
|
9
|
-
import typing
|
10
9
|
import metaflow
|
10
|
+
import typing
|
11
11
|
if typing.TYPE_CHECKING:
|
12
12
|
import metaflow.metadata_provider.metadata
|
13
13
|
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:54.006595 #
|
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.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:53.980121 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -271,6 +271,8 @@ DEBUG_STUBGEN: bool
|
|
271
271
|
|
272
272
|
DEBUG_USERCONF: bool
|
273
273
|
|
274
|
+
DEBUG_CONDA: bool
|
275
|
+
|
274
276
|
AWS_SANDBOX_ENABLED: bool
|
275
277
|
|
276
278
|
AWS_SANDBOX_STS_ENDPOINT_URL: None
|
@@ -1,21 +1,21 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:54.123240 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
8
8
|
|
9
9
|
import typing
|
10
10
|
if typing.TYPE_CHECKING:
|
11
|
+
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator
|
11
12
|
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator
|
12
|
-
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core
|
13
13
|
import metaflow
|
14
14
|
import typing
|
15
|
-
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator
|
16
15
|
import metaflow.events
|
17
|
-
import metaflow.plugins.cards.component_serializer
|
18
16
|
import metaflow.metaflow_current
|
17
|
+
import metaflow.plugins.cards.component_serializer
|
18
|
+
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core
|
19
19
|
|
20
20
|
|
21
21
|
TYPE_CHECKING: bool
|
@@ -228,21 +228,125 @@ class Current(object, metaclass=type):
|
|
228
228
|
def graph(self):
|
229
229
|
...
|
230
230
|
@property
|
231
|
-
def
|
231
|
+
def checkpoint(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator.CurrentCheckpointer":
|
232
232
|
"""
|
233
|
-
(only in the presence of the @
|
233
|
+
(only in the presence of the @checkpoint decorator)
|
234
234
|
|
235
|
-
The `@
|
236
|
-
object
|
237
|
-
distinguish between them using `@card(id=ID)` as the decorator. You will then
|
238
|
-
be able to access that specific card using `current.card[ID].
|
235
|
+
The `@checkpoint` decorator makes saving/loading checkpoints available through the `current.checkpoint`.
|
236
|
+
The object exposes `save`/`load`/`list` methods for saving/loading checkpoints.
|
239
237
|
|
240
|
-
|
238
|
+
You can check if a checkpoint is loaded by `current.checkpoint.is_loaded` and get the checkpoint information
|
239
|
+
by using `current.checkpoint.info`. The `current.checkpoint.directory` returns the path to the checkpoint directory
|
240
|
+
where the checkpoint maybe loaded or saved.
|
241
|
+
|
242
|
+
Usage (Saving Checkpoints):
|
243
|
+
-------
|
244
|
+
```
|
245
|
+
@checkpoint
|
246
|
+
@step
|
247
|
+
def train(self):
|
248
|
+
model = create_model(self.parameters, checkpoint_path = None)
|
249
|
+
for i in range(self.epochs):
|
250
|
+
# some training logic
|
251
|
+
loss = model.train(self.dataset)
|
252
|
+
if i % 10 == 0:
|
253
|
+
model.save(
|
254
|
+
current.checkpoint.directory,
|
255
|
+
)
|
256
|
+
# saves the contents of the `current.checkpoint.directory` as a checkpoint
|
257
|
+
# and returns a reference dictionary to the checkpoint saved in the datastore
|
258
|
+
self.latest_checkpoint = current.checkpoint.save(
|
259
|
+
name="epoch_checkpoint",
|
260
|
+
metadata={
|
261
|
+
"epoch": i,
|
262
|
+
"loss": loss,
|
263
|
+
}
|
264
|
+
)
|
265
|
+
```
|
266
|
+
Usage (Using Loaded Checkpoints):
|
267
|
+
-------
|
268
|
+
```
|
269
|
+
@retry(times=3)
|
270
|
+
@checkpoint
|
271
|
+
@step
|
272
|
+
def train(self):
|
273
|
+
# Assume that the task has restarted and the previous attempt of the task
|
274
|
+
# saved a checkpoint
|
275
|
+
checkpoint_path = None
|
276
|
+
if current.checkpoint.is_loaded: # Check if a checkpoint is loaded
|
277
|
+
print("Loaded checkpoint from the previous attempt")
|
278
|
+
checkpoint_path = current.checkpoint.directory
|
279
|
+
|
280
|
+
model = create_model(self.parameters, checkpoint_path = checkpoint_path)
|
281
|
+
for i in range(self.epochs):
|
282
|
+
...
|
283
|
+
```
|
241
284
|
|
242
285
|
Returns
|
243
286
|
-------
|
244
|
-
|
245
|
-
The
|
287
|
+
CurrentCheckpointer
|
288
|
+
The object for handling checkpointing within a step.
|
289
|
+
"""
|
290
|
+
...
|
291
|
+
@property
|
292
|
+
def model(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core.ModelSerializer":
|
293
|
+
"""
|
294
|
+
(only in the presence of the @model decorator)
|
295
|
+
|
296
|
+
The object used for loading / saving models.
|
297
|
+
`current.model` exposes a `save` method to save models and a `load` method to load models.
|
298
|
+
`current.model.loaded` exposes the paths to the models loaded via the `load` argument in the @model decorator
|
299
|
+
or models loaded via `current.model.load`.
|
300
|
+
|
301
|
+
Usage (Saving a model):
|
302
|
+
-------
|
303
|
+
|
304
|
+
```
|
305
|
+
@model
|
306
|
+
@step
|
307
|
+
def train(self):
|
308
|
+
# current.model.save returns a dictionary reference to the model saved
|
309
|
+
self.my_model = current.model.save(
|
310
|
+
path_to_my_model,
|
311
|
+
label="my_model",
|
312
|
+
metadata={
|
313
|
+
"epochs": 10,
|
314
|
+
"batch-size": 32,
|
315
|
+
"learning-rate": 0.001,
|
316
|
+
}
|
317
|
+
)
|
318
|
+
self.next(self.test)
|
319
|
+
|
320
|
+
@model(load="my_model")
|
321
|
+
@step
|
322
|
+
def test(self):
|
323
|
+
# `current.model.loaded` returns a dictionary of the loaded models
|
324
|
+
# where the key is the name of the artifact and the value is the path to the model
|
325
|
+
print(os.listdir(current.model.loaded["my_model"]))
|
326
|
+
self.next(self.end)
|
327
|
+
```
|
328
|
+
|
329
|
+
Usage (Loading models):
|
330
|
+
-------
|
331
|
+
|
332
|
+
```
|
333
|
+
@step
|
334
|
+
def train(self):
|
335
|
+
# current.model.load returns the path to the model loaded
|
336
|
+
checkpoint_path = current.model.load(
|
337
|
+
self.checkpoint_key,
|
338
|
+
)
|
339
|
+
model_path = current.model.load(
|
340
|
+
self.model,
|
341
|
+
)
|
342
|
+
self.next(self.test)
|
343
|
+
```
|
344
|
+
|
345
|
+
|
346
|
+
Returns
|
347
|
+
-------
|
348
|
+
ModelSerializer
|
349
|
+
The object used for loading / saving models.
|
246
350
|
"""
|
247
351
|
...
|
248
352
|
@property
|
@@ -344,125 +448,21 @@ class Current(object, metaclass=type):
|
|
344
448
|
"""
|
345
449
|
...
|
346
450
|
@property
|
347
|
-
def
|
348
|
-
"""
|
349
|
-
(only in the presence of the @model decorator)
|
350
|
-
|
351
|
-
The object used for loading / saving models.
|
352
|
-
`current.model` exposes a `save` method to save models and a `load` method to load models.
|
353
|
-
`current.model.loaded` exposes the paths to the models loaded via the `load` argument in the @model decorator
|
354
|
-
or models loaded via `current.model.load`.
|
355
|
-
|
356
|
-
Usage (Saving a model):
|
357
|
-
-------
|
358
|
-
|
359
|
-
```
|
360
|
-
@model
|
361
|
-
@step
|
362
|
-
def train(self):
|
363
|
-
# current.model.save returns a dictionary reference to the model saved
|
364
|
-
self.my_model = current.model.save(
|
365
|
-
path_to_my_model,
|
366
|
-
label="my_model",
|
367
|
-
metadata={
|
368
|
-
"epochs": 10,
|
369
|
-
"batch-size": 32,
|
370
|
-
"learning-rate": 0.001,
|
371
|
-
}
|
372
|
-
)
|
373
|
-
self.next(self.test)
|
374
|
-
|
375
|
-
@model(load="my_model")
|
376
|
-
@step
|
377
|
-
def test(self):
|
378
|
-
# `current.model.loaded` returns a dictionary of the loaded models
|
379
|
-
# where the key is the name of the artifact and the value is the path to the model
|
380
|
-
print(os.listdir(current.model.loaded["my_model"]))
|
381
|
-
self.next(self.end)
|
382
|
-
```
|
383
|
-
|
384
|
-
Usage (Loading models):
|
385
|
-
-------
|
386
|
-
|
387
|
-
```
|
388
|
-
@step
|
389
|
-
def train(self):
|
390
|
-
# current.model.load returns the path to the model loaded
|
391
|
-
checkpoint_path = current.model.load(
|
392
|
-
self.checkpoint_key,
|
393
|
-
)
|
394
|
-
model_path = current.model.load(
|
395
|
-
self.model,
|
396
|
-
)
|
397
|
-
self.next(self.test)
|
398
|
-
```
|
399
|
-
|
400
|
-
|
401
|
-
Returns
|
402
|
-
-------
|
403
|
-
ModelSerializer
|
404
|
-
The object used for loading / saving models.
|
405
|
-
"""
|
406
|
-
...
|
407
|
-
@property
|
408
|
-
def checkpoint(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator.CurrentCheckpointer":
|
451
|
+
def card(self) -> "metaflow.plugins.cards.component_serializer.CardComponentCollector":
|
409
452
|
"""
|
410
|
-
(only in the presence of the @
|
411
|
-
|
412
|
-
The `@checkpoint` decorator makes saving/loading checkpoints available through the `current.checkpoint`.
|
413
|
-
The object exposes `save`/`load`/`list` methods for saving/loading checkpoints.
|
414
|
-
|
415
|
-
You can check if a checkpoint is loaded by `current.checkpoint.is_loaded` and get the checkpoint information
|
416
|
-
by using `current.checkpoint.info`. The `current.checkpoint.directory` returns the path to the checkpoint directory
|
417
|
-
where the checkpoint maybe loaded or saved.
|
453
|
+
(only in the presence of the @card decorator)
|
418
454
|
|
419
|
-
|
420
|
-
|
421
|
-
|
422
|
-
|
423
|
-
@step
|
424
|
-
def train(self):
|
425
|
-
model = create_model(self.parameters, checkpoint_path = None)
|
426
|
-
for i in range(self.epochs):
|
427
|
-
# some training logic
|
428
|
-
loss = model.train(self.dataset)
|
429
|
-
if i % 10 == 0:
|
430
|
-
model.save(
|
431
|
-
current.checkpoint.directory,
|
432
|
-
)
|
433
|
-
# saves the contents of the `current.checkpoint.directory` as a checkpoint
|
434
|
-
# and returns a reference dictionary to the checkpoint saved in the datastore
|
435
|
-
self.latest_checkpoint = current.checkpoint.save(
|
436
|
-
name="epoch_checkpoint",
|
437
|
-
metadata={
|
438
|
-
"epoch": i,
|
439
|
-
"loss": loss,
|
440
|
-
}
|
441
|
-
)
|
442
|
-
```
|
443
|
-
Usage (Using Loaded Checkpoints):
|
444
|
-
-------
|
445
|
-
```
|
446
|
-
@retry(times=3)
|
447
|
-
@checkpoint
|
448
|
-
@step
|
449
|
-
def train(self):
|
450
|
-
# Assume that the task has restarted and the previous attempt of the task
|
451
|
-
# saved a checkpoint
|
452
|
-
checkpoint_path = None
|
453
|
-
if current.checkpoint.is_loaded: # Check if a checkpoint is loaded
|
454
|
-
print("Loaded checkpoint from the previous attempt")
|
455
|
-
checkpoint_path = current.checkpoint.directory
|
455
|
+
The `@card` decorator makes the cards available through the `current.card`
|
456
|
+
object. If multiple `@card` decorators are present, you can add an `ID` to
|
457
|
+
distinguish between them using `@card(id=ID)` as the decorator. You will then
|
458
|
+
be able to access that specific card using `current.card[ID].
|
456
459
|
|
457
|
-
|
458
|
-
for i in range(self.epochs):
|
459
|
-
...
|
460
|
-
```
|
460
|
+
Methods available are `append` and `extend`
|
461
461
|
|
462
462
|
Returns
|
463
463
|
-------
|
464
|
-
|
465
|
-
The
|
464
|
+
CardComponentCollector
|
465
|
+
The or one of the cards attached to this step.
|
466
466
|
"""
|
467
467
|
...
|
468
468
|
@property
|
metaflow-stubs/metaflow_git.pyi
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:53.986801 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:53.988686 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:53.997931 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:53.998007 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:53.998065 #
|
5
5
|
######################################################################################################
|
6
6
|
|
metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/__init__.pyi
CHANGED
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:54.051519 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-09T17:10:54.052236 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -9,8 +9,8 @@ from __future__ import annotations
|
|
9
9
|
import threading
|
10
10
|
import typing
|
11
11
|
if typing.TYPE_CHECKING:
|
12
|
-
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.card_utils.async_cards
|
13
12
|
import threading
|
13
|
+
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.card_utils.async_cards
|
14
14
|
|
15
15
|
from ......metaflow_current import current as current
|
16
16
|
|