ob-metaflow-stubs 6.0.4.7__py2.py3-none-any.whl → 6.0.4.8rc1__py2.py3-none-any.whl

metaflow-stubs/packaging_sys/v1.pyi

@@ -0,0 +1,145 @@
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.416384                                                            #
+######################################################################################################
+
+from __future__ import annotations
+
+import metaflow
+import typing
+if typing.TYPE_CHECKING:
+    import metaflow.packaging_sys
+    import typing
+
+from ..exception import MetaflowException as MetaflowException
+from ..user_decorators.user_flow_decorator import FlowMutatorMeta as FlowMutatorMeta
+from ..user_decorators.user_step_decorator import UserStepDecoratorMeta as UserStepDecoratorMeta
+from . import ContentType as ContentType
+from . import MetaflowCodeContentV1Base as MetaflowCodeContentV1Base
+from .distribution_support import modules_to_distributions as modules_to_distributions
+from .utils import suffix_filter as suffix_filter
+from .utils import walk as walk
+
+EXT_EXCLUDE_SUFFIXES: list
+
+MFCONTENT_MARKER: str
+
+class MetaflowCodeContentV1(metaflow.packaging_sys.MetaflowCodeContentV1Base, metaclass=type):
+    def __init__(self, code_dir: str = '.mf_code', other_dir: str = '.mf_meta', criteria: typing.Callable[[module], bool] = ...):
+        ...
+    def create_mfcontent_info(self) -> typing.Dict[str, typing.Any]:
+        ...
+    def get_excluded_tl_entries(self) -> typing.List[str]:
+        """
+        When packaging Metaflow from within an executing Metaflow flow, we need to
+        exclude the files that are inserted by this content from being packaged (possibly).
+        
+        Use this function to return these files or top-level directories.
+        
+        Returns
+        -------
+        List[str]
+            Files or directories to exclude
+        """
+        ...
+    def content_names(self, content_types: typing.Optional[int] = None) -> typing.Generator[typing.Tuple[str, str], None, None]:
+        """
+        Detailed list of the content of this MetaflowCodeContent. This will list all files
+        (or non files -- for the INFO or CONFIG data for example) present in the archive.
+        
+        Parameters
+        ----------
+        content_types : Optional[int]
+            The type of content to get the names of. If None, all content is returned.
+        
+        Yields
+        ------
+        Generator[Tuple[str, str], None, None]
+            Path on the filesystem and the name in the archive
+        """
+        ...
+    def contents(self, content_types: typing.Optional[int] = None) -> typing.Generator[typing.Tuple[typing.Union[bytes, str], str], None, None]:
+        """
+        Very similar to content_names but returns the content of the non-files
+        as well as bytes. For files, identical output as content_names
+        
+        Parameters
+        ----------
+        content_types : Optional[int]
+            The type of content to get the content of. If None, all content is returned.
+        
+        Yields
+        ------
+        Generator[Tuple[Union[str, bytes], str], None, None]
+            Content of the MF content
+        """
+        ...
+    def show(self) -> str:
+        """
+        Returns a more human-readable string representation of the content of this
+        MetaflowCodeContent. This will not, for example, list all files but summarize what
+        is included at a more high level.
+        
+        Returns
+        -------
+        str
+            A human-readable string representation of the content of this MetaflowCodeContent
+        """
+        ...
+    def add_info(self, info: typing.Dict[str, typing.Any]):
+        """
+        Add the content of the INFO file to the Metaflow content
+        
+        Parameters
+        ----------
+        info: Dict[str, Any]
+            The content of the INFO file
+        """
+        ...
+    def add_config(self, config: typing.Dict[str, typing.Any]):
+        """
+        Add the content of the CONFIG file to the Metaflow content
+        
+        Parameters
+        ----------
+        config: Dict[str, Any]
+            The content of the CONFIG file
+        """
+        ...
+    def add_module(self, module: module):
+        """
+        Add a python module to the Metaflow content
+        
+        Parameters
+        ----------
+        module_path: ModuleType
+            The module to add
+        """
+        ...
+    def add_code_file(self, file_path: str, file_name: str):
+        """
+        Add a code file to the Metaflow content
+        
+        Parameters
+        ----------
+        file_path: str
+            The path to the code file to add (on the filesystem)
+        file_name: str
+            The path in the archive to add the code file to
+        """
+        ...
+    def add_other_file(self, file_path: str, file_name: str):
+        """
+        Add a non-python file to the Metaflow content
+        
+        Parameters
+        ----------
+        file_path: str
+            The path to the file to add (on the filesystem)
+        file_name: str
+            The path in the archive to add the file to
+        """
+        ...
+    ...
+

metaflow-stubs/parameters.pyi

@@ -1,17 +1,17 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.851676                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.385362                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import typing
 import metaflow
+import typing
 if typing.TYPE_CHECKING:
     import typing
-    import metaflow._vendor.click.types
     import metaflow.parameters
+    import metaflow._vendor.click.types
 
 from ._vendor import click as click
 from .exception import ParameterFieldFailed as ParameterFieldFailed

metaflow-stubs/plugins/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.858628                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.393457                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -9,25 +9,25 @@ from __future__ import annotations
 
 from . import test_unbounded_foreach_decorator as test_unbounded_foreach_decorator
 from .test_unbounded_foreach_decorator import InternalTestUnboundedForeachInput as InternalTestUnboundedForeachInput
-from . import argo as argo
-from . import secrets as secrets
+from . import resources_decorator as resources_decorator
+from . import kubernetes as kubernetes
+from . import timeout_decorator as timeout_decorator
+from . import aws as aws
 from . import datatools as datatools
 from . import gcp as gcp
 from . import storage_executor as storage_executor
 from . import cards as cards
-from . import parallel_decorator as parallel_decorator
-from . import timeout_decorator as timeout_decorator
 from . import pypi as pypi
-from . import resources_decorator as resources_decorator
+from . import parallel_decorator as parallel_decorator
+from . import argo as argo
 from . import retry_decorator as retry_decorator
-from . import catch_decorator as catch_decorator
 from . import environment_decorator as environment_decorator
-from . import kubernetes as kubernetes
-from . import aws as aws
+from . import catch_decorator as catch_decorator
 from . import frameworks as frameworks
+from . import secrets as secrets
 from . import airflow as airflow
-from . import project_decorator as project_decorator
 from . import events_decorator as events_decorator
+from . import project_decorator as project_decorator
 from . import exit_hook as exit_hook
 from . import uv as uv
 from . import azure as azure

metaflow-stubs/plugins/airflow/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.893080                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.433369                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/airflow/airflow_utils.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.930578                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.469626                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/airflow/exception.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.929684                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.468743                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/airflow/sensors/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.930820                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.469877                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/airflow/sensors/base_sensor.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.984373                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.521095                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/airflow/sensors/external_task_sensor.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.984673                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.521364                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/airflow/sensors/s3_sensor.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.984904                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.521591                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/argo/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.888812                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.431733                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/argo/argo_client.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.988092                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.534017                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/argo/argo_events.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.914341                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.466333                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/argo/argo_workflows.pyi

@@ -1,13 +1,13 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.992186                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.538280                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import typing
 import metaflow
+import typing
 if typing.TYPE_CHECKING:
     import metaflow.graph
     import metaflow.exception
@@ -107,7 +107,7 @@ class ArgoWorkflowsSchedulingException(metaflow.exception.MetaflowException, met
     ...
 
 class ArgoWorkflows(object, metaclass=type):
-    def __init__(self, name, graph: metaflow.graph.FlowGraph, flow, code_package_sha, code_package_url, production_token, metadata, flow_datastore, environment, event_logger, monitor, tags = None, namespace = None, username = None, max_workers = None, workflow_timeout = None, workflow_priority = None, auto_emit_argo_events = False, notify_on_error = False, notify_on_success = False, notify_slack_webhook_url = None, notify_pager_duty_integration_key = None, notify_incident_io_api_key = None, incident_io_alert_source_config_id = None, incident_io_metadata: typing.List[str] = None, enable_heartbeat_daemon = True, enable_error_msg_capture = False):
+    def __init__(self, name, graph: metaflow.graph.FlowGraph, flow, code_package_metadata, code_package_sha, code_package_url, production_token, metadata, flow_datastore, environment, event_logger, monitor, tags = None, namespace = None, username = None, max_workers = None, workflow_timeout = None, workflow_priority = None, auto_emit_argo_events = False, notify_on_error = False, notify_on_success = False, notify_slack_webhook_url = None, notify_pager_duty_integration_key = None, notify_incident_io_api_key = None, incident_io_alert_source_config_id = None, incident_io_metadata: typing.List[str] = None, enable_heartbeat_daemon = True, enable_error_msg_capture = False):
         ...
     def __str__(self):
         ...

metaflow-stubs/plugins/argo/argo_workflows_decorator.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.914662                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.466624                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/argo/argo_workflows_deployer.pyi

@@ -1,13 +1,13 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.915087                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.467068                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import typing
 import metaflow
+import typing
 if typing.TYPE_CHECKING:
     import typing
     import metaflow.plugins.argo.argo_workflows_deployer_objects

metaflow-stubs/plugins/argo/argo_workflows_deployer_objects.pyi

@@ -1,13 +1,13 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.980020                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.520710                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import typing
 import metaflow
+import typing
 if typing.TYPE_CHECKING:
     import metaflow.runner.deployer
     import metaflow.plugins.argo.argo_workflows_deployer_objects

metaflow-stubs/plugins/argo/exit_hooks.pyi

@@ -1,13 +1,13 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:15.004528                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.550434                                                            #
 ######################################################################################################
 
 from __future__ import annotations
 
-import typing
 import metaflow
+import typing
 if typing.TYPE_CHECKING:
     import metaflow.plugins.argo.exit_hooks
 

metaflow-stubs/plugins/aws/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.892608                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.429850                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/aws/aws_client.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.928982                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.456603                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/aws/aws_utils.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.928263                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.455895                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/aws/batch/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.928738                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.456316                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/aws/batch/batch.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.983692                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.488219                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -65,9 +65,9 @@ class Batch(object, metaclass=type):
         ...
     def kill_jobs(self, flow_name, run_id, user, echo):
         ...
-    def create_job(self, step_name, step_cli, task_spec, code_package_sha, code_package_url, code_package_ds, image, queue, iam_role = None, execution_role = None, cpu = None, gpu = None, memory = None, run_time_limit = None, shared_memory = None, max_swap = None, swappiness = None, inferentia = None, efa = None, env = {}, attrs = {}, host_volumes = None, efs_volumes = None, use_tmpfs = None, tmpfs_tempdir = None, tmpfs_size = None, tmpfs_path = None, num_parallel = 0, ephemeral_storage = None, log_driver = None, log_options = None):
+    def create_job(self, step_name, step_cli, task_spec, code_package_metadata, code_package_sha, code_package_url, code_package_ds, image, queue, iam_role = None, execution_role = None, cpu = None, gpu = None, memory = None, run_time_limit = None, shared_memory = None, max_swap = None, swappiness = None, inferentia = None, efa = None, env = {}, attrs = {}, host_volumes = None, efs_volumes = None, use_tmpfs = None, tmpfs_tempdir = None, tmpfs_size = None, tmpfs_path = None, num_parallel = 0, ephemeral_storage = None, log_driver = None, log_options = None):
         ...
-    def launch_job(self, step_name, step_cli, task_spec, code_package_sha, code_package_url, code_package_ds, image, queue, iam_role = None, execution_role = None, cpu = None, gpu = None, memory = None, run_time_limit = None, shared_memory = None, max_swap = None, swappiness = None, inferentia = None, efa = None, host_volumes = None, efs_volumes = None, use_tmpfs = None, tmpfs_tempdir = None, tmpfs_size = None, tmpfs_path = None, num_parallel = 0, env = {}, attrs = {}, ephemeral_storage = None, log_driver = None, log_options = None):
+    def launch_job(self, step_name, step_cli, task_spec, code_package_metadata, code_package_sha, code_package_url, code_package_ds, image, queue, iam_role = None, execution_role = None, cpu = None, gpu = None, memory = None, run_time_limit = None, shared_memory = None, max_swap = None, swappiness = None, inferentia = None, efa = None, host_volumes = None, efs_volumes = None, use_tmpfs = None, tmpfs_tempdir = None, tmpfs_size = None, tmpfs_path = None, num_parallel = 0, env = {}, attrs = {}, ephemeral_storage = None, log_driver = None, log_options = None):
         ...
     def wait(self, stdout_location, stderr_location, echo = None):
         ...

metaflow-stubs/plugins/aws/batch/batch_client.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.982966                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.487544                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/aws/batch/batch_decorator.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.944519                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.483670                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -32,6 +32,8 @@ ECS_FARGATE_EXECUTION_ROLE: None
 
 ECS_S3_ACCESS_IAM_ROLE: None
 
+FEAT_ALWAYS_UPLOAD_CODE_PACKAGE: bool
+
 UBF_CONTROL: str
 
 class BatchDecorator(metaflow.decorators.StepDecorator, metaclass=type):

metaflow-stubs/plugins/aws/secrets_manager/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.929192                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.456786                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/aws/secrets_manager/aws_secrets_manager_secrets_provider.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.984020                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.488519                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -10,8 +10,8 @@ import metaflow
 import abc
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.plugins.secrets
     import abc
+    import metaflow.plugins.secrets
     import metaflow.exception
 
 from ....exception import MetaflowException as MetaflowException

metaflow-stubs/plugins/aws/step_functions/__init__.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.928491                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.456106                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/aws/step_functions/event_bridge_client.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:15.006115                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.550739                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/aws/step_functions/schedule_decorator.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:14.981072                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.485962                                                            #
 ######################################################################################################
 
 from __future__ import annotations

metaflow-stubs/plugins/aws/step_functions/step_functions.pyi

@@ -1,7 +1,7 @@
 ######################################################################################################
 #                                 Auto-generated Metaflow stub file                                  #
-# MF version: 2.15.21.4+obcheckpoint(0.2.4);ob(v1)                                                   #
-# Generated on 2025-07-25T18:05:15.005824                                                            #
+# MF version: 2.16.5.1+obcheckpoint(0.2.4);ob(v1)                                                    #
+# Generated on 2025-07-28T18:04:46.539937                                                            #
 ######################################################################################################
 
 from __future__ import annotations
@@ -37,7 +37,7 @@ class StepFunctionsSchedulingException(metaflow.exception.MetaflowException, met
     ...
 
 class StepFunctions(object, metaclass=type):
-    def __init__(self, name, graph, flow, code_package_sha, code_package_url, production_token, metadata, flow_datastore, environment, event_logger, monitor, tags = None, namespace = None, username = None, max_workers = None, workflow_timeout = None, is_project = False, use_distributed_map = False):
+    def __init__(self, name, graph, flow, code_package_metadata, code_package_sha, code_package_url, production_token, metadata, flow_datastore, environment, event_logger, monitor, tags = None, namespace = None, username = None, max_workers = None, workflow_timeout = None, is_project = False, use_distributed_map = False):
         ...
     def to_json(self):
         ...