]> git.parisson.com Git - timeside.git/commitdiff
Fix no output_file for yaml and numpy analyzer serializers
authorGuillaume Pellerin <yomguy@parisson.com>
Sun, 22 Dec 2013 21:38:39 +0000 (22:38 +0100)
committerGuillaume Pellerin <yomguy@parisson.com>
Sun, 22 Dec 2013 21:38:39 +0000 (22:38 +0100)
timeside/analyzer/core.py

index 1633275abf920850d06bdf0d7502f4f5e7315f6a..0da2341475165e80ccb4736d68f7b2fb3f6fe1b9 100644 (file)
@@ -894,7 +894,7 @@ class AnalyzerResultContainer(dict):
             results.add(res)
         return results
 
-    def to_yaml(self, output_file):
+    def to_yaml(self, output_file=None):
         #if data_list == None: data_list = self.results
         import yaml
 
@@ -932,8 +932,11 @@ class AnalyzerResultContainer(dict):
             results.add(res)
         return results
 
-    def to_numpy(self, output_file):
-        numpy.save(output_file, self)
+    def to_numpy(self, output_file=None):
+        if output_file:
+            numpy.save(output_file, self)
+        else:
+            return self
 
     @staticmethod
     def from_numpy(input_file):