Make pyro.subsample appear in docs (#2353)

docs/source/primitives.rst

@@ -1,25 +1,11 @@
 Primitives
 ==========
 
-.. autofunction:: pyro.sample
-.. autofunction:: pyro.param
-.. autofunction:: pyro.module
-.. autofunction:: pyro.random_module
-.. autofunction:: pyro.factor
-.. autofunction:: pyro.deterministic
-
-.. autoclass:: pyro.plate
+.. automodule:: pyro.primitives
     :members:
     :undoc-members:
+    :show-inheritance:
+    :member-order: bysource
 
-.. autofunction:: pyro.plate_stack
-
-.. autofunction:: pyro.get_param_store
-.. autofunction:: pyro.clear_param_store
-
-.. autofunction:: pyro.validation_enabled
-.. autofunction:: pyro.enable_validation
 
 .. autofunction:: pyro.ops.jit.trace
-
-.. autofunction:: pyro.set_rng_seed

pyro/poutine/runtime.py

@@ -231,6 +231,7 @@ def effectful(fn=None, type=None):
     assert type is not None, "must provide a type label for operation {}".format(fn)
     assert type != "message", "cannot use 'message' as keyword"
 
+    @functools.wraps(fn)
     def _fn(*args, **kwargs):
 
         name = kwargs.pop("name", None)