Merge pull request #253 from BertJorissen/develop

Fix SlotPickleMixin import error

abipy/core/kpoints.py

@@ -13,10 +13,11 @@
 from monty.functools import lazy_property
 from monty.string import marquee
 from pymatgen.core.lattice import Lattice
-from pymatgen.util.serialization import pmg_serialize, SlotPickleMixin
+from pymatgen.util.serialization import pmg_serialize
 from abipy.iotools import ETSF_Reader
 from abipy.tools.derivatives import finite_diff
 from abipy.tools.numtools import add_periodic_replicas, is_diagonal
+from abipy.core.mixins import SlotPickleMixin
 
 import logging
 logger = logging.getLogger(__name__)

abipy/core/mixins.py

@@ -25,6 +25,7 @@
     "Has_PhononBands",
     "NotebookWriter",
     "Has_Header",
+    "SlotPickleMixin"
 ]
 
 
@@ -949,3 +950,17 @@ def hdr(self):
         return self.reader.read_abinit_hdr()
 
     #def compare_hdr(self, other_hdr):
+
+
+class SlotPickleMixin:
+    """
+    This mixin makes it possible to pickle/unpickle objects with __slots__
+    defined.
+    """
+
+    def __getstate__(self):
+        return {slot: getattr(self, slot) for slot in self.__slots__ if hasattr(self, slot)}
+
+    def __setstate__(self, state):
+        for slot, value in state.items():
+            setattr(self, slot, value)

abipy/core/symmetries.py

@@ -13,8 +13,8 @@
 from monty.termcolor import cprint
 from monty.collections import dict2namedtuple
 from pymatgen.symmetry.analyzer import SpacegroupAnalyzer
-from pymatgen.util.serialization import SlotPickleMixin
 from abipy.core.kpoints import wrap_to_ws, issamek, has_timrev_from_kptopt
+from abipy.core.mixins import SlotPickleMixin
 
 
 __all__ = [