ob-metaflow-stubs 6.0.3.153__py2.py3-none-any.whl → 6.0.3.154__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 +654 -654
- 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 +6 -6
- metaflow-stubs/client/filecache.pyi +3 -3
- metaflow-stubs/events.pyi +2 -2
- metaflow-stubs/exception.pyi +2 -2
- metaflow-stubs/flowspec.pyi +4 -4
- metaflow-stubs/generated_for.txt +1 -1
- metaflow-stubs/includefile.pyi +3 -3
- 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 +2 -2
- metaflow-stubs/metadata_provider/util.pyi +2 -2
- metaflow-stubs/metaflow_config.pyi +2 -2
- metaflow-stubs/metaflow_current.pyi +133 -133
- 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 +3 -3
- 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 +4 -4
- 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 +3 -3
- 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 +3 -3
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/decorator.pyi +3 -3
- 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 +3 -3
- 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/core.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 +3 -3
- 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 +2 -2
- 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 +3 -3
- 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 +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/tar_utils.pyi +2 -2
- 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/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 +3 -3
- metaflow-stubs/plugins/__init__.pyi +12 -12
- 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 +11 -5
- metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +3 -3
- metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +2 -2
- 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 +3 -3
- 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 +2 -2
- metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi +2 -2
- 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 +3 -3
- 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 +6 -6
- metaflow-stubs/plugins/cards/card_client.pyi +3 -3
- 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 +3 -3
- 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 +2 -2
- 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 +3 -3
- 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 +2 -2
- 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 +2 -2
- metaflow-stubs/plugins/pypi/conda_decorator.pyi +2 -2
- metaflow-stubs/plugins/pypi/conda_environment.pyi +3 -3
- 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 +2 -2
- metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +3 -3
- 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 +2 -2
- metaflow-stubs/plugins/timeout_decorator.pyi +3 -3
- 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 +29 -29
- metaflow-stubs/runner/deployer_impl.pyi +2 -2
- metaflow-stubs/runner/metaflow_runner.pyi +2 -2
- 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 +2 -2
- 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 +5 -5
- metaflow-stubs/user_configs/config_options.pyi +2 -2
- metaflow-stubs/user_configs/config_parameters.pyi +7 -7
- {ob_metaflow_stubs-6.0.3.153.dist-info → ob_metaflow_stubs-6.0.3.154.dist-info}/METADATA +1 -1
- ob_metaflow_stubs-6.0.3.154.dist-info/RECORD +198 -0
- ob_metaflow_stubs-6.0.3.153.dist-info/RECORD +0 -198
- {ob_metaflow_stubs-6.0.3.153.dist-info → ob_metaflow_stubs-6.0.3.154.dist-info}/WHEEL +0 -0
- {ob_metaflow_stubs-6.0.3.153.dist-info → ob_metaflow_stubs-6.0.3.154.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.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.882633 #
|
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.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.896238 #
|
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.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.895065 #
|
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.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.920992 #
|
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.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.885602 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
metaflow-stubs/client/core.pyi
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.890022 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -9,12 +9,12 @@ from __future__ import annotations
|
|
9
9
|
import typing
|
10
10
|
import metaflow
|
11
11
|
if typing.TYPE_CHECKING:
|
12
|
-
import
|
13
|
-
import
|
12
|
+
import datetime
|
13
|
+
import tarfile
|
14
14
|
import metaflow.events
|
15
15
|
import tempfile
|
16
|
-
import
|
17
|
-
import
|
16
|
+
import metaflow.client.core
|
17
|
+
import typing
|
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.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.916118 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -9,8 +9,8 @@ from __future__ import annotations
|
|
9
9
|
import metaflow
|
10
10
|
import typing
|
11
11
|
if typing.TYPE_CHECKING:
|
12
|
-
import metaflow.datastore.content_addressed_store
|
13
12
|
import metaflow.exception
|
13
|
+
import metaflow.datastore.content_addressed_store
|
14
14
|
|
15
15
|
from ..exception import MetaflowException as MetaflowException
|
16
16
|
|
metaflow-stubs/events.pyi
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.882243 #
|
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.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.874423 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
metaflow-stubs/flowspec.pyi
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.879106 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -10,10 +10,10 @@ import typing
|
|
10
10
|
import metaflow
|
11
11
|
if typing.TYPE_CHECKING:
|
12
12
|
import metaflow.exception
|
13
|
-
import metaflow.flowspec
|
14
|
-
import typing
|
15
13
|
import metaflow.unbounded_foreach
|
16
14
|
import metaflow.datastore.inputs
|
15
|
+
import metaflow.flowspec
|
16
|
+
import typing
|
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.7.
|
1
|
+
2.15.7.2+obcheckpoint(0.1.9);ob(v1) 2025-04-10T22:13:14.870901
|
metaflow-stubs/includefile.pyi
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.884965 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -9,9 +9,9 @@ from __future__ import annotations
|
|
9
9
|
import typing
|
10
10
|
import metaflow
|
11
11
|
if typing.TYPE_CHECKING:
|
12
|
-
import metaflow.parameters
|
13
12
|
import typing
|
14
13
|
import metaflow._vendor.click.types
|
14
|
+
import metaflow.parameters
|
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.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.873604 #
|
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.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.878016 #
|
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.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.900503 #
|
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.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.900017 #
|
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.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.900232 #
|
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.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.875548 #
|
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.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:15.013319 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -10,12 +10,12 @@ import typing
|
|
10
10
|
if typing.TYPE_CHECKING:
|
11
11
|
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator
|
12
12
|
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core
|
13
|
-
import typing
|
14
|
-
import metaflow
|
15
|
-
import metaflow.events
|
16
|
-
import metaflow.plugins.cards.component_serializer
|
17
13
|
import metaflow.metaflow_current
|
14
|
+
import metaflow.events
|
18
15
|
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator
|
16
|
+
import metaflow.plugins.cards.component_serializer
|
17
|
+
import typing
|
18
|
+
import metaflow
|
19
19
|
|
20
20
|
|
21
21
|
TYPE_CHECKING: bool
|
@@ -228,82 +228,111 @@ class Current(object, metaclass=type):
|
|
228
228
|
def graph(self):
|
229
229
|
...
|
230
230
|
@property
|
231
|
-
def
|
231
|
+
def parallel(self) -> "metaflow.metaflow_current.Parallel":
|
232
232
|
"""
|
233
|
-
(only in the presence of the @
|
234
|
-
|
235
|
-
The `@card` decorator makes the cards available through the `current.card`
|
236
|
-
object. If multiple `@card` decorators are present, you can add an `ID` to
|
237
|
-
distinguish between them using `@card(id=ID)` as the decorator. You will then
|
238
|
-
be able to access that specific card using `current.card[ID].
|
233
|
+
(only in the presence of the @parallel decorator)
|
239
234
|
|
240
|
-
|
235
|
+
Returns a namedtuple with relevant information about the parallel task.
|
241
236
|
|
242
237
|
Returns
|
243
238
|
-------
|
244
|
-
|
245
|
-
|
239
|
+
Parallel
|
240
|
+
`namedtuple` with the following fields:
|
241
|
+
- main_ip (`str`)
|
242
|
+
The IP address of the control task.
|
243
|
+
- num_nodes (`int`)
|
244
|
+
The total number of tasks created by @parallel
|
245
|
+
- node_index (`int`)
|
246
|
+
The index of the current task in all the @parallel tasks.
|
247
|
+
- control_task_id (`Optional[str]`)
|
248
|
+
The task ID of the control task. Available to all tasks.
|
246
249
|
"""
|
247
250
|
...
|
248
251
|
@property
|
249
|
-
def
|
252
|
+
def is_parallel(self) -> bool:
|
250
253
|
"""
|
251
|
-
(only in the presence of the @
|
254
|
+
(only in the presence of the @parallel decorator)
|
252
255
|
|
253
|
-
|
254
|
-
|
256
|
+
True if the current step is a @parallel step.
|
257
|
+
"""
|
258
|
+
...
|
259
|
+
@property
|
260
|
+
def model(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core.ModelSerializer":
|
261
|
+
"""
|
262
|
+
(only in the presence of the @model decorator)
|
255
263
|
|
256
|
-
|
257
|
-
|
258
|
-
|
264
|
+
The object used for loading / saving models.
|
265
|
+
`current.model` exposes a `save` method to save models and a `load` method to load models.
|
266
|
+
`current.model.loaded` exposes the paths to the models loaded via the `load` argument in the @model decorator
|
267
|
+
or models loaded via `current.model.load`.
|
259
268
|
|
260
|
-
Usage (Saving
|
269
|
+
Usage (Saving a model):
|
261
270
|
-------
|
271
|
+
|
262
272
|
```
|
263
|
-
@
|
273
|
+
@model
|
264
274
|
@step
|
265
275
|
def train(self):
|
266
|
-
model
|
267
|
-
|
268
|
-
|
269
|
-
|
270
|
-
|
271
|
-
|
272
|
-
|
273
|
-
|
274
|
-
|
275
|
-
|
276
|
-
|
277
|
-
|
278
|
-
|
279
|
-
|
280
|
-
|
281
|
-
|
282
|
-
|
276
|
+
# current.model.save returns a dictionary reference to the model saved
|
277
|
+
self.my_model = current.model.save(
|
278
|
+
path_to_my_model,
|
279
|
+
label="my_model",
|
280
|
+
metadata={
|
281
|
+
"epochs": 10,
|
282
|
+
"batch-size": 32,
|
283
|
+
"learning-rate": 0.001,
|
284
|
+
}
|
285
|
+
)
|
286
|
+
self.next(self.test)
|
287
|
+
|
288
|
+
@model(load="my_model")
|
289
|
+
@step
|
290
|
+
def test(self):
|
291
|
+
# `current.model.loaded` returns a dictionary of the loaded models
|
292
|
+
# where the key is the name of the artifact and the value is the path to the model
|
293
|
+
print(os.listdir(current.model.loaded["my_model"]))
|
294
|
+
self.next(self.end)
|
283
295
|
```
|
284
|
-
|
296
|
+
|
297
|
+
Usage (Loading models):
|
285
298
|
-------
|
299
|
+
|
286
300
|
```
|
287
|
-
@retry(times=3)
|
288
|
-
@checkpoint
|
289
301
|
@step
|
290
302
|
def train(self):
|
291
|
-
#
|
292
|
-
|
293
|
-
|
294
|
-
|
295
|
-
|
296
|
-
|
297
|
-
|
298
|
-
|
299
|
-
for i in range(self.epochs):
|
300
|
-
...
|
303
|
+
# current.model.load returns the path to the model loaded
|
304
|
+
checkpoint_path = current.model.load(
|
305
|
+
self.checkpoint_key,
|
306
|
+
)
|
307
|
+
model_path = current.model.load(
|
308
|
+
self.model,
|
309
|
+
)
|
310
|
+
self.next(self.test)
|
301
311
|
```
|
302
312
|
|
313
|
+
|
303
314
|
Returns
|
304
315
|
-------
|
305
|
-
|
306
|
-
The object for
|
316
|
+
ModelSerializer
|
317
|
+
The object used for loading / saving models.
|
318
|
+
"""
|
319
|
+
...
|
320
|
+
@property
|
321
|
+
def card(self) -> "metaflow.plugins.cards.component_serializer.CardComponentCollector":
|
322
|
+
"""
|
323
|
+
(only in the presence of the @card decorator)
|
324
|
+
|
325
|
+
The `@card` decorator makes the cards available through the `current.card`
|
326
|
+
object. If multiple `@card` decorators are present, you can add an `ID` to
|
327
|
+
distinguish between them using `@card(id=ID)` as the decorator. You will then
|
328
|
+
be able to access that specific card using `current.card[ID].
|
329
|
+
|
330
|
+
Methods available are `append` and `extend`
|
331
|
+
|
332
|
+
Returns
|
333
|
+
-------
|
334
|
+
CardComponentCollector
|
335
|
+
The or one of the cards attached to this step.
|
307
336
|
"""
|
308
337
|
...
|
309
338
|
@property
|
@@ -376,93 +405,77 @@ class Current(object, metaclass=type):
|
|
376
405
|
"""
|
377
406
|
...
|
378
407
|
@property
|
379
|
-
def
|
408
|
+
def checkpoint(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator.CurrentCheckpointer":
|
380
409
|
"""
|
381
|
-
(only in the presence of the @
|
410
|
+
(only in the presence of the @checkpoint decorator)
|
382
411
|
|
383
|
-
The
|
384
|
-
|
385
|
-
`current.model.loaded` exposes the paths to the models loaded via the `load` argument in the @model decorator
|
386
|
-
or models loaded via `current.model.load`.
|
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.
|
387
414
|
|
388
|
-
|
389
|
-
|
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.
|
390
418
|
|
419
|
+
Usage (Saving Checkpoints):
|
420
|
+
-------
|
391
421
|
```
|
392
|
-
@
|
422
|
+
@checkpoint
|
393
423
|
@step
|
394
424
|
def train(self):
|
395
|
-
|
396
|
-
|
397
|
-
|
398
|
-
|
399
|
-
|
400
|
-
|
401
|
-
|
402
|
-
|
403
|
-
|
404
|
-
|
405
|
-
|
406
|
-
|
407
|
-
|
408
|
-
|
409
|
-
|
410
|
-
|
411
|
-
|
412
|
-
print(os.listdir(current.model.loaded["my_model"]))
|
413
|
-
self.next(self.end)
|
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
|
+
)
|
414
442
|
```
|
415
|
-
|
416
|
-
Usage (Loading models):
|
443
|
+
Usage (Using Loaded Checkpoints):
|
417
444
|
-------
|
418
|
-
|
419
445
|
```
|
446
|
+
@retry(times=3)
|
447
|
+
@checkpoint
|
420
448
|
@step
|
421
449
|
def train(self):
|
422
|
-
#
|
423
|
-
|
424
|
-
|
425
|
-
|
426
|
-
|
427
|
-
|
428
|
-
)
|
429
|
-
self.next(self.test)
|
430
|
-
```
|
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
|
431
456
|
|
457
|
+
model = create_model(self.parameters, checkpoint_path = checkpoint_path)
|
458
|
+
for i in range(self.epochs):
|
459
|
+
...
|
460
|
+
```
|
432
461
|
|
433
462
|
Returns
|
434
463
|
-------
|
435
|
-
|
436
|
-
The object
|
464
|
+
CurrentCheckpointer
|
465
|
+
The object for handling checkpointing within a step.
|
437
466
|
"""
|
438
467
|
...
|
439
468
|
@property
|
440
|
-
def
|
469
|
+
def trigger(self) -> "metaflow.events.Trigger":
|
441
470
|
"""
|
442
|
-
(only in the presence of the @
|
471
|
+
(only in the presence of the @trigger, or @trigger_on_finish decorators)
|
443
472
|
|
444
|
-
Returns
|
473
|
+
Returns `Trigger` if the current run is triggered by an event
|
445
474
|
|
446
475
|
Returns
|
447
476
|
-------
|
448
|
-
|
449
|
-
`
|
450
|
-
- main_ip (`str`)
|
451
|
-
The IP address of the control task.
|
452
|
-
- num_nodes (`int`)
|
453
|
-
The total number of tasks created by @parallel
|
454
|
-
- node_index (`int`)
|
455
|
-
The index of the current task in all the @parallel tasks.
|
456
|
-
- control_task_id (`Optional[str]`)
|
457
|
-
The task ID of the control task. Available to all tasks.
|
458
|
-
"""
|
459
|
-
...
|
460
|
-
@property
|
461
|
-
def is_parallel(self) -> bool:
|
462
|
-
"""
|
463
|
-
(only in the presence of the @parallel decorator)
|
464
|
-
|
465
|
-
True if the current step is a @parallel step.
|
477
|
+
Trigger
|
478
|
+
`Trigger` if triggered by an event
|
466
479
|
"""
|
467
480
|
...
|
468
481
|
@property
|
@@ -532,19 +545,6 @@ class Current(object, metaclass=type):
|
|
532
545
|
True if the flow is deployed with `--production`.
|
533
546
|
"""
|
534
547
|
...
|
535
|
-
@property
|
536
|
-
def trigger(self) -> "metaflow.events.Trigger":
|
537
|
-
"""
|
538
|
-
(only in the presence of the @trigger_on_finish, or @trigger decorators)
|
539
|
-
|
540
|
-
Returns `Trigger` if the current run is triggered by an event
|
541
|
-
|
542
|
-
Returns
|
543
|
-
-------
|
544
|
-
Trigger
|
545
|
-
`Trigger` if triggered by an event
|
546
|
-
"""
|
547
|
-
...
|
548
548
|
...
|
549
549
|
|
550
550
|
current: Current
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.883839 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.892848 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.7.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.7.2+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-04-10T22:13:14.892906 #
|
5
5
|
######################################################################################################
|
6
6
|
|