From 53c69b627156f1c4dcf9f8786a3ca0e221de3cb9 Mon Sep 17 00:00:00 2001
From: Joshua Harlow <harlowja@yahoo-inc.com>
Date: Tue, 21 May 2013 23:28:45 -0700
Subject: [PATCH] Ensure we fetch the appropriate inputs for the running task.

---
 taskflow/patterns/linear_flow.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/taskflow/patterns/linear_flow.py b/taskflow/patterns/linear_flow.py
index 3d6d330f..bcf7f960 100644
--- a/taskflow/patterns/linear_flow.py
+++ b/taskflow/patterns/linear_flow.py
@@ -29,6 +29,15 @@ class Flow(ordered_flow.Flow):
         super(Flow, self).__init__(name, tolerant, parents)
         self._tasks = []
 
+    def _fetch_task_inputs(self, task):
+        inputs = {}
+        if self.results:
+            (_last_task, last_results) = self.results[-1]
+            for k in task.requires():
+                if last_results and k in last_results:
+                    inputs[k] = last_results[k]
+        return inputs
+
     def _validate_provides(self, task):
         last_provides = set()
         last_provider = None