nextpipe 0.1.5.dev0__tar.gz → 0.1.5.dev2__tar.gz

{nextpipe-0.1.5.dev0 → nextpipe-0.1.5.dev2}/.github/workflows/test.yml

@@ -5,7 +5,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: ["3.9", "3.11", "3.13"]
+        python-version: ["3.9", "3.13"]
     steps:
       - name: git clone
         uses: actions/checkout@v4

{nextpipe-0.1.5.dev0 → nextpipe-0.1.5.dev2}/PKG-INFO

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

nextpipe-0.1.5.dev2/examples/pipeline-chain/requirements.txt

@@ -0,0 +1 @@
+nextpipe==0.1.5.dev.2

nextpipe-0.1.5.dev2/examples/pipeline-complex/requirements.txt

@@ -0,0 +1 @@
+nextpipe==0.1.5.dev.2

nextpipe-0.1.5.dev2/examples/pipeline-ensemble/requirements.txt

@@ -0,0 +1 @@
+nextpipe==0.1.5.dev.2

nextpipe-0.1.5.dev2/examples/pipeline-foreach/requirements.txt

@@ -0,0 +1 @@
+nextpipe==0.1.5.dev.2

nextpipe-0.1.5.dev2/examples/pipeline-preprocess/requirements.txt

@@ -0,0 +1 @@
+nextpipe==0.1.5.dev.2

nextpipe-0.1.5.dev2/nextpipe/__about__.py

@@ -0,0 +1 @@
+__version__ = "v0.1.5.dev2"

{nextpipe-0.1.5.dev0 → nextpipe-0.1.5.dev2}/nextpipe/flow.py

@@ -69,7 +69,7 @@ class FlowSpec:
         self.client = Client() if client is None else client
         self.uplink = uplink.UplinkClient(self.client, uplink_config)
         # Create the graph
-        self.graph = FlowGraph(self.__class__)
+        self.graph = FlowGraph(self)
         # Inform platform about the graph
         self.uplink.submit_update(self.graph._to_uplink_dto())
         # Prepare for running the flow
@@ -104,9 +104,9 @@ class FlowSpec:
 
 
 class FlowGraph:
-    def __init__(self, flow_spec_type: type):
-        self.flow_spec_type = flow_spec_type
-        self.__create_graph(flow_spec_type)
+    def __init__(self, flow_spec: FlowSpec):
+        self.flow_spec = flow_spec
+        self.__create_graph(flow_spec)
         self.__debug_print()
         # Create a Mermaid diagram of the graph and log it
         mermaid = self._to_mermaid()
@@ -118,12 +118,12 @@ class FlowGraph:
     def get_step(self, definition: decorators.Step) -> FlowStep:
         return self.steps_by_definition[definition]
 
-    def __create_graph(self, flow_spec):
+    def __create_graph(self, flow_spec: FlowSpec):
         root = utils.get_ast_root(flow_spec)
 
         # Build the graph
         self.steps: list[FlowStep] = []
-        visitor = StepVisitor(self.steps, flow_spec)
+        visitor = StepVisitor(self.steps, flow_spec.__class__)
         visitor.visit(root)
 
         # Init steps for all step definitions
@@ -217,7 +217,7 @@ class FlowGraph:
         )
 
     def __debug_print(self):
-        utils.log_internal(f"Flow: {self.flow_spec_type.__name__}")
+        utils.log_internal(f"Flow: {self.flow_spec.__class__.__name__}")
         utils.log_internal(f"nextpipe: {version('nextpipe')}")
         utils.log_internal(f"nextmv: {version('nextmv')}")
         utils.log_internal("Flow graph steps:")
@@ -228,13 +228,13 @@ class FlowGraph:
 
 
 class StepVisitor(ast.NodeVisitor):
-    def __init__(self, steps: list[FlowStep], flow: FlowSpec):
+    def __init__(self, steps: list[FlowStep], flow_class: type):
         self.steps = steps
-        self.flow = flow
+        self.flow_class = flow_class
         super().__init__()
 
     def visit_FunctionDef(self, step_function):
-        func = getattr(self.flow, step_function.name)
+        func = getattr(self.flow_class, step_function.name)
         if hasattr(func, "is_step"):
             self.steps.append(FlowStep(step_function, func.step, func.__doc__))
 
@@ -307,7 +307,7 @@ class Runner:
         if len(inputs) > self.spec.config.max_step_inputs:
             raise Exception(
                 f"Step {step.definition.get_id()} has too many inputs ({len(inputs)}). "
-                + f"Maximum allowed is {self.graph.flow_spec_type.config.max_step_inputs}."
+                + f"Maximum allowed is {self.graph.flow_spec.config.max_step_inputs}."
             )
         return inputs
 

{nextpipe-0.1.5.dev0 → nextpipe-0.1.5.dev2}/nextpipe/utils.py

@@ -139,7 +139,7 @@ def __get_notebook_ast_root(obj: object) -> ast.ClassDef:
     ipython = get_ipython()
 
     # Go backwards in the history to find the cell where the object's class was defined.
-    for i in range(len(ipython.history_manager.input_hist_parsed), 0, -1):
+    for i in range(len(ipython.history_manager.input_hist_parsed) - 1, -1, -1):
         # Parse the code of the cell into an AST.
         tree = ast.parse(ipython.history_manager.input_hist_parsed[i])
 
@@ -155,8 +155,8 @@ def __get_normal_ast_root(obj: object) -> ast.ClassDef:
     """
     Find the root AST of the given object.
     """
-    module = importlib.import_module(obj.__module__)
-    class_name = obj.__name__
+    module = importlib.import_module(obj.__class__.__module__)
+    class_name = obj.__class__.__name__
     tree = ast.parse(inspect.getsource(module)).body
     root = [n for n in tree if isinstance(n, ast.ClassDef) and n.name == class_name][0]
     return root

nextpipe-0.1.5.dev0/examples/pipeline-chain/requirements.txt

@@ -1 +0,0 @@
-nextpipe==0.1.5.dev.0

nextpipe-0.1.5.dev0/examples/pipeline-complex/requirements.txt

@@ -1 +0,0 @@
-nextpipe==0.1.5.dev.0

nextpipe-0.1.5.dev0/examples/pipeline-ensemble/requirements.txt

@@ -1 +0,0 @@
-nextpipe==0.1.5.dev.0

nextpipe-0.1.5.dev0/examples/pipeline-foreach/requirements.txt

@@ -1 +0,0 @@
-nextpipe==0.1.5.dev.0

nextpipe-0.1.5.dev0/examples/pipeline-preprocess/requirements.txt

@@ -1 +0,0 @@
-nextpipe==0.1.5.dev.0

nextpipe-0.1.5.dev0/nextpipe/__about__.py

@@ -1 +0,0 @@
-__version__ = "v0.1.5.dev0"