From ad2f54974636d54e6bc98d1b8fcf40efac46cf81 Mon Sep 17 00:00:00 2001
From: Colm Talbot <colm.talbot@ligo.org>
Date: Tue, 11 Jun 2019 21:45:10 -0500
Subject: [PATCH] Remove dynesty from __getstate__
---
bilby/core/sampler/dynesty.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/bilby/core/sampler/dynesty.py b/bilby/core/sampler/dynesty.py
index ddd59ff08..f6c40ebf1 100644
--- a/bilby/core/sampler/dynesty.py
+++ b/bilby/core/sampler/dynesty.py
@@ -121,6 +121,12 @@ class Dynesty(NestedSampler):
signal.signal(signal.SIGINT, self.write_current_state_and_exit)
signal.signal(signal.SIGALRM, self.write_current_state_and_exit)
+ def __getstate__(self):
+ """ For pickle: remove external_sampler, which can be an unpicklable "module" """
+ state = self.__dict__.copy()
+ del state['external_sampler']
+ return state
+
@property
def sampler_function_kwargs(self):
keys = ['dlogz', 'print_progress', 'print_func', 'maxiter',
--
GitLab