metaflow-stubs 2.12.28__py2.py3-none-any.whl → 2.12.29__py2.py3-none-any.whl

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

@@ -0,0 +1,127 @@
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.29                                                                                #
+# Generated on 2024-11-07T22:19:34.645856                                                            #
+######################################################################################################
+
+from __future__ import annotations
+
+import typing
+import metaflow
+if typing.TYPE_CHECKING:
+    import metaflow.runner.deployer
+    import metaflow.plugins.aws.step_functions.step_functions_deployer_objects
+
+from .step_functions import StepFunctions as StepFunctions
+from ....runner.deployer import DeployedFlow as DeployedFlow
+from ....runner.deployer import TriggeredRun as TriggeredRun
+from ....runner.utils import get_lower_level_group as get_lower_level_group
+from ....runner.utils import handle_timeout as handle_timeout
+
+class StepFunctionsTriggeredRun(metaflow.runner.deployer.TriggeredRun, metaclass=type):
+    """
+    A class representing a triggered AWS Step Functions state machine execution.
+    """
+    def terminate(self, **kwargs) -> bool:
+        """
+        Terminate the running state machine execution.
+        
+        Parameters
+        ----------
+        authorize : str, optional, default None
+            Authorize the termination with a production token.
+        
+        Returns
+        -------
+        bool
+            True if the command was successful, False otherwise.
+        """
+        ...
+    ...
+
+class StepFunctionsDeployedFlow(metaflow.runner.deployer.DeployedFlow, metaclass=metaflow.runner.deployer.DeployedFlowMeta):
+    """
+    A class representing a deployed AWS Step Functions state machine.
+    """
+    @classmethod
+    def from_deployment(cls, identifier: str, metadata: typing.Optional[str] = None):
+        """
+        This method is not currently implemented for Step Functions.
+        
+        Raises
+        ------
+        NotImplementedError
+            This method is not implemented for Step Functions.
+        """
+        ...
+    @property
+    def production_token(self: metaflow.runner.deployer.DeployedFlow) -> typing.Optional[str]:
+        """
+        Get the production token for the deployed flow.
+        
+        Returns
+        -------
+        str, optional
+            The production token, None if it cannot be retrieved.
+        """
+        ...
+    def list_runs(self, states: typing.Optional[typing.List[str]] = None) -> typing.List[StepFunctionsTriggeredRun]:
+        """
+        List runs of the deployed flow.
+        
+        Parameters
+        ----------
+        states : List[str], optional, default None
+            A list of states to filter the runs by. Allowed values are:
+            RUNNING, SUCCEEDED, FAILED, TIMED_OUT, ABORTED.
+            If not provided, all states will be considered.
+        
+        Returns
+        -------
+        List[StepFunctionsTriggeredRun]
+            A list of TriggeredRun objects representing the runs of the deployed flow.
+        
+        Raises
+        ------
+        ValueError
+            If any of the provided states are invalid or if there are duplicate states.
+        """
+        ...
+    def delete(self, **kwargs) -> bool:
+        """
+        Delete the deployed state machine.
+        
+        Parameters
+        ----------
+        authorize : str, optional, default None
+            Authorize the deletion with a production token.
+        
+        Returns
+        -------
+        bool
+            True if the command was successful, False otherwise.
+        """
+        ...
+    def trigger(self, **kwargs) -> StepFunctionsTriggeredRun:
+        """
+        Trigger a new run for the deployed flow.
+        
+        Parameters
+        ----------
+        **kwargs : Any
+            Additional arguments to pass to the trigger command,
+            `Parameters` in particular
+        
+        Returns
+        -------
+        StepFunctionsTriggeredRun
+            The triggered run instance.
+        
+        Raises
+        ------
+        Exception
+            If there is an error during the trigger process.
+        """
+        ...
+    ...
+

metaflow-stubs/plugins/azure/__init__.pyi

@@ -1,12 +1,17 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.28                                                            #
-# Generated on 2024-11-01T10:21:04.484043                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.29                                                                                #
+# Generated on 2024-11-07T22:19:34.589772                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
 
-def create_azure_credential():
-    ...
+from . import azure_credential as azure_credential
+from .azure_credential import create_cacheable_azure_credential as create_azure_credential
+from . import azure_exceptions as azure_exceptions
+from . import azure_utils as azure_utils
+from . import blob_service_client_factory as blob_service_client_factory
+from . import includefile_support as includefile_support
+from . import azure_secret_manager_secrets_provider as azure_secret_manager_secrets_provider
 

metaflow-stubs/plugins/azure/azure_credential.pyi

@@ -1,12 +1,13 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.28                                                            #
-# Generated on 2024-11-01T10:21:04.542418                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.29                                                                                #
+# Generated on 2024-11-07T22:19:34.630980                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
 
+
 class AzureDefaultClientProvider(object, metaclass=type):
     @staticmethod
     def create_cacheable_azure_credential(*args, **kwargs):

metaflow-stubs/plugins/azure/azure_exceptions.pyi

@@ -1,21 +1,17 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.28                                                            #
-# Generated on 2024-11-01T10:21:04.542673                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.29                                                                                #
+# Generated on 2024-11-07T22:19:34.631160                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
+import metaflow
 import typing
 if typing.TYPE_CHECKING:
     import metaflow.exception
 
-class MetaflowException(Exception, metaclass=type):
-    def __init__(self, msg = "", lineno = None):
-        ...
-    def __str__(self):
-        ...
-    ...
+from ...exception import MetaflowException as MetaflowException
 
 class MetaflowAzureAuthenticationError(metaflow.exception.MetaflowException, metaclass=type):
     ...

metaflow-stubs/plugins/azure/azure_secret_manager_secrets_provider.pyi

@@ -1,38 +1,25 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.28                                                            #
-# Generated on 2024-11-01T10:21:04.544077                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.29                                                                                #
+# Generated on 2024-11-07T22:19:34.632773                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
+import abc
+import metaflow
 import typing
 if typing.TYPE_CHECKING:
-    import metaflow.plugins.secrets
     import abc
     import metaflow.exception
+    import metaflow.plugins.secrets
 
-class SecretsProvider(abc.ABC, metaclass=abc.ABCMeta):
-    def get_secret_as_dict(self, secret_id, options = {}, role = None) -> typing.Dict[str, str]:
-        """
-        Retrieve the secret from secrets backend, and return a dictionary of
-        environment variables.
-        """
-        ...
-    ...
-
-class MetaflowException(Exception, metaclass=type):
-    def __init__(self, msg = "", lineno = None):
-        ...
-    def __str__(self):
-        ...
-    ...
+from ..secrets import SecretsProvider as SecretsProvider
+from ...exception import MetaflowException as MetaflowException
+from .azure_credential import create_cacheable_azure_credential as create_cacheable_azure_credential
 
 AZURE_KEY_VAULT_PREFIX: None
 
-def create_cacheable_azure_credential():
-    ...
-
 class MetaflowAzureKeyVaultBadVault(metaflow.exception.MetaflowException, metaclass=type):
     """
     Raised when the secretid is fully qualified but does not have the right key vault domain

metaflow-stubs/plugins/azure/azure_utils.pyi

@@ -1,36 +1,18 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.28                                                            #
-# Generated on 2024-11-01T10:21:04.543152                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.29                                                                                #
+# Generated on 2024-11-07T22:19:34.631563                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
-import typing
-if typing.TYPE_CHECKING:
-    import metaflow.exception
 
-class MetaflowAzureAuthenticationError(metaflow.exception.MetaflowException, metaclass=type):
-    ...
-
-class MetaflowAzureResourceError(metaflow.exception.MetaflowException, metaclass=type):
-    ...
-
-class MetaflowAzurePackageError(metaflow.exception.MetaflowException, metaclass=type):
-    ...
-
-class MetaflowInternalError(metaflow.exception.MetaflowException, metaclass=type):
-    ...
-
-class MetaflowException(Exception, metaclass=type):
-    def __init__(self, msg = "", lineno = None):
-        ...
-    def __str__(self):
-        ...
-    ...
-
-def create_cacheable_azure_credential():
-    ...
+from .azure_exceptions import MetaflowAzureAuthenticationError as MetaflowAzureAuthenticationError
+from .azure_exceptions import MetaflowAzureResourceError as MetaflowAzureResourceError
+from .azure_exceptions import MetaflowAzurePackageError as MetaflowAzurePackageError
+from ...exception import MetaflowInternalError as MetaflowInternalError
+from ...exception import MetaflowException as MetaflowException
+from .azure_credential import create_cacheable_azure_credential as create_cacheable_azure_credential
 
 def check_azure_deps(func):
     """

metaflow-stubs/plugins/azure/blob_service_client_factory.pyi

@@ -1,33 +1,18 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.28                                                            #
-# Generated on 2024-11-01T10:21:04.543546                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.29                                                                                #
+# Generated on 2024-11-07T22:19:34.631890                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
 
-class MetaflowException(Exception, metaclass=type):
-    def __init__(self, msg = "", lineno = None):
-        ...
-    def __str__(self):
-        ...
-    ...
+from ...exception import MetaflowException as MetaflowException
+from .azure_utils import check_azure_deps as check_azure_deps
+from .azure_credential import create_cacheable_azure_credential as create_cacheable_azure_credential
 
 AZURE_STORAGE_BLOB_SERVICE_ENDPOINT: None
 
-def check_azure_deps(func):
-    """
-    The decorated function checks Azure dependencies (as needed for Azure storage backend). This includes
-    various Azure SDK packages, as well as a Python version of >3.6
-    
-    We also tune some warning and logging configurations to reduce excessive log lines from Azure SDK.
-    """
-    ...
-
-def create_cacheable_azure_credential():
-    ...
-
 AZURE_CLIENT_CONNECTION_DATA_BLOCK_SIZE: int
 
 AZURE_CLIENT_MAX_SINGLE_GET_SIZE_MB: int

metaflow-stubs/plugins/azure/includefile_support.pyi

@@ -1,24 +1,14 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.28                                                            #
-# Generated on 2024-11-01T10:21:04.476293                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.29                                                                                #
+# Generated on 2024-11-07T22:19:34.632415                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
-import typing
-if typing.TYPE_CHECKING:
-    import metaflow.exception
 
-class MetaflowException(Exception, metaclass=type):
-    def __init__(self, msg = "", lineno = None):
-        ...
-    def __str__(self):
-        ...
-    ...
-
-class MetaflowInternalError(metaflow.exception.MetaflowException, metaclass=type):
-    ...
+from ...exception import MetaflowException as MetaflowException
+from ...exception import MetaflowInternalError as MetaflowInternalError
 
 class Azure(object, metaclass=type):
     @classmethod

metaflow-stubs/plugins/cards/__init__.pyi

@@ -1,9 +1,19 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.28                                                            #
-# Generated on 2024-11-01T10:21:04.482539                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.29                                                                                #
+# Generated on 2024-11-07T22:19:34.586684                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
 
+from . import card_modules as card_modules
+from . import exception as exception
+from . import component_serializer as component_serializer
+from . import card_creator as card_creator
+from . import card_decorator as card_decorator
+from . import card_datastore as card_datastore
+from . import card_resolver as card_resolver
+from . import card_client as card_client
+from . import card_cli as card_cli
+