fixes (#580)

pyro-ppl · Nov 4, 2021 · c1a7258 · c1a7258
1 parent 20b12ed
commit c1a7258
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/funsor/__init__.py b/funsor/__init__.py
@@ -10,6 +10,7 @@
 from funsor.sum_product import MarkovProduct
 from funsor.tensor import Tensor, function
 from funsor.terms import (
+    Approximate,
     Cat,
     Funsor,
     Independent,
@@ -55,6 +56,7 @@
 
 __all__ = [
     "__version__",
+    "Approximate",
     "Array",
     "Bint",
     "Cat",

diff --git a/funsor/terms.py b/funsor/terms.py
@@ -1299,7 +1299,7 @@ class Approximate(Funsor):
 
     :param op: An associative operator.
     :type op: ~funsor.ops.AssociativeOp
-    :param Funsor model: An exact funsor depending on ``reduced_vars``.
+    :param Funsor model: An exact funsor depending on ``approx_vars``.
     :param Funsor guide: A proposal funsor guiding optional approximation.
     :param frozenset approx_vars: A set of variables over which to approximate.
     """