diff --git a/python/lammps/mliap/mliap_unified_abc.py b/python/lammps/mliap/mliap_unified_abc.py index f5751b14f2..3243df85c4 100644 --- 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)