Multiselection included. (Without input validation).

workflows/go_workflow.py

@@ -81,7 +81,7 @@ class GODistanceFunction(workflow.Function):
         workflow.Function.__init__(self, 'go_diatance', 'GO Distances')
         self.output = None
 
-    def run(self, data):
+    def run(self, (data,)):
         logger.log('debug', 'datatype: %s' % type(data))
         if not type(data) == Annotations:
             return None
@@ -108,7 +108,7 @@ class DatasetLog(workflow.Function):
     def __init__(self):
         workflow.Function.__init__(self, 'log', 'Log')
 
-    def run(self, data):
+    def run(self, (data,)):
         logger.log('notice', 'Taking the log of dataset %s' % data.get_name())
         d = data.asarray()
         d = log(d)
@@ -151,7 +151,8 @@ class DatasetSaveFunction(workflow.Function):
         if not data:
             logger.log("notice", "No data to save.")
             return
-
+        else:
+            data = data[0]
         chooser = gtk.FileChooserDialog(title="Save pickled data...", parent=None,
                                         action=gtk.FILE_CHOOSER_ACTION_SAVE,
                                         buttons=(gtk.STOCK_CANCEL, gtk.RESPONSE_CANCEL,
@@ -179,7 +180,7 @@ class CelFileImportFunction(workflow.Function):
     def __init__(self):
         workflow.Function.__init__(self, 'cel_import', 'Import Affy')
 
-    def run(self, data):
+    def run(self, (data,)):
         import rpy
         chooser = gtk.FileChooserDialog(title="Select cel files...", parent=None,
                                         action=gtk.FILE_CHOOSER_ACTION_OPEN,
@@ -228,7 +229,7 @@ class PCAFunction(workflow.Function):
         workflow.Function.__init__(self, 'pca', 'PCA')
         self._workflow = wf
 
-    def run(self, data):
+    def run(self, (data,)):
         import rpy
         
         dim_2, dim_1 = data.get_dim_names()