From 3045f1d3cbf5fad3fa44c32b3f890d33c9a783cb Mon Sep 17 00:00:00 2001 From: Steven Ray Anaya Date: Fri, 13 May 2022 14:45:44 -0600 Subject: [PATCH] Add pickle method to base class --- python/lammps/mliap/mliap_unified_abc.py | 5 +++++ 1 file changed, 5 insertions(+) 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)