ob-metaflow-stubs 6.0.3.158__py2.py3-none-any.whl → 6.0.3.159__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 +768 -768
- metaflow-stubs/cards.pyi +1 -1
- metaflow-stubs/cli.pyi +1 -1
- metaflow-stubs/cli_components/__init__.pyi +1 -1
- metaflow-stubs/cli_components/utils.pyi +1 -1
- metaflow-stubs/client/__init__.pyi +1 -1
- metaflow-stubs/client/core.pyi +4 -4
- metaflow-stubs/client/filecache.pyi +1 -1
- metaflow-stubs/events.pyi +1 -1
- metaflow-stubs/exception.pyi +1 -1
- metaflow-stubs/flowspec.pyi +3 -3
- metaflow-stubs/generated_for.txt +1 -1
- metaflow-stubs/includefile.pyi +2 -2
- metaflow-stubs/info_file.pyi +1 -1
- metaflow-stubs/metadata_provider/__init__.pyi +1 -1
- metaflow-stubs/metadata_provider/heartbeat.pyi +1 -1
- metaflow-stubs/metadata_provider/metadata.pyi +1 -1
- metaflow-stubs/metadata_provider/util.pyi +1 -1
- metaflow-stubs/metaflow_config.pyi +1 -1
- metaflow-stubs/metaflow_current.pyi +119 -119
- metaflow-stubs/mf_extensions/__init__.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/__init__.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/__init__.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/__init__.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/__init__.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/async_cards.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/deco_injection_mixin.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/extra_components.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/__init__.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/__init__.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/checkpoint_lister.pyi +3 -3
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/lineage_card.pyi +1 -1
- 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 +1 -1
- 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 +1 -1
- 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 +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/__init__.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/context.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/core.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/decorator.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/exceptions.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/task_utils.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/utils.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastructures.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/exceptions.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/__init__.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/hf_hub/decorator.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/__init__.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/core.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/exceptions.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/modeling_utils/model_storage.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/__init__.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/flowspec_utils.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/general.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/identity_utils.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/__init__.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/base.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/serialization_handler/tar.pyi +1 -1
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/utils/tar_utils.pyi +1 -1
- metaflow-stubs/mf_extensions/outerbounds/__init__.pyi +1 -1
- metaflow-stubs/mf_extensions/outerbounds/plugins/__init__.pyi +1 -1
- metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/__init__.pyi +1 -1
- metaflow-stubs/mf_extensions/outerbounds/plugins/card_utilities/injector.pyi +1 -1
- metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/__init__.pyi +1 -1
- metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/ollama.pyi +1 -1
- metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/__init__.pyi +1 -1
- metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/snowflake.pyi +1 -1
- metaflow-stubs/mf_extensions/outerbounds/profilers/__init__.pyi +1 -1
- metaflow-stubs/mf_extensions/outerbounds/profilers/gpu.pyi +1 -1
- metaflow-stubs/mf_extensions/outerbounds/remote_config.pyi +1 -1
- metaflow-stubs/mf_extensions/outerbounds/toplevel/__init__.pyi +1 -1
- metaflow-stubs/mf_extensions/outerbounds/toplevel/global_aliases_for_metaflow_package.pyi +1 -1
- metaflow-stubs/multicore_utils.pyi +1 -1
- metaflow-stubs/parameters.pyi +2 -2
- metaflow-stubs/plugins/__init__.pyi +10 -10
- metaflow-stubs/plugins/airflow/__init__.pyi +1 -1
- metaflow-stubs/plugins/airflow/airflow_utils.pyi +1 -1
- metaflow-stubs/plugins/airflow/exception.pyi +1 -1
- metaflow-stubs/plugins/airflow/sensors/__init__.pyi +1 -1
- metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi +1 -1
- metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi +1 -1
- metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi +1 -1
- metaflow-stubs/plugins/argo/__init__.pyi +1 -1
- metaflow-stubs/plugins/argo/argo_client.pyi +1 -1
- metaflow-stubs/plugins/argo/argo_events.pyi +1 -1
- metaflow-stubs/plugins/argo/argo_workflows.pyi +2 -2
- metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +2 -2
- metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +1 -1
- metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi +2 -2
- metaflow-stubs/plugins/aws/__init__.pyi +1 -1
- metaflow-stubs/plugins/aws/aws_client.pyi +1 -1
- metaflow-stubs/plugins/aws/aws_utils.pyi +1 -1
- metaflow-stubs/plugins/aws/batch/__init__.pyi +1 -1
- metaflow-stubs/plugins/aws/batch/batch.pyi +1 -1
- metaflow-stubs/plugins/aws/batch/batch_client.pyi +1 -1
- metaflow-stubs/plugins/aws/batch/batch_decorator.pyi +1 -1
- metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi +1 -1
- metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi +2 -2
- metaflow-stubs/plugins/aws/step_functions/__init__.pyi +1 -1
- metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi +1 -1
- metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi +1 -1
- metaflow-stubs/plugins/aws/step_functions/step_functions.pyi +1 -1
- metaflow-stubs/plugins/aws/step_functions/step_functions_client.pyi +1 -1
- 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 +1 -1
- metaflow-stubs/plugins/azure/azure_credential.pyi +1 -1
- metaflow-stubs/plugins/azure/azure_exceptions.pyi +1 -1
- metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi +2 -2
- metaflow-stubs/plugins/azure/azure_utils.pyi +1 -1
- metaflow-stubs/plugins/azure/blob_service_client_factory.pyi +1 -1
- metaflow-stubs/plugins/azure/includefile_support.pyi +1 -1
- metaflow-stubs/plugins/cards/__init__.pyi +5 -5
- metaflow-stubs/plugins/cards/card_client.pyi +1 -1
- metaflow-stubs/plugins/cards/card_creator.pyi +1 -1
- metaflow-stubs/plugins/cards/card_datastore.pyi +1 -1
- metaflow-stubs/plugins/cards/card_decorator.pyi +1 -1
- metaflow-stubs/plugins/cards/card_modules/__init__.pyi +1 -1
- metaflow-stubs/plugins/cards/card_modules/basic.pyi +2 -2
- metaflow-stubs/plugins/cards/card_modules/card.pyi +1 -1
- metaflow-stubs/plugins/cards/card_modules/components.pyi +2 -2
- metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi +1 -1
- metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi +1 -1
- metaflow-stubs/plugins/cards/card_modules/test_cards.pyi +1 -1
- metaflow-stubs/plugins/cards/card_resolver.pyi +1 -1
- metaflow-stubs/plugins/cards/component_serializer.pyi +1 -1
- metaflow-stubs/plugins/cards/exception.pyi +1 -1
- metaflow-stubs/plugins/catch_decorator.pyi +1 -1
- metaflow-stubs/plugins/datatools/__init__.pyi +1 -1
- metaflow-stubs/plugins/datatools/local.pyi +1 -1
- metaflow-stubs/plugins/datatools/s3/__init__.pyi +1 -1
- metaflow-stubs/plugins/datatools/s3/s3.pyi +2 -2
- metaflow-stubs/plugins/datatools/s3/s3tail.pyi +1 -1
- metaflow-stubs/plugins/datatools/s3/s3util.pyi +1 -1
- metaflow-stubs/plugins/debug_logger.pyi +1 -1
- metaflow-stubs/plugins/debug_monitor.pyi +1 -1
- metaflow-stubs/plugins/environment_decorator.pyi +1 -1
- metaflow-stubs/plugins/events_decorator.pyi +1 -1
- metaflow-stubs/plugins/frameworks/__init__.pyi +1 -1
- metaflow-stubs/plugins/frameworks/pytorch.pyi +1 -1
- metaflow-stubs/plugins/gcp/__init__.pyi +1 -1
- metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi +2 -2
- metaflow-stubs/plugins/gcp/gs_exceptions.pyi +1 -1
- metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi +1 -1
- metaflow-stubs/plugins/gcp/gs_utils.pyi +1 -1
- metaflow-stubs/plugins/gcp/includefile_support.pyi +1 -1
- metaflow-stubs/plugins/kubernetes/__init__.pyi +1 -1
- metaflow-stubs/plugins/kubernetes/kube_utils.pyi +1 -1
- metaflow-stubs/plugins/kubernetes/kubernetes.pyi +1 -1
- metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +1 -1
- metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +1 -1
- metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi +1 -1
- metaflow-stubs/plugins/kubernetes/spot_monitor_sidecar.pyi +1 -1
- metaflow-stubs/plugins/ollama/__init__.pyi +1 -1
- metaflow-stubs/plugins/parallel_decorator.pyi +1 -1
- metaflow-stubs/plugins/perimeters.pyi +1 -1
- metaflow-stubs/plugins/project_decorator.pyi +1 -1
- metaflow-stubs/plugins/pypi/__init__.pyi +2 -2
- metaflow-stubs/plugins/pypi/conda_decorator.pyi +1 -1
- metaflow-stubs/plugins/pypi/conda_environment.pyi +2 -2
- metaflow-stubs/plugins/pypi/parsers.pyi +1 -1
- metaflow-stubs/plugins/pypi/pypi_decorator.pyi +1 -1
- metaflow-stubs/plugins/pypi/pypi_environment.pyi +1 -1
- metaflow-stubs/plugins/pypi/utils.pyi +1 -1
- metaflow-stubs/plugins/resources_decorator.pyi +1 -1
- metaflow-stubs/plugins/retry_decorator.pyi +1 -1
- metaflow-stubs/plugins/secrets/__init__.pyi +2 -2
- metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +2 -2
- metaflow-stubs/plugins/secrets/secrets_decorator.pyi +1 -1
- metaflow-stubs/plugins/snowflake/__init__.pyi +1 -1
- metaflow-stubs/plugins/storage_executor.pyi +1 -1
- metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +1 -1
- metaflow-stubs/plugins/timeout_decorator.pyi +1 -1
- metaflow-stubs/profilers/__init__.pyi +1 -1
- metaflow-stubs/pylint_wrapper.pyi +1 -1
- metaflow-stubs/runner/__init__.pyi +1 -1
- metaflow-stubs/runner/deployer.pyi +29 -29
- metaflow-stubs/runner/deployer_impl.pyi +1 -1
- metaflow-stubs/runner/metaflow_runner.pyi +2 -2
- metaflow-stubs/runner/nbdeploy.pyi +1 -1
- metaflow-stubs/runner/nbrun.pyi +1 -1
- metaflow-stubs/runner/subprocess_manager.pyi +1 -1
- metaflow-stubs/runner/utils.pyi +2 -2
- metaflow-stubs/system/__init__.pyi +1 -1
- metaflow-stubs/system/system_logger.pyi +1 -1
- metaflow-stubs/system/system_monitor.pyi +1 -1
- metaflow-stubs/tagging_util.pyi +1 -1
- metaflow-stubs/tuple_util.pyi +1 -1
- metaflow-stubs/user_configs/__init__.pyi +1 -1
- 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 +6 -6
- {ob_metaflow_stubs-6.0.3.158.dist-info → ob_metaflow_stubs-6.0.3.159.dist-info}/METADATA +1 -1
- ob_metaflow_stubs-6.0.3.159.dist-info/RECORD +200 -0
- ob_metaflow_stubs-6.0.3.158.dist-info/RECORD +0 -200
- {ob_metaflow_stubs-6.0.3.158.dist-info → ob_metaflow_stubs-6.0.3.159.dist-info}/WHEEL +0 -0
- {ob_metaflow_stubs-6.0.3.158.dist-info → ob_metaflow_stubs-6.0.3.159.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
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.805750 #
|
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
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.820911 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.819708 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.845742 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.809532 #
|
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
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.813917 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -10,11 +10,11 @@ import typing
|
|
10
10
|
import metaflow
|
11
11
|
if typing.TYPE_CHECKING:
|
12
12
|
import datetime
|
13
|
-
import
|
13
|
+
import metaflow.client.core
|
14
14
|
import typing
|
15
|
-
import
|
15
|
+
import tarfile
|
16
16
|
import metaflow.events
|
17
|
-
import
|
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
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.840707 #
|
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
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.806338 #
|
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
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.798350 #
|
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
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.803012 #
|
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.datastore.inputs
|
13
|
-
import metaflow.exception
|
14
13
|
import typing
|
15
|
-
import metaflow.
|
14
|
+
import metaflow.exception
|
16
15
|
import metaflow.unbounded_foreach
|
16
|
+
import metaflow.flowspec
|
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.2+obcheckpoint(0.2.1);ob(v1) 2025-04-
|
1
|
+
2.15.7.2+obcheckpoint(0.2.1);ob(v1) 2025-04-18T04:20:36.794762
|
metaflow-stubs/includefile.pyi
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.808878 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -9,8 +9,8 @@ from __future__ import annotations
|
|
9
9
|
import typing
|
10
10
|
import metaflow
|
11
11
|
if typing.TYPE_CHECKING:
|
12
|
-
import typing
|
13
12
|
import metaflow._vendor.click.types
|
13
|
+
import typing
|
14
14
|
import metaflow.parameters
|
15
15
|
|
16
16
|
from ._vendor import click as click
|
metaflow-stubs/info_file.pyi
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.797520 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.801903 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.825212 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.824734 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.824956 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.799441 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.938461 #
|
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.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator
|
13
|
-
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator
|
14
|
-
import metaflow
|
15
|
-
import typing
|
16
12
|
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core
|
17
|
-
import metaflow.events
|
18
13
|
import metaflow.metaflow_current
|
14
|
+
import typing
|
15
|
+
import metaflow
|
16
|
+
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator
|
17
|
+
import metaflow.events
|
18
|
+
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator
|
19
19
|
|
20
20
|
|
21
21
|
TYPE_CHECKING: bool
|
@@ -228,82 +228,64 @@ class Current(object, metaclass=type):
|
|
228
228
|
def graph(self):
|
229
229
|
...
|
230
230
|
@property
|
231
|
-
def
|
231
|
+
def model(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core.ModelSerializer":
|
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 @model decorator)
|
239
234
|
|
240
|
-
|
235
|
+
The object used for loading / saving models.
|
236
|
+
`current.model` exposes a `save` method to save models and a `load` method to load models.
|
237
|
+
`current.model.loaded` exposes the paths to the models loaded via the `load` argument in the @model decorator
|
238
|
+
or models loaded via `current.model.load`.
|
241
239
|
|
242
|
-
|
240
|
+
Usage (Saving a model):
|
243
241
|
-------
|
244
|
-
CardComponentCollector
|
245
|
-
The or one of the cards attached to this step.
|
246
|
-
"""
|
247
|
-
...
|
248
|
-
@property
|
249
|
-
def checkpoint(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator.CurrentCheckpointer":
|
250
|
-
"""
|
251
|
-
(only in the presence of the @checkpoint decorator)
|
252
242
|
|
253
|
-
The `@checkpoint` decorator makes saving/loading checkpoints available through the `current.checkpoint`.
|
254
|
-
The object exposes `save`/`load`/`list` methods for saving/loading checkpoints.
|
255
|
-
|
256
|
-
You can check if a checkpoint is loaded by `current.checkpoint.is_loaded` and get the checkpoint information
|
257
|
-
by using `current.checkpoint.info`. The `current.checkpoint.directory` returns the path to the checkpoint directory
|
258
|
-
where the checkpoint maybe loaded or saved.
|
259
|
-
|
260
|
-
Usage (Saving Checkpoints):
|
261
|
-
-------
|
262
243
|
```
|
263
|
-
@
|
244
|
+
@model
|
264
245
|
@step
|
265
246
|
def train(self):
|
266
|
-
model
|
267
|
-
|
268
|
-
|
269
|
-
|
270
|
-
|
271
|
-
|
272
|
-
|
273
|
-
|
274
|
-
|
275
|
-
|
276
|
-
|
277
|
-
|
278
|
-
|
279
|
-
|
280
|
-
|
281
|
-
|
282
|
-
|
247
|
+
# current.model.save returns a dictionary reference to the model saved
|
248
|
+
self.my_model = current.model.save(
|
249
|
+
path_to_my_model,
|
250
|
+
label="my_model",
|
251
|
+
metadata={
|
252
|
+
"epochs": 10,
|
253
|
+
"batch-size": 32,
|
254
|
+
"learning-rate": 0.001,
|
255
|
+
}
|
256
|
+
)
|
257
|
+
self.next(self.test)
|
258
|
+
|
259
|
+
@model(load="my_model")
|
260
|
+
@step
|
261
|
+
def test(self):
|
262
|
+
# `current.model.loaded` returns a dictionary of the loaded models
|
263
|
+
# where the key is the name of the artifact and the value is the path to the model
|
264
|
+
print(os.listdir(current.model.loaded["my_model"]))
|
265
|
+
self.next(self.end)
|
283
266
|
```
|
284
|
-
|
267
|
+
|
268
|
+
Usage (Loading models):
|
285
269
|
-------
|
270
|
+
|
286
271
|
```
|
287
|
-
@retry(times=3)
|
288
|
-
@checkpoint
|
289
272
|
@step
|
290
273
|
def train(self):
|
291
|
-
#
|
292
|
-
|
293
|
-
|
294
|
-
|
295
|
-
|
296
|
-
|
297
|
-
|
298
|
-
|
299
|
-
for i in range(self.epochs):
|
300
|
-
...
|
274
|
+
# current.model.load returns the path to the model loaded
|
275
|
+
checkpoint_path = current.model.load(
|
276
|
+
self.checkpoint_key,
|
277
|
+
)
|
278
|
+
model_path = current.model.load(
|
279
|
+
self.model,
|
280
|
+
)
|
281
|
+
self.next(self.test)
|
301
282
|
```
|
302
283
|
|
284
|
+
|
303
285
|
Returns
|
304
286
|
-------
|
305
|
-
|
306
|
-
The object for
|
287
|
+
ModelSerializer
|
288
|
+
The object used for loading / saving models.
|
307
289
|
"""
|
308
290
|
...
|
309
291
|
@property
|
@@ -376,64 +358,82 @@ class Current(object, metaclass=type):
|
|
376
358
|
"""
|
377
359
|
...
|
378
360
|
@property
|
379
|
-
def
|
361
|
+
def card(self) -> "metaflow.plugins.cards.component_serializer.CardComponentCollector":
|
380
362
|
"""
|
381
|
-
(only in the presence of the @
|
363
|
+
(only in the presence of the @card decorator)
|
382
364
|
|
383
|
-
The
|
384
|
-
|
385
|
-
|
386
|
-
|
365
|
+
The `@card` decorator makes the cards available through the `current.card`
|
366
|
+
object. If multiple `@card` decorators are present, you can add an `ID` to
|
367
|
+
distinguish between them using `@card(id=ID)` as the decorator. You will then
|
368
|
+
be able to access that specific card using `current.card[ID].
|
387
369
|
|
388
|
-
|
370
|
+
Methods available are `append` and `extend`
|
371
|
+
|
372
|
+
Returns
|
389
373
|
-------
|
374
|
+
CardComponentCollector
|
375
|
+
The or one of the cards attached to this step.
|
376
|
+
"""
|
377
|
+
...
|
378
|
+
@property
|
379
|
+
def checkpoint(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator.CurrentCheckpointer":
|
380
|
+
"""
|
381
|
+
(only in the presence of the @checkpoint decorator)
|
382
|
+
|
383
|
+
The `@checkpoint` decorator makes saving/loading checkpoints available through the `current.checkpoint`.
|
384
|
+
The object exposes `save`/`load`/`list` methods for saving/loading checkpoints.
|
390
385
|
|
386
|
+
You can check if a checkpoint is loaded by `current.checkpoint.is_loaded` and get the checkpoint information
|
387
|
+
by using `current.checkpoint.info`. The `current.checkpoint.directory` returns the path to the checkpoint directory
|
388
|
+
where the checkpoint maybe loaded or saved.
|
389
|
+
|
390
|
+
Usage (Saving Checkpoints):
|
391
|
+
-------
|
391
392
|
```
|
392
|
-
@
|
393
|
+
@checkpoint
|
393
394
|
@step
|
394
395
|
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)
|
396
|
+
model = create_model(self.parameters, checkpoint_path = None)
|
397
|
+
for i in range(self.epochs):
|
398
|
+
# some training logic
|
399
|
+
loss = model.train(self.dataset)
|
400
|
+
if i % 10 == 0:
|
401
|
+
model.save(
|
402
|
+
current.checkpoint.directory,
|
403
|
+
)
|
404
|
+
# saves the contents of the `current.checkpoint.directory` as a checkpoint
|
405
|
+
# and returns a reference dictionary to the checkpoint saved in the datastore
|
406
|
+
self.latest_checkpoint = current.checkpoint.save(
|
407
|
+
name="epoch_checkpoint",
|
408
|
+
metadata={
|
409
|
+
"epoch": i,
|
410
|
+
"loss": loss,
|
411
|
+
}
|
412
|
+
)
|
414
413
|
```
|
415
|
-
|
416
|
-
Usage (Loading models):
|
414
|
+
Usage (Using Loaded Checkpoints):
|
417
415
|
-------
|
418
|
-
|
419
416
|
```
|
417
|
+
@retry(times=3)
|
418
|
+
@checkpoint
|
420
419
|
@step
|
421
420
|
def train(self):
|
422
|
-
#
|
423
|
-
|
424
|
-
|
425
|
-
|
426
|
-
|
427
|
-
|
428
|
-
)
|
429
|
-
self.next(self.test)
|
430
|
-
```
|
421
|
+
# Assume that the task has restarted and the previous attempt of the task
|
422
|
+
# saved a checkpoint
|
423
|
+
checkpoint_path = None
|
424
|
+
if current.checkpoint.is_loaded: # Check if a checkpoint is loaded
|
425
|
+
print("Loaded checkpoint from the previous attempt")
|
426
|
+
checkpoint_path = current.checkpoint.directory
|
431
427
|
|
428
|
+
model = create_model(self.parameters, checkpoint_path = checkpoint_path)
|
429
|
+
for i in range(self.epochs):
|
430
|
+
...
|
431
|
+
```
|
432
432
|
|
433
433
|
Returns
|
434
434
|
-------
|
435
|
-
|
436
|
-
The object
|
435
|
+
CurrentCheckpointer
|
436
|
+
The object for handling checkpointing within a step.
|
437
437
|
"""
|
438
438
|
...
|
439
439
|
@property
|
@@ -466,19 +466,6 @@ class Current(object, metaclass=type):
|
|
466
466
|
"""
|
467
467
|
...
|
468
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
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.807739 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.816772 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.816832 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.816889 #
|
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
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.868983 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.869597 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.869030 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.909736 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -9,8 +9,8 @@ from __future__ import annotations
|
|
9
9
|
import typing
|
10
10
|
import metaflow
|
11
11
|
if typing.TYPE_CHECKING:
|
12
|
-
import metaflow.plugins.cards.card_modules.card
|
13
12
|
import metaflow.plugins.cards.card_modules.components
|
13
|
+
import metaflow.plugins.cards.card_modules.card
|
14
14
|
|
15
15
|
from ......plugins.cards.card_modules.components import VegaChart as VegaChart
|
16
16
|
from ......plugins.cards.card_modules.card import MetaflowCardComponent as MetaflowCardComponent
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
3
|
# MF version: 2.15.7.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
-
# Generated on 2025-04-
|
4
|
+
# Generated on 2025-04-18T04:20:36.816942 #
|
5
5
|
######################################################################################################
|
6
6
|
|