ob-metaflow-stubs 6.0.3.117__py2.py3-none-any.whl → 6.0.3.118__py2.py3-none-any.whl

metaflow-stubs/runner/nbrun.pyi

@@ -1,153 +1,16 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.27.1+obcheckpoint(0.1.2);ob(v1)                               #
-# Generated on 2024-11-13T19:05:29.366131                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-11-13T23:34:52.844920                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
 import typing
-if typing.TYPE_CHECKING:
-    import metaflow.runner.metaflow_runner
 
-class Runner(object, metaclass=type):
-    """
-    Metaflow's Runner API that presents a programmatic interface
-    to run flows and perform other operations either synchronously or asynchronously.
-    The class expects a path to the flow file along with optional arguments
-    that match top-level options on the command-line.
-    
-    This class works as a context manager, calling `cleanup()` to remove
-    temporary files at exit.
-    
-    Example:
-    ```python
-    with Runner('slowflow.py', pylint=False) as runner:
-        result = runner.run(alpha=5, tags=["abc", "def"], max_workers=5)
-        print(result.run.finished)
-    ```
-    
-    Parameters
-    ----------
-    flow_file : str
-        Path to the flow file to run
-    show_output : bool, default True
-        Show the 'stdout' and 'stderr' to the console by default,
-        Only applicable for synchronous 'run' and 'resume' functions.
-    profile : Optional[str], default None
-        Metaflow profile to use to run this run. If not specified, the default
-        profile is used (or the one already set using `METAFLOW_PROFILE`)
-    env : Optional[Dict], default None
-        Additional environment variables to set for the Run. This overrides the
-        environment set for this process.
-    cwd : Optional[str], default None
-        The directory to run the subprocess in; if not specified, the current
-        directory is used.
-    file_read_timeout : int, default 3600
-        The timeout until which we try to read the runner attribute file.
-    **kwargs : Any
-        Additional arguments that you would pass to `python myflow.py` before
-        the `run` command.
-    """
-    def __init__(self, flow_file: str, show_output: bool = True, profile: typing.Optional[str] = None, env: typing.Optional[typing.Dict] = None, cwd: typing.Optional[str] = None, file_read_timeout: int = 3600, **kwargs):
-        ...
-    def __enter__(self) -> metaflow.runner.metaflow_runner.Runner:
-        ...
-    def __aenter__(self) -> metaflow.runner.metaflow_runner.Runner:
-        ...
-    def _Runner__get_executing_run(self, tfp_runner_attribute, command_obj):
-        ...
-    def run(self, **kwargs) -> metaflow.runner.metaflow_runner.ExecutingRun:
-        """
-        Blocking execution of the run. This method will wait until
-        the run has completed execution.
-        
-        Parameters
-        ----------
-        **kwargs : Any
-            Additional arguments that you would pass to `python myflow.py` after
-            the `run` command, in particular, any parameters accepted by the flow.
-        
-        Returns
-        -------
-        ExecutingRun
-            ExecutingRun containing the results of the run.
-        """
-        ...
-    def resume(self, **kwargs):
-        """
-        Blocking resume execution of the run.
-        This method will wait until the resumed run has completed execution.
-        
-        Parameters
-        ----------
-        **kwargs : Any
-            Additional arguments that you would pass to `python ./myflow.py` after
-            the `resume` command.
-        
-        Returns
-        -------
-        ExecutingRun
-            ExecutingRun containing the results of the resumed run.
-        """
-        ...
-    def async_run(self, **kwargs) -> metaflow.runner.metaflow_runner.ExecutingRun:
-        """
-        Non-blocking execution of the run. This method will return as soon as the
-        run has launched.
-        
-        Note that this method is asynchronous and needs to be `await`ed.
-        
-        Parameters
-        ----------
-        **kwargs : Any
-            Additional arguments that you would pass to `python myflow.py` after
-            the `run` command, in particular, any parameters accepted by the flow.
-        
-        Returns
-        -------
-        ExecutingRun
-            ExecutingRun representing the run that was started.
-        """
-        ...
-    def async_resume(self, **kwargs):
-        """
-        Non-blocking resume execution of the run.
-        This method will return as soon as the resume has launched.
-        
-        Note that this method is asynchronous and needs to be `await`ed.
-        
-        Parameters
-        ----------
-        **kwargs : Any
-            Additional arguments that you would pass to `python myflow.py` after
-            the `resume` command.
-        
-        Returns
-        -------
-        ExecutingRun
-            ExecutingRun representing the resumed run that was started.
-        """
-        ...
-    def __exit__(self, exc_type, exc_value, traceback):
-        ...
-    def __aexit__(self, exc_type, exc_value, traceback):
-        ...
-    def cleanup(self):
-        """
-        Delete any temporary files created during execution.
-        """
-        ...
-    ...
-
-def get_current_cell(ipython):
-    ...
-
-def format_flowfile(cell):
-    """
-    Formats the given cell content to create a valid Python script that can be executed as a Metaflow flow.
-    """
-    ...
+from .metaflow_runner import Runner as Runner
+from .utils import get_current_cell as get_current_cell
+from .utils import format_flowfile as format_flowfile
 
 class NBRunnerInitializationError(Exception, metaclass=type):
     """
@@ -177,13 +40,13 @@ class NBRunner(object, metaclass=type):
     show_output : bool, default True
         Show the 'stdout' and 'stderr' to the console by default,
         Only applicable for synchronous 'run' and 'resume' functions.
-    profile : Optional[str], default None
+    profile : str, optional, default None
         Metaflow profile to use to run this run. If not specified, the default
         profile is used (or the one already set using `METAFLOW_PROFILE`)
-    env : Optional[Dict], default None
+    env : Dict[str, str], optional, default None
         Additional environment variables to set for the Run. This overrides the
         environment set for this process.
-    base_dir : Optional[str], default None
+    base_dir : str, optional, default None
         The directory to run the subprocess in; if not specified, the current
         working directory is used.
     file_read_timeout : int, default 3600

metaflow-stubs/runner/subprocess_manager.pyi

@@ -1,14 +1,13 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.27.1+obcheckpoint(0.1.2);ob(v1)                               #
-# Generated on 2024-11-13T19:05:29.382678                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-11-13T23:34:52.857794                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
 import typing
-if typing.TYPE_CHECKING:
-    import metaflow.runner.subprocess_manager
+
 
 def kill_process_and_descendants(pid, termination_timeout):
     ...
@@ -26,7 +25,7 @@ class SubprocessManager(object, metaclass=type):
     """
     def __init__(self):
         ...
-    def __aenter__(self) -> SubprocessManager:
+    def __aenter__(self) -> "SubprocessManager":
         ...
     def __aexit__(self, exc_type, exc_value, traceback):
         ...
@@ -120,7 +119,7 @@ class CommandManager(object, metaclass=type):
             directory is used.
         """
         ...
-    def __aenter__(self) -> CommandManager:
+    def __aenter__(self) -> "CommandManager":
         ...
     def __aexit__(self, exc_type, exc_value, traceback):
         ...
@@ -195,7 +194,7 @@ class CommandManager(object, metaclass=type):
             for example.
         """
         ...
-    def emit_logs(self, stream: str = "stdout", custom_logger: typing.Callable[..., None] = print):
+    def emit_logs(self, stream: str = 'stdout', custom_logger: typing.Callable[..., None] = print):
         """
         Helper function that can easily emit all the logs for a given stream.
         

metaflow-stubs/runner/utils.pyi

@@ -1,11 +1,18 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.27.1+obcheckpoint(0.1.2);ob(v1)                               #
-# Generated on 2024-11-13T19:05:29.383059                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-11-13T23:34:52.858306                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
+import typing
+if typing.TYPE_CHECKING:
+    import metaflow.runner.click_api
+    import metaflow.runner.subprocess_manager
+    import typing
+    import tempfile
+
 
 TYPE_CHECKING: bool
 
@@ -14,17 +21,18 @@ def get_current_cell(ipython):
 
 def format_flowfile(cell):
     """
-    Formats the given cell content to create a valid Python script that can be executed as a Metaflow flow.
+    Formats the given cell content to create a valid Python script that can be
+    executed as a Metaflow flow.
     """
     ...
 
-def check_process_status(command_obj: "CommandManager"):
+def check_process_status(command_obj: "metaflow.runner.subprocess_manager.CommandManager"):
     ...
 
-def read_from_file_when_ready(file_path: str, command_obj: "CommandManager", timeout: float = 5):
+def read_from_file_when_ready(file_path: str, command_obj: "metaflow.runner.subprocess_manager.CommandManager", timeout: float = 5):
     ...
 
-def handle_timeout(tfp_runner_attribute, command_obj: "CommandManager", file_read_timeout: int):
+def handle_timeout(tfp_runner_attribute: "tempfile._TemporaryFileWrapper[str]", command_obj: "metaflow.runner.subprocess_manager.CommandManager", file_read_timeout: int):
     """
     Handle the timeout for a running subprocess command that reads a file
     and raises an error with appropriate logs if a TimeoutError occurs.
@@ -51,3 +59,30 @@ def handle_timeout(tfp_runner_attribute, command_obj: "CommandManager", file_rea
     """
     ...
 
+def get_lower_level_group(api: "metaflow.runner.click_api.MetaflowAPI", top_level_kwargs: typing.Dict[str, typing.Any], sub_command: str, sub_command_kwargs: typing.Dict[str, typing.Any]) -> "metaflow.runner.click_api.MetaflowAPI":
+    """
+    Retrieve a lower-level group from the API based on the type and provided arguments.
+    
+    Parameters
+    ----------
+    api : MetaflowAPI
+        Metaflow API instance.
+    top_level_kwargs : Dict[str, Any]
+        Top-level keyword arguments to pass to the API.
+    sub_command : str
+        Sub-command of API to get the API for
+    sub_command_kwargs : Dict[str, Any]
+        Sub-command arguments
+    
+    Returns
+    -------
+    MetaflowAPI
+        The lower-level group object retrieved from the API.
+    
+    Raises
+    ------
+    ValueError
+        If the `_type` is None.
+    """
+    ...
+

metaflow-stubs/system/__init__.pyi

@@ -1,112 +1,14 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.27.1+obcheckpoint(0.1.2);ob(v1)                               #
-# Generated on 2024-11-13T19:05:29.368490                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-11-13T23:34:52.847493                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
-import typing
-if typing.TYPE_CHECKING:
-    import metaflow.event_logger
-    import metaflow.monitor
-    import typing
 
-class SystemMonitor(object, metaclass=type):
-    def __init__(self):
-        ...
-    def __del__(self):
-        ...
-    def update_context(self, context: typing.Dict[str, typing.Any]):
-        """
-        Update the global context maintained by the system monitor.
-        
-        Parameters
-        ----------
-        context : Dict[str, Any]
-            A dictionary containing the context to update.
-        """
-        ...
-    def init_system_monitor(self, flow_name: str, monitor: "metaflow.monitor.NullMonitor"):
-        ...
-    @property
-    def monitor(self) -> typing.Optional["metaflow.monitor.NullMonitor"]:
-        ...
-    def measure(self, name: str):
-        """
-        Context manager to measure the execution duration and counter of a block of code.
-        
-        Parameters
-        ----------
-        name : str
-            The name to associate with the timer and counter.
-        
-        Yields
-        ------
-        None
-        """
-        ...
-    def count(self, name: str):
-        """
-        Context manager to increment a counter.
-        
-        Parameters
-        ----------
-        name : str
-            The name of the counter.
-        
-        Yields
-        ------
-        None
-        """
-        ...
-    def gauge(self, gauge: "metaflow.monitor.Gauge"):
-        """
-        Log a gauge.
-        
-        Parameters
-        ----------
-        gauge : metaflow.monitor.Gauge
-            The gauge to log.
-        """
-        ...
-    ...
-
-class SystemLogger(object, metaclass=type):
-    def __init__(self):
-        ...
-    def __del__(self):
-        ...
-    def update_context(self, context: typing.Dict[str, typing.Any]):
-        """
-        Update the global context maintained by the system logger.
-        
-        Parameters
-        ----------
-        context : Dict[str, Any]
-            A dictionary containing the context to update.
-        """
-        ...
-    def init_system_logger(self, flow_name: str, logger: "metaflow.event_logger.NullEventLogger"):
-        ...
-    @property
-    def logger(self) -> typing.Optional["metaflow.event_logger.NullEventLogger"]:
-        ...
-    def log_event(self, level: str, module: str, name: str, payload: typing.Optional[typing.Any] = None):
-        """
-        Log an event to the event logger.
-        
-        Parameters
-        ----------
-        level : str
-            Log level of the event. Can be one of "info", "warning", "error", "critical", "debug".
-        module : str
-            Module of the event. Usually the name of the class, function, or module that the event is being logged from.
-        name : str
-            Name of the event. Used to qualify the event type.
-        payload : Optional[Any], default None
-            Payload of the event. Contains the event data.
-        """
-        ...
-    ...
+from . import system_monitor as system_monitor
+from .system_monitor import SystemMonitor as SystemMonitor
+from . import system_logger as system_logger
+from .system_logger import SystemLogger as SystemLogger
 

metaflow-stubs/system/system_logger.pyi

@@ -1,8 +1,8 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.27.1+obcheckpoint(0.1.2);ob(v1)                               #
-# Generated on 2024-11-13T19:05:29.393582                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-11-13T23:34:52.871265                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
@@ -11,21 +11,12 @@ if typing.TYPE_CHECKING:
     import metaflow.event_logger
     import typing
 
+
 class SystemLogger(object, metaclass=type):
     def __init__(self):
         ...
     def __del__(self):
         ...
-    def update_context(self, context: typing.Dict[str, typing.Any]):
-        """
-        Update the global context maintained by the system logger.
-        
-        Parameters
-        ----------
-        context : Dict[str, Any]
-            A dictionary containing the context to update.
-        """
-        ...
     def init_system_logger(self, flow_name: str, logger: "metaflow.event_logger.NullEventLogger"):
         ...
     @property

metaflow-stubs/system/system_monitor.pyi

@@ -1,31 +1,21 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.27.1+obcheckpoint(0.1.2);ob(v1)                               #
-# Generated on 2024-11-13T19:05:29.393203                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-11-13T23:34:52.870931                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
 import typing
 if typing.TYPE_CHECKING:
     import metaflow.monitor
-    import typing
+
 
 class SystemMonitor(object, metaclass=type):
     def __init__(self):
         ...
     def __del__(self):
         ...
-    def update_context(self, context: typing.Dict[str, typing.Any]):
-        """
-        Update the global context maintained by the system monitor.
-        
-        Parameters
-        ----------
-        context : Dict[str, Any]
-            A dictionary containing the context to update.
-        """
-        ...
     def init_system_monitor(self, flow_name: str, monitor: "metaflow.monitor.NullMonitor"):
         ...
     @property

metaflow-stubs/tagging_util.pyi

@@ -1,17 +1,13 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.27.1+obcheckpoint(0.1.2);ob(v1)                               #
-# Generated on 2024-11-13T19:05:29.345866                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-11-13T23:34:52.834220                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
-import typing
-if typing.TYPE_CHECKING:
-    import metaflow.exception
 
-class MetaflowTaggingError(metaflow.exception.MetaflowException, metaclass=type):
-    ...
+from .exception import MetaflowTaggingError as MetaflowTaggingError
 
 def is_utf8_encodable(x):
     """

metaflow-stubs/tuple_util.pyi

@@ -1,12 +1,13 @@
-##################################################################################
-#                       Auto-generated Metaflow stub file                        #
-# MF version: 2.12.27.1+obcheckpoint(0.1.2);ob(v1)                               #
-# Generated on 2024-11-13T19:05:29.352127                                        #
-##################################################################################
+######################################################################################################
+#                                 Auto-generated Metaflow stub file                                  #
+# MF version: 2.12.30.1+obcheckpoint(0.1.4);ob(v1)                                                   #
+# Generated on 2024-11-13T23:34:52.835718                                                            #
+######################################################################################################
 
 from __future__ import annotations
 
 
+
 def namedtuple_with_defaults(typename, field_descr, defaults = ()):
     ...
 

{ob_metaflow_stubs-6.0.3.117.dist-info → ob_metaflow_stubs-6.0.3.118.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ob-metaflow-stubs
-Version: 6.0.3.117
+Version: 6.0.3.118
 Summary: Metaflow Stubs: Stubs for the metaflow package
 Author: Netflix, Outerbounds & the Metaflow Community
 Author-email: help@outerbounds.co