fixed bug in writing out data

UDST · May 29, 2014 · d460770 · d460770
1 parent 063e8e4
commit d460770
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/urbansim/utils/dataset.py b/urbansim/utils/dataset.py
@@ -33,13 +33,19 @@ def list_tbls(self):
     def save_tmptbl(self, name, tbl):
         self.d[name] = tbl
 
-    def save_output(self, filename):
-        outstore = pd.HDFStore(filename, savetbls)
+    def save_output(self, filename, savetbls=None, debug=False):
+
+        if savetbls is None:
+            savetbls = self.d.keys()
+
+        outstore = pd.HDFStore(filename, "w")
 
         for key in savetbls:
             df = self.fetch(key)
             df = misc.df64bitto32bit(df)
-            print key + "\n", df.describe()
+            print key
+            if debug:
+                print df.describe()
             outstore[key] = df
 
         outstore.close()