ob-metaflow-stubs 6.0.3.148__py2.py3-none-any.whl → 6.0.3.151__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 +668 -668
- 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 +2 -2
- metaflow-stubs/events.pyi +3 -3
- metaflow-stubs/exception.pyi +2 -2
- metaflow-stubs/flowspec.pyi +5 -5
- metaflow-stubs/generated_for.txt +1 -1
- metaflow-stubs/includefile.pyi +5 -5
- 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 +2 -2
- metaflow-stubs/metaflow_current.pyi +141 -141
- 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 +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/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 +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 +5 -5
- 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 +2 -2
- 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/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 +5 -5
- 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 +3 -3
- metaflow-stubs/plugins/argo/argo_events.pyi +2 -2
- metaflow-stubs/plugins/argo/argo_workflows.pyi +3 -3
- 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 +6 -4
- 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 +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 +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 +2 -2
- 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 +3 -3
- 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 +2 -2
- metaflow-stubs/plugins/pypi/conda_decorator.pyi +2 -2
- metaflow-stubs/plugins/pypi/conda_environment.pyi +4 -4
- 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 +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 +3 -3
- metaflow-stubs/plugins/timeout_decorator.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 +8 -6
- metaflow-stubs/runner/deployer_impl.pyi +2 -2
- 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 +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 +6 -6
- metaflow-stubs/user_configs/config_options.pyi +4 -4
- metaflow-stubs/user_configs/config_parameters.pyi +7 -7
- {ob_metaflow_stubs-6.0.3.148.dist-info → ob_metaflow_stubs-6.0.3.151.dist-info}/METADATA +1 -1
- ob_metaflow_stubs-6.0.3.151.dist-info/RECORD +198 -0
- ob_metaflow_stubs-6.0.3.148.dist-info/RECORD +0 -198
- {ob_metaflow_stubs-6.0.3.148.dist-info → ob_metaflow_stubs-6.0.3.151.dist-info}/WHEEL +0 -0
- {ob_metaflow_stubs-6.0.3.148.dist-info → ob_metaflow_stubs-6.0.3.151.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.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.673332 #
|
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.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.687499 #
|
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.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.686291 #
|
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.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.712394 #
|
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.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.676946 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
metaflow-stubs/client/core.pyi
CHANGED
@@ -1,19 +1,19 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.681365 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
8
8
|
|
9
|
-
import metaflow
|
10
9
|
import typing
|
10
|
+
import metaflow
|
11
11
|
if typing.TYPE_CHECKING:
|
12
|
-
import tarfile
|
13
|
-
import tempfile
|
14
12
|
import metaflow.client.core
|
15
|
-
import datetime
|
16
13
|
import metaflow.events
|
14
|
+
import datetime
|
15
|
+
import tempfile
|
16
|
+
import tarfile
|
17
17
|
import typing
|
18
18
|
|
19
19
|
from ..metaflow_current import current as current
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.707500 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
metaflow-stubs/events.pyi
CHANGED
@@ -1,15 +1,15 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.673897 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
8
8
|
|
9
9
|
import typing
|
10
10
|
if typing.TYPE_CHECKING:
|
11
|
-
import metaflow
|
12
11
|
import metaflow.events
|
12
|
+
import metaflow
|
13
13
|
|
14
14
|
|
15
15
|
TYPE_CHECKING: bool
|
metaflow-stubs/exception.pyi
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.665636 #
|
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.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.670295 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
8
8
|
|
9
|
-
import metaflow
|
10
9
|
import typing
|
10
|
+
import metaflow
|
11
11
|
if typing.TYPE_CHECKING:
|
12
|
-
import metaflow.unbounded_foreach
|
13
12
|
import metaflow.flowspec
|
13
|
+
import metaflow.datastore.inputs
|
14
14
|
import typing
|
15
|
+
import metaflow.unbounded_foreach
|
15
16
|
import metaflow.exception
|
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.
|
1
|
+
2.15.6.1+obcheckpoint(0.1.9);ob(v1) 2025-03-17T20:51:51.662151
|
metaflow-stubs/includefile.pyi
CHANGED
@@ -1,17 +1,17 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.676315 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
8
8
|
|
9
|
-
import metaflow
|
10
9
|
import typing
|
10
|
+
import metaflow
|
11
11
|
if typing.TYPE_CHECKING:
|
12
|
-
import metaflow._vendor.click.types
|
13
|
-
import typing
|
14
12
|
import metaflow.parameters
|
13
|
+
import typing
|
14
|
+
import metaflow._vendor.click.types
|
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.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.664921 #
|
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.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.669194 #
|
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.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.691927 #
|
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.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.691448 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
8
8
|
|
9
|
-
import metaflow
|
10
9
|
import typing
|
10
|
+
import metaflow
|
11
11
|
if typing.TYPE_CHECKING:
|
12
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.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.691669 #
|
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.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.666747 #
|
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.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.821819 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -9,13 +9,13 @@ from __future__ import annotations
|
|
9
9
|
import typing
|
10
10
|
if typing.TYPE_CHECKING:
|
11
11
|
import metaflow.plugins.cards.component_serializer
|
12
|
-
import metaflow
|
13
|
-
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core
|
14
12
|
import metaflow.events
|
15
|
-
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.
|
13
|
+
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core
|
16
14
|
import metaflow.metaflow_current
|
17
|
-
import typing
|
18
15
|
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator
|
16
|
+
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator
|
17
|
+
import typing
|
18
|
+
import metaflow
|
19
19
|
|
20
20
|
|
21
21
|
TYPE_CHECKING: bool
|
@@ -228,6 +228,128 @@ class Current(object, metaclass=type):
|
|
228
228
|
def graph(self):
|
229
229
|
...
|
230
230
|
@property
|
231
|
+
def checkpoint(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator.CurrentCheckpointer":
|
232
|
+
"""
|
233
|
+
(only in the presence of the @checkpoint decorator)
|
234
|
+
|
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.
|
237
|
+
|
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
|
+
```
|
284
|
+
|
285
|
+
Returns
|
286
|
+
-------
|
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.
|
350
|
+
"""
|
351
|
+
...
|
352
|
+
@property
|
231
353
|
def parallel(self) -> "metaflow.metaflow_current.Parallel":
|
232
354
|
"""
|
233
355
|
(only in the presence of the @parallel decorator)
|
@@ -344,141 +466,6 @@ class Current(object, metaclass=type):
|
|
344
466
|
"""
|
345
467
|
...
|
346
468
|
@property
|
347
|
-
def model(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core.ModelSerializer":
|
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":
|
409
|
-
"""
|
410
|
-
(only in the presence of the @checkpoint decorator)
|
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.
|
418
|
-
|
419
|
-
Usage (Saving Checkpoints):
|
420
|
-
-------
|
421
|
-
```
|
422
|
-
@checkpoint
|
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
|
456
|
-
|
457
|
-
model = create_model(self.parameters, checkpoint_path = checkpoint_path)
|
458
|
-
for i in range(self.epochs):
|
459
|
-
...
|
460
|
-
```
|
461
|
-
|
462
|
-
Returns
|
463
|
-
-------
|
464
|
-
CurrentCheckpointer
|
465
|
-
The object for handling checkpointing within a step.
|
466
|
-
"""
|
467
|
-
...
|
468
|
-
@property
|
469
|
-
def trigger(self) -> "metaflow.events.Trigger":
|
470
|
-
"""
|
471
|
-
(only in the presence of the @trigger_on_finish, or @trigger decorators)
|
472
|
-
|
473
|
-
Returns `Trigger` if the current run is triggered by an event
|
474
|
-
|
475
|
-
Returns
|
476
|
-
-------
|
477
|
-
Trigger
|
478
|
-
`Trigger` if triggered by an event
|
479
|
-
"""
|
480
|
-
...
|
481
|
-
@property
|
482
469
|
def project_name(self) -> str:
|
483
470
|
"""
|
484
471
|
(only in the presence of the @project decorator)
|
@@ -545,6 +532,19 @@ class Current(object, metaclass=type):
|
|
545
532
|
True if the flow is deployed with `--production`.
|
546
533
|
"""
|
547
534
|
...
|
535
|
+
@property
|
536
|
+
def trigger(self) -> "metaflow.events.Trigger":
|
537
|
+
"""
|
538
|
+
(only in the presence of the @trigger, or @trigger_on_finish 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.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.675198 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.684180 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.684238 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.684291 #
|
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.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.739573 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.
|
4
|
-
# Generated on 2025-03-
|
3
|
+
# MF version: 2.15.6.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-17T20:51:51.740192 #
|
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
|
|