ICODE-ADC/lammps
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add pickle method to base class

Browse Source
This commit is contained in:
Steven Ray Anaya
2022-05-13 14:45:44 -06:00
parent f909d7760d
commit 3045f1d3cb
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
python/lammps/mliap/mliap_unified_abc.py
View File

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