From 438f76dc43d2dc7e6fb844f64395b1c4d1e17b27 Mon Sep 17 00:00:00 2001
From: Cristian C Lalescu <Cristian.Lalescu@ds.mpg.de>
Date: Sun, 3 Mar 2019 11:08:35 +0100
Subject: [PATCH] add more general checkpoint file name method

---
 bfps/DNS.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/bfps/DNS.py b/bfps/DNS.py
index fdaa0f63..9dd9f182 100644
--- a/bfps/DNS.py
+++ b/bfps/DNS.py
@@ -781,6 +781,10 @@ class DNS(_code):
         return os.path.join(
                     self.work_dir,
                     self.simname + '_checkpoint_0.h5')
+    def get_checkpoint_fname(self, checkpoint = 0):
+        return os.path.join(
+                    self.work_dir,
+                    self.simname + '_checkpoint_{0}.h5'.format(checkpoint))
     def generate_tracer_state(
             self,
             rseed = None,
-- 
GitLab