bug #757 move get_output_file to pipeline object

pipelet/pipeline.py

@@ -434,6 +434,19 @@ class Pipeline:
         """ 
         return self._curr_dirs[seg]
 
+    def get_output_file(self, seg):
+        """ Return the segment output file
+        
+        Parameters
+        ----------
+        seg : string, segment name.
+        
+        Returns
+        -------
+        string, segment directory. 
+        """ 
+        return path.join(self.get_curr_dir(seg),'seg_%s.output'%seg)
+
     def get_param_file(self, seg):
         """ Return the segment directory.  
 

pipelet/worker.py

@@ -150,7 +150,7 @@ class Worker(object):
         ----------
         res: task result. 
         """
-        fn = self.get_data_fn('.pipe_res')  # TODO move this to  pipe
+        fn = self.pipe.get_output_fn()  
         with closing (file(fn,'w')) as f:
             r = pickle.dump(res,f)