fix #291 : excluding 0D arrays when saving a session.

alixdamman · gdementen · commit a6c60d70d35d · 2017-06-14T15:52:52.000+02:00
diff --git a/larray/io/session.py b/larray/io/session.py
@@ -4,6 +4,7 @@
 from collections import OrderedDict
 from pandas import ExcelWriter, ExcelFile, HDFStore
 
+from larray.core.abc import ABCLArray
 from larray.util.misc import pickle
 from larray.io.excel import open_excel
 from larray.io.array import df_aslarray, read_csv, read_hdf
@@ -114,6 +115,10 @@ def dump_arrays(self, key_values, *args, **kwargs):
         display = kwargs.pop('display', False)
         self._open_for_write()
         for key, value in key_values:
+            if isinstance(value, ABCLArray) and value.ndim == 0:
+                if display:
+                    print('Cannot dump {}. Dumping 0D arrays is not supported currently.'.format(key))
+                continue
             if display:
                 print("dumping", key, "...", end=' ')
             self._dump(key, value, *args, **kwargs)
