Remove dynesty from __getstate__

Remove dynesty from getstate
ad2f5497 · Colm Talbot · Gregory Ashton · 206f412b · ad2f5497
Commit ad2f5497 authored Jun 11, 2019 by Colm Talbot Committed by Gregory Ashton Jun 11, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

bilby/core/sampler/dynesty.py bilby/core/sampler/dynesty.py +6 -0

No files found.
--- 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',