ob-metaflow-stubs 6.0.3.165__py2.py3-none-any.whl → 6.0.3.167__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 +670 -670
- 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 +4 -4
- metaflow-stubs/client/filecache.pyi +3 -3
- metaflow-stubs/events.pyi +3 -3
- metaflow-stubs/exception.pyi +2 -2
- metaflow-stubs/flowspec.pyi +4 -4
- metaflow-stubs/generated_for.txt +1 -1
- metaflow-stubs/includefile.pyi +2 -2
- 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 +80 -80
- metaflow-stubs/metaflow_git.pyi +2 -2
- metaflow-stubs/mf_extensions/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/async_cards.pyi +3 -3
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/deco_injection_mixin.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/extra_components.pyi +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 +4 -4
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/cards/lineage_card.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/checkpoint_storage.pyi +2 -2
- 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 +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/decorator.pyi +3 -3
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/exceptions.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/final_api.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/checkpoints/lineage.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/context.pyi +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/decorator.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/exceptions.pyi +2 -2
- metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/datastore/task_utils.pyi +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 +3 -3
- 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 +2 -2
- 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/checkpoint_datastores/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/coreweave.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/plugins/checkpoint_datastores/nebius.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/plugins/ollama/ollama.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/plugins/snowflake/snowflake.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/profilers/__init__.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/profilers/gpu.pyi +2 -2
- metaflow-stubs/mf_extensions/outerbounds/remote_config.pyi +9 -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 +13 -13
- 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 +3 -3
- metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi +3 -3
- metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi +4 -4
- 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 +4 -4
- metaflow-stubs/plugins/aws/step_functions/step_functions_deployer_objects.pyi +2 -2
- metaflow-stubs/plugins/azure/__init__.pyi +2 -2
- metaflow-stubs/plugins/azure/azure_credential.pyi +2 -2
- metaflow-stubs/plugins/azure/azure_exceptions.pyi +2 -2
- metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi +3 -3
- metaflow-stubs/plugins/azure/azure_utils.pyi +2 -2
- metaflow-stubs/plugins/azure/blob_service_client_factory.pyi +2 -2
- metaflow-stubs/plugins/azure/includefile_support.pyi +2 -2
- metaflow-stubs/plugins/cards/__init__.pyi +6 -6
- metaflow-stubs/plugins/cards/card_client.pyi +3 -3
- metaflow-stubs/plugins/cards/card_creator.pyi +2 -2
- metaflow-stubs/plugins/cards/card_datastore.pyi +2 -2
- metaflow-stubs/plugins/cards/card_decorator.pyi +2 -2
- metaflow-stubs/plugins/cards/card_modules/__init__.pyi +2 -2
- metaflow-stubs/plugins/cards/card_modules/basic.pyi +2 -2
- metaflow-stubs/plugins/cards/card_modules/card.pyi +2 -2
- metaflow-stubs/plugins/cards/card_modules/components.pyi +3 -3
- metaflow-stubs/plugins/cards/card_modules/convert_to_native_type.pyi +2 -2
- metaflow-stubs/plugins/cards/card_modules/renderer_tools.pyi +2 -2
- metaflow-stubs/plugins/cards/card_modules/test_cards.pyi +2 -2
- metaflow-stubs/plugins/cards/card_resolver.pyi +2 -2
- metaflow-stubs/plugins/cards/component_serializer.pyi +2 -2
- metaflow-stubs/plugins/cards/exception.pyi +2 -2
- metaflow-stubs/plugins/catch_decorator.pyi +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 +4 -4
- metaflow-stubs/plugins/datatools/s3/s3tail.pyi +2 -2
- metaflow-stubs/plugins/datatools/s3/s3util.pyi +2 -2
- metaflow-stubs/plugins/debug_logger.pyi +2 -2
- metaflow-stubs/plugins/debug_monitor.pyi +2 -2
- metaflow-stubs/plugins/environment_decorator.pyi +2 -2
- metaflow-stubs/plugins/events_decorator.pyi +2 -2
- metaflow-stubs/plugins/frameworks/__init__.pyi +2 -2
- metaflow-stubs/plugins/frameworks/pytorch.pyi +2 -2
- metaflow-stubs/plugins/gcp/__init__.pyi +2 -2
- metaflow-stubs/plugins/gcp/gcp_secret_manager_secrets_provider.pyi +3 -3
- metaflow-stubs/plugins/gcp/gs_exceptions.pyi +2 -2
- metaflow-stubs/plugins/gcp/gs_storage_client_factory.pyi +2 -2
- metaflow-stubs/plugins/gcp/gs_utils.pyi +2 -2
- metaflow-stubs/plugins/gcp/includefile_support.pyi +2 -2
- metaflow-stubs/plugins/kubernetes/__init__.pyi +2 -2
- metaflow-stubs/plugins/kubernetes/kube_utils.pyi +2 -2
- metaflow-stubs/plugins/kubernetes/kubernetes.pyi +2 -2
- metaflow-stubs/plugins/kubernetes/kubernetes_client.pyi +2 -2
- metaflow-stubs/plugins/kubernetes/kubernetes_decorator.pyi +2 -2
- metaflow-stubs/plugins/kubernetes/kubernetes_jobsets.pyi +2 -2
- metaflow-stubs/plugins/kubernetes/spot_monitor_sidecar.pyi +2 -2
- metaflow-stubs/plugins/ollama/__init__.pyi +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 +3 -3
- metaflow-stubs/plugins/pypi/conda_decorator.pyi +2 -2
- metaflow-stubs/plugins/pypi/conda_environment.pyi +5 -5
- metaflow-stubs/plugins/pypi/parsers.pyi +2 -2
- metaflow-stubs/plugins/pypi/pypi_decorator.pyi +2 -2
- metaflow-stubs/plugins/pypi/pypi_environment.pyi +2 -2
- metaflow-stubs/plugins/pypi/utils.pyi +2 -2
- metaflow-stubs/plugins/resources_decorator.pyi +2 -2
- metaflow-stubs/plugins/retry_decorator.pyi +2 -2
- metaflow-stubs/plugins/secrets/__init__.pyi +3 -3
- metaflow-stubs/plugins/secrets/inline_secrets_provider.pyi +3 -3
- metaflow-stubs/plugins/secrets/secrets_decorator.pyi +2 -2
- metaflow-stubs/plugins/snowflake/__init__.pyi +2 -2
- metaflow-stubs/plugins/storage_executor.pyi +2 -2
- metaflow-stubs/plugins/test_unbounded_foreach_decorator.pyi +2 -2
- metaflow-stubs/plugins/timeout_decorator.pyi +2 -2
- metaflow-stubs/plugins/uv/__init__.pyi +2 -2
- metaflow-stubs/plugins/uv/uv_environment.pyi +3 -3
- metaflow-stubs/profilers/__init__.pyi +2 -2
- metaflow-stubs/pylint_wrapper.pyi +2 -2
- metaflow-stubs/runner/__init__.pyi +2 -2
- metaflow-stubs/runner/deployer.pyi +6 -6
- 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 +3 -3
- metaflow-stubs/system/__init__.pyi +2 -2
- metaflow-stubs/system/system_logger.pyi +2 -2
- metaflow-stubs/system/system_monitor.pyi +2 -2
- metaflow-stubs/tagging_util.pyi +2 -2
- metaflow-stubs/tuple_util.pyi +2 -2
- metaflow-stubs/user_configs/__init__.pyi +2 -2
- metaflow-stubs/user_configs/config_decorators.pyi +5 -5
- metaflow-stubs/user_configs/config_options.pyi +2 -2
- metaflow-stubs/user_configs/config_parameters.pyi +5 -5
- {ob_metaflow_stubs-6.0.3.165.dist-info → ob_metaflow_stubs-6.0.3.167.dist-info}/METADATA +1 -1
- ob_metaflow_stubs-6.0.3.167.dist-info/RECORD +206 -0
- ob_metaflow_stubs-6.0.3.165.dist-info/RECORD +0 -206
- {ob_metaflow_stubs-6.0.3.165.dist-info → ob_metaflow_stubs-6.0.3.167.dist-info}/WHEEL +0 -0
- {ob_metaflow_stubs-6.0.3.165.dist-info → ob_metaflow_stubs-6.0.3.167.dist-info}/top_level.txt +0 -0
metaflow-stubs/cards.pyi
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.049294 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
metaflow-stubs/cli.pyi
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.064610 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.063420 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.090493 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.052486 #
|
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.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.056890 #
|
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 metaflow.client.core
|
13
14
|
import tempfile
|
15
|
+
import metaflow.events
|
14
16
|
import tarfile
|
15
|
-
import metaflow.client.core
|
16
17
|
import typing
|
17
|
-
import metaflow.events
|
18
18
|
|
19
19
|
from ..metaflow_current import current as current
|
20
20
|
from ..events import Trigger as Trigger
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.085624 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -9,8 +9,8 @@ from __future__ import annotations
|
|
9
9
|
import metaflow
|
10
10
|
import typing
|
11
11
|
if typing.TYPE_CHECKING:
|
12
|
-
import metaflow.datastore.content_addressed_store
|
13
12
|
import metaflow.exception
|
13
|
+
import metaflow.datastore.content_addressed_store
|
14
14
|
|
15
15
|
from ..exception import MetaflowException as MetaflowException
|
16
16
|
|
metaflow-stubs/events.pyi
CHANGED
@@ -1,15 +1,15 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.048702 #
|
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.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.040700 #
|
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.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.045555 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -9,11 +9,11 @@ from __future__ import annotations
|
|
9
9
|
import typing
|
10
10
|
import metaflow
|
11
11
|
if typing.TYPE_CHECKING:
|
12
|
+
import metaflow.exception
|
13
|
+
import metaflow.flowspec
|
12
14
|
import metaflow.unbounded_foreach
|
13
15
|
import metaflow.datastore.inputs
|
14
16
|
import typing
|
15
|
-
import metaflow.flowspec
|
16
|
-
import metaflow.exception
|
17
17
|
|
18
18
|
from . import parameters as parameters
|
19
19
|
from .parameters import DelayedEvaluationParameter as DelayedEvaluationParameter
|
metaflow-stubs/generated_for.txt
CHANGED
@@ -1 +1 @@
|
|
1
|
-
2.15.11.
|
1
|
+
2.15.11.2+obcheckpoint(0.2.1);ob(v1) 2025-05-13T17:00:07.036982
|
metaflow-stubs/includefile.pyi
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.051835 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
metaflow-stubs/info_file.pyi
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.040014 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.044398 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.068955 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.068459 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.068675 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.041840 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -271,6 +271,8 @@ DEBUG_STUBGEN: bool
|
|
271
271
|
|
272
272
|
DEBUG_USERCONF: bool
|
273
273
|
|
274
|
+
DEBUG_CONDA: bool
|
275
|
+
|
274
276
|
AWS_SANDBOX_ENABLED: bool
|
275
277
|
|
276
278
|
AWS_SANDBOX_STS_ENDPOINT_URL: None
|
@@ -1,21 +1,21 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.189119 #
|
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.plugins.cards.component_serializer
|
11
12
|
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator
|
12
13
|
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core
|
13
|
-
import metaflow
|
14
|
-
import typing
|
15
14
|
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.hf_hub.decorator
|
16
|
-
import metaflow.events
|
17
|
-
import metaflow.plugins.cards.component_serializer
|
18
15
|
import metaflow.metaflow_current
|
16
|
+
import metaflow.events
|
17
|
+
import typing
|
18
|
+
import metaflow
|
19
19
|
|
20
20
|
|
21
21
|
TYPE_CHECKING: bool
|
@@ -246,6 +246,67 @@ class Current(object, metaclass=type):
|
|
246
246
|
"""
|
247
247
|
...
|
248
248
|
@property
|
249
|
+
def model(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.modeling_utils.core.ModelSerializer":
|
250
|
+
"""
|
251
|
+
(only in the presence of the @model decorator)
|
252
|
+
|
253
|
+
The object used for loading / saving models.
|
254
|
+
`current.model` exposes a `save` method to save models and a `load` method to load models.
|
255
|
+
`current.model.loaded` exposes the paths to the models loaded via the `load` argument in the @model decorator
|
256
|
+
or models loaded via `current.model.load`.
|
257
|
+
|
258
|
+
Usage (Saving a model):
|
259
|
+
-------
|
260
|
+
|
261
|
+
```
|
262
|
+
@model
|
263
|
+
@step
|
264
|
+
def train(self):
|
265
|
+
# current.model.save returns a dictionary reference to the model saved
|
266
|
+
self.my_model = current.model.save(
|
267
|
+
path_to_my_model,
|
268
|
+
label="my_model",
|
269
|
+
metadata={
|
270
|
+
"epochs": 10,
|
271
|
+
"batch-size": 32,
|
272
|
+
"learning-rate": 0.001,
|
273
|
+
}
|
274
|
+
)
|
275
|
+
self.next(self.test)
|
276
|
+
|
277
|
+
@model(load="my_model")
|
278
|
+
@step
|
279
|
+
def test(self):
|
280
|
+
# `current.model.loaded` returns a dictionary of the loaded models
|
281
|
+
# where the key is the name of the artifact and the value is the path to the model
|
282
|
+
print(os.listdir(current.model.loaded["my_model"]))
|
283
|
+
self.next(self.end)
|
284
|
+
```
|
285
|
+
|
286
|
+
Usage (Loading models):
|
287
|
+
-------
|
288
|
+
|
289
|
+
```
|
290
|
+
@step
|
291
|
+
def train(self):
|
292
|
+
# current.model.load returns the path to the model loaded
|
293
|
+
checkpoint_path = current.model.load(
|
294
|
+
self.checkpoint_key,
|
295
|
+
)
|
296
|
+
model_path = current.model.load(
|
297
|
+
self.model,
|
298
|
+
)
|
299
|
+
self.next(self.test)
|
300
|
+
```
|
301
|
+
|
302
|
+
|
303
|
+
Returns
|
304
|
+
-------
|
305
|
+
ModelSerializer
|
306
|
+
The object used for loading / saving models.
|
307
|
+
"""
|
308
|
+
...
|
309
|
+
@property
|
249
310
|
def parallel(self) -> "metaflow.metaflow_current.Parallel":
|
250
311
|
"""
|
251
312
|
(only in the presence of the @parallel decorator)
|
@@ -344,67 +405,6 @@ class Current(object, metaclass=type):
|
|
344
405
|
"""
|
345
406
|
...
|
346
407
|
@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
408
|
def checkpoint(self) -> "metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.checkpoints.decorator.CurrentCheckpointer":
|
409
409
|
"""
|
410
410
|
(only in the presence of the @checkpoint decorator)
|
@@ -466,6 +466,19 @@ 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, or @trigger_on_finish 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_on_finish, or @trigger decorators)
|
539
|
-
|
540
|
-
Returns `Trigger` if the current run is triggered by an event
|
541
|
-
|
542
|
-
Returns
|
543
|
-
-------
|
544
|
-
Trigger
|
545
|
-
`Trigger` if triggered by an event
|
546
|
-
"""
|
547
|
-
...
|
548
548
|
...
|
549
549
|
|
550
550
|
current: Current
|
metaflow-stubs/metaflow_git.pyi
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.048913 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.050571 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.059908 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.059968 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.060024 #
|
5
5
|
######################################################################################################
|
6
6
|
|
metaflow-stubs/mf_extensions/obcheckpoint/plugins/machine_learning_utilities/card_utils/__init__.pyi
CHANGED
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.115040 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.115736 #
|
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
|
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.115088 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.159966 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.060084 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,6 +1,6 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.116515 #
|
5
5
|
######################################################################################################
|
6
6
|
|
@@ -1,18 +1,18 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.116563 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
8
8
|
|
9
|
+
import threading
|
9
10
|
import typing
|
10
11
|
import metaflow
|
11
|
-
import threading
|
12
12
|
if typing.TYPE_CHECKING:
|
13
|
-
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.card_utils.async_cards
|
14
13
|
import threading
|
15
14
|
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.datastructures
|
15
|
+
import metaflow.mf_extensions.obcheckpoint.plugins.machine_learning_utilities.card_utils.async_cards
|
16
16
|
|
17
17
|
from ...card_utils.deco_injection_mixin import CardDecoratorInjector as CardDecoratorInjector
|
18
18
|
from ...card_utils.async_cards import CardRefresher as CardRefresher
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.116926 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|
@@ -1,7 +1,7 @@
|
|
1
1
|
######################################################################################################
|
2
2
|
# Auto-generated Metaflow stub file #
|
3
|
-
# MF version: 2.15.11.
|
4
|
-
# Generated on 2025-05-
|
3
|
+
# MF version: 2.15.11.2+obcheckpoint(0.2.1);ob(v1) #
|
4
|
+
# Generated on 2025-05-13T17:00:07.159154 #
|
5
5
|
######################################################################################################
|
6
6
|
|
7
7
|
from __future__ import annotations
|