fixing #115

src/exoplanet/theano_ops/starry.py

@@ -62,6 +62,12 @@ def __init__(self):
         self.ld = driver.LimbDark()
         super().__init__()
 
+    def __getstate__(self):
+        return {}
+
+    def __setstate__(self, data):
+        self.ld = driver.LimbDark()
+
     def make_node(self, *inputs):
         in_args = [as_tensor_variable(i) for i in inputs]
         if any(i.dtype != "float64" for i in in_args):

tests/theano_ops/starry_test.py

@@ -7,13 +7,13 @@
 import theano.tensor as tt
 from theano.tests import unittest_tools as utt
 
+from exoplanet.theano_ops.driver import SimpleLimbDark
 from exoplanet.theano_ops.starry import (
     GetCl,
     GetClRev,
     LimbDark,
     RadiusFromOccArea,
 )
-from exoplanet.theano_ops.driver import SimpleLimbDark
 
 
 class TestGetCl(utt.InferShapeTester):
@@ -104,6 +104,12 @@ def test_grad(self):
         func = lambda *args: self.op(*args)[0]  # NOQA
         utt.verify_grad(func, in_args)
 
+    def test_pickle(self):
+        f, _, in_args = self.get_args()
+        data = pickle.dumps(self.op, -1)
+        new_op = pickle.loads(data)
+        utt.assert_allclose(f(*in_args), new_op(*in_args)[0].eval())
+
 
 class TestRadiusFromOccArea(utt.InferShapeTester):
     def setUp(self):