nextpipe 0.6.1__tar.gz → 0.6.2.dev0__tar.gz

{nextpipe-0.6.1 → nextpipe-0.6.2.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nextpipe
-Version: 0.6.1
+Version: 0.6.2.dev0
 Summary: Framework for Decision Pipeline modeling and execution
 Project-URL: Homepage, https://www.nextmv.io
 Project-URL: Documentation, https://nextpipe.docs.nextmv.io/en/latest/

nextpipe-0.6.2.dev0/nextpipe/__about__.py

@@ -0,0 +1 @@
+__version__ = "v0.6.2-dev.0"

{nextpipe-0.6.1 → nextpipe-0.6.2.dev0}/nextpipe/decorators.py

@@ -781,6 +781,7 @@ class App:
         full_result: bool = False,
         run_configuration: nextmv.RunConfiguration = None,
         polling_options: nextmv.PollingOptions | None = nextmv.DEFAULT_POLLING_OPTIONS,
+        bypass_result: typing.Any = None,
     ):
         """
         Initialize an App object.
@@ -799,6 +800,9 @@ class App:
             The configuration to apply when running the app, by default None.
         polling_options : Optional[nextmv.PollingOptions], optional
             Options for polling for the results of the app run, by default _DEFAULT_POLLING_OPTIONS.
+        bypass_result : typing.Any, optional
+            A value that can be provided to bypass the actual execution of the app.
+            This is mainly used for testing purposes, by default None.
         """
 
         if input_type:
@@ -824,6 +828,7 @@ class App:
         self.full_result = full_result
         self.run_configuration = run_configuration
         self.polling_options = polling_options
+        self.bypass_result = bypass_result
 
     def __repr__(self):
         """Return a string representation of the app.
@@ -846,6 +851,7 @@ def app(
     full_result: bool = False,
     run_configuration: nextmv.RunConfiguration = None,
     polling_options: nextmv.PollingOptions | None = nextmv.DEFAULT_POLLING_OPTIONS,
+    bypass_result: typing.Any = None,
 ):
     """
     Decorator to mark a step as a Nextmv Application (external application)
@@ -885,6 +891,9 @@ def app(
         class itself. Please note that the `.initial_delay` attribute will be
         overridden internally, as a strategy to stagger multiple parallel runs
         and avoid overloading the Platform.
+    bypass_result : typing.Any, optional
+        A value that can be provided to bypass the actual execution of the app.
+        This is mainly used for testing purposes, by default None.
 
     Example
     -------
@@ -957,6 +966,7 @@ def app(
             full_result=full_result,
             run_configuration=run_configuration,
             polling_options=polling_options,
+            bypass_result=bypass_result,
         )
         wrapper.step.type = StepType.APP
 

{nextpipe-0.6.1 → nextpipe-0.6.2.dev0}/nextpipe/flow.py

@@ -29,6 +29,7 @@ import inspect
 import io
 import os
 import random
+import shutil
 import tempfile
 import threading
 import time
@@ -991,14 +992,24 @@ class Runner:
 
             # Run the application
             try:
-                run_id = app.new_run(*run_args[0], **run_args[1])
-                console_url = f"{client.console_url}/app/{app_step.app_id}/run/{run_id}?view=details"
-                utils.log_internal(f"Started app step {node.id} run, find it at {console_url}")
-                node.run_id = run_id
-                update_dag()  # Update the node with the run_id before we start polling for results
-                result = app.run_result_with_polling(
-                    run_id=run_id, polling_options=polling_options, output_dir_path=temp_dir
-                )
+                if app_step.bypass_result is None:
+                    run_id = app.new_run(*run_args[0], **run_args[1])
+                    console_url = f"{client.console_url}/app/{app_step.app_id}/run/{run_id}?view=details"
+                    utils.log_internal(f"Started app step {node.id} run, find it at {console_url}")
+                    node.run_id = run_id
+                    update_dag()  # Update the node with the run_id before we start polling for results
+                    result = app.run_result_with_polling(
+                        run_id=run_id, polling_options=polling_options, output_dir_path=temp_dir
+                    )
+                else:
+                    utils.log_internal(f"Bypassing app step {node.id} run, using bypass result")
+                    # If the user supplied a bypass result, we need to write it to the temp dir if it's in dir mode
+                    if isinstance(app_step.bypass_result, str) and os.path.isdir(app_step.bypass_result):
+                        # If the bypass result is a directory, we copy its contents to the temp dir
+                        for filename in os.listdir(app_step.bypass_result):
+                            shutil.copy(os.path.join(app_step.bypass_result, filename), temp_dir)
+                    else:
+                        result = app_step.bypass_result
             finally:  # Make sure we clean up temp dir on failure too
                 # If the temp dir is empty, remove it
                 dir_result = False

nextpipe-0.6.1/nextpipe/__about__.py

@@ -1 +0,0 @@
-__version__ = "v0.6.1"