Add pickle method to base class

2022-05-13 14:45:44 -06:00
parent f909d7760d
commit 3045f1d3cb
1 changed files with 5 additions and 0 deletions
--- a/python/lammps/mliap/mliap_unified_abc.py
+++ b/python/lammps/mliap/mliap_unified_abc.py
@ -1,4 +1,5 @@
 from abc import ABC, abstractmethod
+import pickle

 class MLIAPUnified(ABC):
    """Abstract base class for MLIAPUnified."""
@ -21,3 +22,7 @@ class MLIAPUnified(ABC):
    @abstractmethod
    def compute_forces(self, data):
        """Compute forces."""
+
+    def pickle(self, fname):
+        with open(fname, 'wb') as fp:
+            pickle.dump(self, fp)