Fixes pyro-ppl#3430 (serializing an easyguide corrupts it).

Author: BenZickel

pyro/contrib/easyguide/easyguide.py

@@ -230,7 +230,7 @@ def __init__(self, guide, sites):
         self.event_shape = torch.Size([sum(self._site_sizes.values())])
 
     def __getstate__(self):
-        state = getattr(super(), "__getstate__", self.__dict__.copy)()
+        state = getattr(super(), "__getstate__", lambda: self.__dict__)().copy()
         state["_guide"] = state["_guide"]()  # weakref -> ref
         return state
 

tests/contrib/easyguide/test_easyguide.py

@@ -79,7 +79,6 @@ def guide(self, batch, subsample, full_size):
             self.group(match="state_[0-9]*").map_estimate()
 
 
-@pytest.mark.xfail(reason="https://github.com/pyro-ppl/pyro/issues/3430")
 def test_serialize():
     guide = PickleGuide(model)
     check_guide(guide)