ob-metaflow-stubs 6.0.3.146__py2.py3-none-any.whl → 6.0.3.148__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 +715 -716
- 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 +3 -3
- metaflow-stubs/client/filecache.pyi +2 -2
- 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 +4 -2
- metaflow-stubs/metaflow_current.pyi +90 -90
- 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 +2 -2
- 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 +2 -2
- 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 +3 -3
- 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 +4 -4
- 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 +4 -4
- 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 +2 -2
- 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 +2 -2
- 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 +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 +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 +15 -11
- metaflow-stubs/plugins/airflow/__init__.pyi +2 -2
- metaflow-stubs/plugins/airflow/airflow_utils.pyi +2 -2
- metaflow-stubs/plugins/airflow/exception.pyi +2 -2
- metaflow-stubs/plugins/airflow/sensors/__init__.pyi +2 -2
- metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi +2 -2
- metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi +2 -2
- metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi +2 -2
- metaflow-stubs/plugins/argo/__init__.pyi +2 -2
- metaflow-stubs/plugins/argo/argo_client.pyi +2 -2
- metaflow-stubs/plugins/argo/argo_events.pyi +2 -2
- metaflow-stubs/plugins/argo/argo_workflows.pyi +2 -2
- metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +3 -3
- metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +3 -3
- metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi +3 -3
- metaflow-stubs/plugins/aws/__init__.pyi +2 -2
- metaflow-stubs/plugins/aws/aws_client.pyi +2 -2
- metaflow-stubs/plugins/aws/aws_utils.pyi +2 -2
- metaflow-stubs/plugins/aws/batch/__init__.pyi +2 -2
- metaflow-stubs/plugins/aws/batch/batch.pyi +2 -2
- metaflow-stubs/plugins/aws/batch/batch_client.pyi +2 -2
- metaflow-stubs/plugins/aws/batch/batch_decorator.pyi +2 -2
- metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi +2 -2
- metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi +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 +3 -3
- 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 +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 +2 -2
- metaflow-stubs/plugins/cards/card_creator.pyi +2 -2
- metaflow-stubs/plugins/cards/card_datastore.pyi +2 -2
- metaflow-stubs/plugins/cards/card_decorator.pyi +2 -2
- metaflow-stubs/plugins/cards/card_modules/__init__.pyi +2 -2
- metaflow-stubs/plugins/cards/card_modules/basic.pyi +2 -2
- metaflow-stubs/plugins/cards/card_modules/card.pyi +2 -2
- metaflow-stubs/plugins/cards/card_modules/components.pyi +2 -2
- 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 +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 +4 -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 +2 -2
- 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 +4 -3
- metaflow-stubs/plugins/pypi/conda_decorator.pyi +2 -2
- metaflow-stubs/plugins/pypi/conda_environment.pyi +3 -3
- metaflow-stubs/plugins/pypi/parsers.pyi +113 -0
- 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 +2 -2
- 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 +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 +4 -4
- metaflow-stubs/runner/deployer_impl.pyi +3 -3
- metaflow-stubs/runner/metaflow_runner.pyi +3 -3
- 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 +4 -4
- metaflow-stubs/system/__init__.pyi +2 -2
- metaflow-stubs/system/system_logger.pyi +3 -3
- metaflow-stubs/system/system_monitor.pyi +2 -2
- metaflow-stubs/tagging_util.pyi +2 -2
- metaflow-stubs/tuple_util.pyi +2 -2
- metaflow-stubs/user_configs/__init__.pyi +2 -2
- metaflow-stubs/user_configs/config_decorators.pyi +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.146.dist-info → ob_metaflow_stubs-6.0.3.148.dist-info}/METADATA +1 -1
- ob_metaflow_stubs-6.0.3.148.dist-info/RECORD +198 -0
- ob_metaflow_stubs-6.0.3.146.dist-info/RECORD +0 -197
- {ob_metaflow_stubs-6.0.3.146.dist-info → ob_metaflow_stubs-6.0.3.148.dist-info}/WHEEL +0 -0
- {ob_metaflow_stubs-6.0.3.146.dist-info → ob_metaflow_stubs-6.0.3.148.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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.963412 #
|
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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.978395 #
|
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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.977135 #
|
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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:42.013958 #
|
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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.967174 #
|
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.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.971845 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -11,10 +11,10 @@ import typing
|
|
11
11
|
if typing.TYPE_CHECKING:
|
12
12
|
import tarfile
|
13
13
|
import tempfile
|
14
|
-
import typing
|
15
14
|
import metaflow.client.core
|
16
15
|
import datetime
|
17
16
|
import metaflow.events
|
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.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:42.006294 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
metaflow-stubs/events.pyi
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.964143 #
|
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.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.955711 #
|
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.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.960406 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -9,10 +9,10 @@ from __future__ import annotations
|
|
9
9
|
import metaflow
|
10
10
|
import typing
|
11
11
|
if typing.TYPE_CHECKING:
|
12
|
+
import metaflow.unbounded_foreach
|
13
|
+
import metaflow.flowspec
|
12
14
|
import typing
|
13
15
|
import metaflow.exception
|
14
|
-
import metaflow.flowspec
|
15
|
-
import metaflow.unbounded_foreach
|
16
16
|
import metaflow.datastore.inputs
|
17
17
|
|
18
18
|
from . import parameters as parameters
|
metaflow-stubs/generated_for.txt
CHANGED
@@ -1 +1 @@
|
|
1
|
-
2.15.
|
1
|
+
2.15.3.1+obcheckpoint(0.1.9);ob(v1) 2025-03-03T22:55:41.951979
|
metaflow-stubs/includefile.pyi
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.966529 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -9,9 +9,9 @@ from __future__ import annotations
|
|
9
9
|
import metaflow
|
10
10
|
import typing
|
11
11
|
if typing.TYPE_CHECKING:
|
12
|
-
import metaflow.parameters
|
13
12
|
import metaflow._vendor.click.types
|
14
13
|
import typing
|
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.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.954905 #
|
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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.959302 #
|
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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.982974 #
|
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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.982464 #
|
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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.982709 #
|
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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.956824 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -209,6 +209,8 @@ KUBERNETES_DISK: None
|
|
209
209
|
|
210
210
|
KUBERNETES_QOS: str
|
211
211
|
|
212
|
+
KUBERNETES_CONDA_ARCH: None
|
213
|
+
|
212
214
|
ARGO_WORKFLOWS_KUBERNETES_SECRETS: str
|
213
215
|
|
214
216
|
ARGO_WORKFLOWS_ENV_VARS_TO_SKIP: str
|
@@ -1,21 +1,21 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:42.141136 #
|
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.metaflow_current
|
12
|
-
import typing
|
13
|
-
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core
|
14
|
-
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator
|
15
|
-
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator
|
16
11
|
import metaflow.plugins.cards.component_serializer
|
17
12
|
import metaflow
|
13
|
+
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core
|
18
14
|
import metaflow.events
|
15
|
+
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator
|
16
|
+
import metaflow.metaflow_current
|
17
|
+
import typing
|
18
|
+
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator
|
19
19
|
|
20
20
|
|
21
21
|
TYPE_CHECKING: bool
|
@@ -257,24 +257,6 @@ class Current(object, metaclass=type):
|
|
257
257
|
"""
|
258
258
|
...
|
259
259
|
@property
|
260
|
-
def card(self) -> "metaflow.plugins.cards.component_serializer.CardComponentCollector":
|
261
|
-
"""
|
262
|
-
(only in the presence of the @card decorator)
|
263
|
-
|
264
|
-
The `@card` decorator makes the cards available through the `current.card`
|
265
|
-
object. If multiple `@card` decorators are present, you can add an `ID` to
|
266
|
-
distinguish between them using `@card(id=ID)` as the decorator. You will then
|
267
|
-
be able to access that specific card using `current.card[ID].
|
268
|
-
|
269
|
-
Methods available are `append` and `extend`
|
270
|
-
|
271
|
-
Returns
|
272
|
-
-------
|
273
|
-
CardComponentCollector
|
274
|
-
The or one of the cards attached to this step.
|
275
|
-
"""
|
276
|
-
...
|
277
|
-
@property
|
278
260
|
def huggingface_hub(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator.HuggingfaceRegistry":
|
279
261
|
"""
|
280
262
|
(only in the presence of the @huggingface_hub decorator)
|
@@ -344,64 +326,21 @@ class Current(object, metaclass=type):
|
|
344
326
|
"""
|
345
327
|
...
|
346
328
|
@property
|
347
|
-
def
|
329
|
+
def card(self) -> "metaflow.plugins.cards.component_serializer.CardComponentCollector":
|
348
330
|
"""
|
349
|
-
(only in the presence of the @
|
350
|
-
|
351
|
-
The `@checkpoint` decorator makes saving/loading checkpoints available through the `current.checkpoint`.
|
352
|
-
The object exposes `save`/`load`/`list` methods for saving/loading checkpoints.
|
353
|
-
|
354
|
-
You can check if a checkpoint is loaded by `current.checkpoint.is_loaded` and get the checkpoint information
|
355
|
-
by using `current.checkpoint.info`. The `current.checkpoint.directory` returns the path to the checkpoint directory
|
356
|
-
where the checkpoint maybe loaded or saved.
|
331
|
+
(only in the presence of the @card decorator)
|
357
332
|
|
358
|
-
|
359
|
-
|
360
|
-
|
361
|
-
|
362
|
-
@step
|
363
|
-
def train(self):
|
364
|
-
model = create_model(self.parameters, checkpoint_path = None)
|
365
|
-
for i in range(self.epochs):
|
366
|
-
# some training logic
|
367
|
-
loss = model.train(self.dataset)
|
368
|
-
if i % 10 == 0:
|
369
|
-
model.save(
|
370
|
-
current.checkpoint.directory,
|
371
|
-
)
|
372
|
-
# saves the contents of the `current.checkpoint.directory` as a checkpoint
|
373
|
-
# and returns a reference dictionary to the checkpoint saved in the datastore
|
374
|
-
self.latest_checkpoint = current.checkpoint.save(
|
375
|
-
name="epoch_checkpoint",
|
376
|
-
metadata={
|
377
|
-
"epoch": i,
|
378
|
-
"loss": loss,
|
379
|
-
}
|
380
|
-
)
|
381
|
-
```
|
382
|
-
Usage (Using Loaded Checkpoints):
|
383
|
-
-------
|
384
|
-
```
|
385
|
-
@retry(times=3)
|
386
|
-
@checkpoint
|
387
|
-
@step
|
388
|
-
def train(self):
|
389
|
-
# Assume that the task has restarted and the previous attempt of the task
|
390
|
-
# saved a checkpoint
|
391
|
-
checkpoint_path = None
|
392
|
-
if current.checkpoint.is_loaded: # Check if a checkpoint is loaded
|
393
|
-
print("Loaded checkpoint from the previous attempt")
|
394
|
-
checkpoint_path = current.checkpoint.directory
|
333
|
+
The `@card` decorator makes the cards available through the `current.card`
|
334
|
+
object. If multiple `@card` decorators are present, you can add an `ID` to
|
335
|
+
distinguish between them using `@card(id=ID)` as the decorator. You will then
|
336
|
+
be able to access that specific card using `current.card[ID].
|
395
337
|
|
396
|
-
|
397
|
-
for i in range(self.epochs):
|
398
|
-
...
|
399
|
-
```
|
338
|
+
Methods available are `append` and `extend`
|
400
339
|
|
401
340
|
Returns
|
402
341
|
-------
|
403
|
-
|
404
|
-
The
|
342
|
+
CardComponentCollector
|
343
|
+
The or one of the cards attached to this step.
|
405
344
|
"""
|
406
345
|
...
|
407
346
|
@property
|
@@ -466,6 +405,80 @@ class Current(object, metaclass=type):
|
|
466
405
|
"""
|
467
406
|
...
|
468
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
|
469
482
|
def project_name(self) -> str:
|
470
483
|
"""
|
471
484
|
(only in the presence of the @project decorator)
|
@@ -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, 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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.965411 #
|
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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.974937 #
|
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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.974998 #
|
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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.975053 #
|
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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:42.051614 #
|
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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:42.052589 #
|
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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:42.051678 #
|
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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:42.110947 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:41.975106 #
|
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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:42.053840 #
|
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-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:42.053905 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -10,8 +10,8 @@ import metaflow
|
|
10
10
|
import typing
|
11
11
|
import threading
|
12
12
|
if typing.TYPE_CHECKING:
|
13
|
-
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
|
14
13
|
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.card_utils.async_cards
|
14
|
+
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
|
15
15
|
import threading
|
16
16
|
|
17
17
|
from ...card_utils.deco_injection_mixin import CardDecoratorInjector as CardDecoratorInjector
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.
|
4
|
-
# Generated on 2025-
|
3
|
+
# MF version: 2.15.3.1+obcheckpoint(0.1.9);ob(v1) #
|
4
|
+
# Generated on 2025-03-03T22:55:42.054504 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|