diff --git a/docs/source/inference_algos.rst b/docs/source/inference_algos.rst index 2440f28377..153028d1f7 100644 --- a/docs/source/inference_algos.rst +++ b/docs/source/inference_algos.rst @@ -51,6 +51,12 @@ ELBO :show-inheritance: :member-order: bysource +.. automodule:: pyro.infer.tracetmc_elbo + :members: + :undoc-members: + :show-inheritance: + :member-order: bysource + Importance ---------- diff --git a/examples/air/main.py b/examples/air/main.py index 770666c706..55a4a0a459 100644 --- a/examples/air/main.py +++ b/examples/air/main.py @@ -245,7 +245,7 @@ def per_param_optim_args(module_name, param_name): if __name__ == '__main__': - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="Pyro AIR example", argument_default=argparse.SUPPRESS) parser.add_argument('-n', '--num-steps', type=int, default=int(1e8), help='number of optimization steps to take') diff --git a/examples/baseball.py b/examples/baseball.py index 7170e78ae6..aa9d0fd042 100644 --- a/examples/baseball.py +++ b/examples/baseball.py @@ -327,7 +327,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="Baseball batting average using HMC") parser.add_argument("-n", "--num-samples", nargs="?", default=200, type=int) parser.add_argument("--num-chains", nargs='?', default=4, type=int) diff --git a/examples/contrib/autoname/mixture.py b/examples/contrib/autoname/mixture.py index 37c06cb6ef..2993f83c54 100644 --- a/examples/contrib/autoname/mixture.py +++ b/examples/contrib/autoname/mixture.py @@ -72,7 +72,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="parse args") parser.add_argument('-n', '--num-epochs', default=200, type=int) parser.add_argument('--jit', action='store_true') diff --git a/examples/contrib/autoname/scoping_mixture.py b/examples/contrib/autoname/scoping_mixture.py index 384957c9f8..757bd7748b 100644 --- a/examples/contrib/autoname/scoping_mixture.py +++ b/examples/contrib/autoname/scoping_mixture.py @@ -64,7 +64,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="parse args") parser.add_argument('-n', '--num-epochs', default=200, type=int) args = parser.parse_args() diff --git a/examples/contrib/autoname/tree_data.py b/examples/contrib/autoname/tree_data.py index 368a74fddd..8054064d6e 100644 --- a/examples/contrib/autoname/tree_data.py +++ b/examples/contrib/autoname/tree_data.py @@ -102,7 +102,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="parse args") parser.add_argument('-n', '--num-epochs', default=100, type=int) args = parser.parse_args() diff --git a/examples/contrib/cevae/synthetic.py b/examples/contrib/cevae/synthetic.py index 6705878b2d..07732ec4f9 100644 --- a/examples/contrib/cevae/synthetic.py +++ b/examples/contrib/cevae/synthetic.py @@ -78,7 +78,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="Causal Effect Variational Autoencoder") parser.add_argument("--num-data", default=1000, type=int) parser.add_argument("--feature-dim", default=5, type=int) diff --git a/examples/contrib/gp/sv-dkl.py b/examples/contrib/gp/sv-dkl.py index 3eeb71c3f3..a31db5839f 100644 --- a/examples/contrib/gp/sv-dkl.py +++ b/examples/contrib/gp/sv-dkl.py @@ -162,7 +162,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description='Pyro GP MNIST Example') parser.add_argument('--data-dir', type=str, default=None, metavar='PATH', help='default directory to cache MNIST data') diff --git a/examples/contrib/oed/ab_test.py b/examples/contrib/oed/ab_test.py index 3b53575fd1..a2df774b59 100644 --- a/examples/contrib/oed/ab_test.py +++ b/examples/contrib/oed/ab_test.py @@ -112,7 +112,7 @@ def main(num_vi_steps, num_bo_steps, seed): if __name__ == "__main__": - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="A/B test experiment design using VI") parser.add_argument("-n", "--num-vi-steps", nargs="?", default=5000, type=int) parser.add_argument('--num-bo-steps', nargs="?", default=5, type=int) diff --git a/examples/contrib/timeseries/gp_models.py b/examples/contrib/timeseries/gp_models.py index 874460445d..939e81b065 100644 --- a/examples/contrib/timeseries/gp_models.py +++ b/examples/contrib/timeseries/gp_models.py @@ -146,7 +146,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="contrib.timeseries example usage") parser.add_argument("-n", "--num-steps", default=300, type=int) parser.add_argument("-s", "--seed", default=0, type=int) diff --git a/examples/dmm/dmm.py b/examples/dmm/dmm.py index a253d9a6c0..605f4889df 100644 --- a/examples/dmm/dmm.py +++ b/examples/dmm/dmm.py @@ -447,7 +447,7 @@ def do_evaluation(): # parse command-line arguments and execute the main method if __name__ == '__main__': - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="parse args") parser.add_argument('-n', '--num-epochs', type=int, default=5000) diff --git a/examples/eight_schools/mcmc.py b/examples/eight_schools/mcmc.py index 7e451b0d76..d41e58682a 100644 --- a/examples/eight_schools/mcmc.py +++ b/examples/eight_schools/mcmc.py @@ -39,7 +39,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description='Eight Schools MCMC') parser.add_argument('--num-samples', type=int, default=1000, help='number of MCMC samples (default: 1000)') diff --git a/examples/eight_schools/svi.py b/examples/eight_schools/svi.py index 9f5825293f..6bd6de69e8 100644 --- a/examples/eight_schools/svi.py +++ b/examples/eight_schools/svi.py @@ -72,7 +72,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description='Eight Schools SVI') parser.add_argument('--lr', type=float, default=0.01, help='learning rate (default: 0.01)') diff --git a/examples/hmm.py b/examples/hmm.py index 59fe0e6bbf..3975632220 100644 --- a/examples/hmm.py +++ b/examples/hmm.py @@ -635,7 +635,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="MAP Baum-Welch learning Bach Chorales") parser.add_argument("-m", "--model", default="1", type=str, help="one of: {}".format(", ".join(sorted(models.keys())))) diff --git a/examples/inclined_plane.py b/examples/inclined_plane.py index a6b89fdf82..3e9acb1a4d 100644 --- a/examples/inclined_plane.py +++ b/examples/inclined_plane.py @@ -121,7 +121,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="parse args") parser.add_argument('-n', '--num-samples', default=500, type=int) args = parser.parse_args() diff --git a/examples/lda.py b/examples/lda.py index 29ffbe686a..3664111b33 100644 --- a/examples/lda.py +++ b/examples/lda.py @@ -135,7 +135,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="Amortized Latent Dirichlet Allocation") parser.add_argument("-t", "--num-topics", default=8, type=int) parser.add_argument("-w", "--num-words", default=1024, type=int) diff --git a/examples/lkj.py b/examples/lkj.py index 030b983005..894ec6af34 100644 --- a/examples/lkj.py +++ b/examples/lkj.py @@ -46,7 +46,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="Demonstrate the use of an LKJ Prior") parser.add_argument("--num-samples", nargs="?", default=200, type=int) parser.add_argument("--n", nargs="?", default=500, type=int) diff --git a/examples/minipyro.py b/examples/minipyro.py index a9eba96960..03471cbb6d 100644 --- a/examples/minipyro.py +++ b/examples/minipyro.py @@ -62,7 +62,7 @@ def guide(data): if __name__ == "__main__": - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="Mini Pyro demo") parser.add_argument("-b", "--backend", default="minipyro") parser.add_argument("-n", "--num-steps", default=1001, type=int) diff --git a/examples/rsa/generics.py b/examples/rsa/generics.py index cf002ba971..92ddda7819 100644 --- a/examples/rsa/generics.py +++ b/examples/rsa/generics.py @@ -154,7 +154,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="parse args") parser.add_argument('-n', '--num-samples', default=10, type=int) args = parser.parse_args() diff --git a/examples/rsa/hyperbole.py b/examples/rsa/hyperbole.py index 6eb6908290..825e8b369f 100644 --- a/examples/rsa/hyperbole.py +++ b/examples/rsa/hyperbole.py @@ -151,7 +151,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="parse args") parser.add_argument('-n', '--num-samples', default=10, type=int) parser.add_argument('--price', default=10000, type=int) diff --git a/examples/rsa/schelling.py b/examples/rsa/schelling.py index 303e761a6a..0ce3d3774d 100644 --- a/examples/rsa/schelling.py +++ b/examples/rsa/schelling.py @@ -73,7 +73,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="parse args") parser.add_argument('-n', '--num-samples', default=10, type=int) parser.add_argument('--depth', default=2, type=int) diff --git a/examples/rsa/schelling_false.py b/examples/rsa/schelling_false.py index 2d304f16f7..da3da066a4 100644 --- a/examples/rsa/schelling_false.py +++ b/examples/rsa/schelling_false.py @@ -86,7 +86,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="parse args") parser.add_argument('-n', '--num-samples', default=10, type=int) parser.add_argument('--depth', default=3, type=int) diff --git a/examples/rsa/semantic_parsing.py b/examples/rsa/semantic_parsing.py index d6d9d08da2..cf75bc954c 100644 --- a/examples/rsa/semantic_parsing.py +++ b/examples/rsa/semantic_parsing.py @@ -337,7 +337,7 @@ def is_all_qud(world): if __name__ == "__main__": - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="parse args") parser.add_argument('-n', '--num-samples', default=10, type=int) args = parser.parse_args() diff --git a/examples/sparse_gamma_def.py b/examples/sparse_gamma_def.py index 169ffa8d82..5518d1fbca 100644 --- a/examples/sparse_gamma_def.py +++ b/examples/sparse_gamma_def.py @@ -231,7 +231,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') # parse command line arguments parser = argparse.ArgumentParser(description="parse args") parser.add_argument('-n', '--num-epochs', default=1500, type=int, help='number of training epochs') diff --git a/examples/sparse_regression.py b/examples/sparse_regression.py index 14e369cbf8..8b87be326e 100644 --- a/examples/sparse_regression.py +++ b/examples/sparse_regression.py @@ -309,7 +309,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description='Krylov KIT') parser.add_argument('--num-data', type=int, default=750) parser.add_argument('--num-steps', type=int, default=1000) diff --git a/examples/vae/ss_vae_M2.py b/examples/vae/ss_vae_M2.py index 8c2fc60a14..07928fbbee 100644 --- a/examples/vae/ss_vae_M2.py +++ b/examples/vae/ss_vae_M2.py @@ -380,7 +380,7 @@ def main(args): "-sup 3000 -zd 50 -hl 500 -lr 0.00042 -b1 0.95 -bs 200 -log ./tmp.log" if __name__ == "__main__": - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description="SS-VAE\n{}".format(EXAMPLE_RUN)) diff --git a/examples/vae/vae.py b/examples/vae/vae.py index aba7e2ad7a..a89ee4801b 100644 --- a/examples/vae/vae.py +++ b/examples/vae/vae.py @@ -198,7 +198,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') # parse command line arguments parser = argparse.ArgumentParser(description="parse args") parser.add_argument('-n', '--num-epochs', default=101, type=int, help='number of training epochs') diff --git a/examples/vae/vae_comparison.py b/examples/vae/vae_comparison.py index 6da8fb498c..8192ef74b4 100644 --- a/examples/vae/vae_comparison.py +++ b/examples/vae/vae_comparison.py @@ -243,7 +243,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('1.0.0') + assert pyro.__version__.startswith('1.1.0') parser = argparse.ArgumentParser(description='VAE using MNIST dataset') parser.add_argument('-n', '--num-epochs', nargs='?', default=10, type=int) parser.add_argument('--batch_size', nargs='?', default=128, type=int) diff --git a/pyro/__init__.py b/pyro/__init__.py index 7194bb88a5..7eb7ec7575 100644 --- a/pyro/__init__.py +++ b/pyro/__init__.py @@ -6,7 +6,7 @@ param, plate, plate_stack, random_module, sample, validation_enabled) from pyro.util import set_rng_seed -version_prefix = '1.0.0' +version_prefix = '1.1.0' # Get the __version__ string from the auto-generated _version.py file, if exists. try: diff --git a/pyro/util.py b/pyro/util.py index 192c6e1cf7..1fc1d5ed41 100644 --- a/pyro/util.py +++ b/pyro/util.py @@ -6,7 +6,6 @@ from collections import defaultdict from itertools import zip_longest -import graphviz import torch from contextlib import contextmanager @@ -92,22 +91,28 @@ def warn_if_inf(value, msg="", allow_posinf=False, allow_neginf=False): def save_visualization(trace, graph_output): """ - :param pyro.poutine.Trace trace: a trace to be visualized - :param graph_output: the graph will be saved to graph_output.pdf - :type graph_output: str - - Take a trace generated by poutine.trace with `graph_type='dense'` and render - the graph with the output saved to file. + (DEPRECATED) Take a trace generated by poutine.trace with `graph_type='dense'` + and render the graph with the output saved to file. - non-reparameterized stochastic nodes are salmon - reparameterized stochastic nodes are half salmon, half grey - observation nodes are green + :param pyro.poutine.Trace trace: a trace to be visualized + :param graph_output: the graph will be saved to graph_output.pdf + :type graph_output: str + + Example: trace = pyro.poutine.trace(model, graph_type="dense").get_trace() save_visualization(trace, 'output') """ + warnings.warn("`save_visualization` function is deprecated and will be removed in " + "a future version.") + + import graphviz + g = graphviz.Digraph() for label, node in trace.nodes.items(): diff --git a/setup.py b/setup.py index fcc86c9d1d..f5fc42981b 100644 --- a/setup.py +++ b/setup.py @@ -55,6 +55,7 @@ # examples/tutorials EXTRAS_REQUIRE = [ 'jupyter>=1.0.0', + 'graphviz>=0.8', 'matplotlib>=1.3', 'pillow', 'torchvision>=0.4.0', @@ -78,7 +79,6 @@ install_requires=[ # if you add any additional libraries, please also # add them to `docs/requirements.txt` - 'graphviz>=0.8', # numpy is necessary for some functionality of PyTorch 'numpy>=1.7', 'opt_einsum>=2.3.2', diff --git a/tutorial/source/air.ipynb b/tutorial/source/air.ipynb index 1cb6d9a8a8..4633c51c69 100644 --- a/tutorial/source/air.ipynb +++ b/tutorial/source/air.ipynb @@ -41,7 +41,7 @@ "import numpy as np\n", "\n", "smoke_test = ('CI' in os.environ)\n", - "assert pyro.__version__.startswith('1.0.0')\n", + "assert pyro.__version__.startswith('1.1.0')\n", "pyro.enable_validation(True)" ] }, diff --git a/tutorial/source/bayesian_regression.ipynb b/tutorial/source/bayesian_regression.ipynb index 36281b1a89..698d9b815d 100644 --- a/tutorial/source/bayesian_regression.ipynb +++ b/tutorial/source/bayesian_regression.ipynb @@ -69,7 +69,7 @@ "\n", "# for CI testing\n", "smoke_test = ('CI' in os.environ)\n", - "assert pyro.__version__.startswith('1.0.0')\n", + "assert pyro.__version__.startswith('1.1.0')\n", "pyro.enable_validation(True)\n", "pyro.set_rng_seed(1)\n", "pyro.enable_validation(True)\n", diff --git a/tutorial/source/bayesian_regression_ii.ipynb b/tutorial/source/bayesian_regression_ii.ipynb index 079028d6e5..72d5789bbf 100644 --- a/tutorial/source/bayesian_regression_ii.ipynb +++ b/tutorial/source/bayesian_regression_ii.ipynb @@ -44,7 +44,7 @@ "import pyro.optim as optim\n", "\n", "pyro.set_rng_seed(1)\n", - "assert pyro.__version__.startswith('1.0.0')" + "assert pyro.__version__.startswith('1.1.0')" ] }, { diff --git a/tutorial/source/bo.ipynb b/tutorial/source/bo.ipynb index 1505b4b01e..b05cd40254 100644 --- a/tutorial/source/bo.ipynb +++ b/tutorial/source/bo.ipynb @@ -54,7 +54,7 @@ "import pyro\n", "import pyro.contrib.gp as gp\n", "\n", - "assert pyro.__version__.startswith('1.0.0')\n", + "assert pyro.__version__.startswith('1.1.0')\n", "pyro.enable_validation(True) # can help with debugging\n", "pyro.set_rng_seed(1)" ] diff --git a/tutorial/source/easyguide.ipynb b/tutorial/source/easyguide.ipynb index 0921214e89..1ca63b58fa 100644 --- a/tutorial/source/easyguide.ipynb +++ b/tutorial/source/easyguide.ipynb @@ -45,7 +45,7 @@ "\n", "pyro.enable_validation(True)\n", "smoke_test = ('CI' in os.environ)\n", - "assert pyro.__version__.startswith('1.0.0')" + "assert pyro.__version__.startswith('1.1.0')" ] }, { diff --git a/tutorial/source/ekf.ipynb b/tutorial/source/ekf.ipynb index ceb56c85ae..6a73303efc 100644 --- a/tutorial/source/ekf.ipynb +++ b/tutorial/source/ekf.ipynb @@ -98,7 +98,7 @@ "from pyro.contrib.tracking.measurements import PositionMeasurement\n", "\n", "smoke_test = ('CI' in os.environ)\n", - "assert pyro.__version__.startswith('1.0.0')\n", + "assert pyro.__version__.startswith('1.1.0')\n", "pyro.enable_validation(True)" ] }, diff --git a/tutorial/source/enumeration.ipynb b/tutorial/source/enumeration.ipynb index 05039f1be2..ae42af690a 100644 --- a/tutorial/source/enumeration.ipynb +++ b/tutorial/source/enumeration.ipynb @@ -50,7 +50,7 @@ "from pyro.ops.indexing import Vindex\n", "\n", "smoke_test = ('CI' in os.environ)\n", - "assert pyro.__version__.startswith('1.0.0')\n", + "assert pyro.__version__.startswith('1.1.0')\n", "pyro.enable_validation()\n", "pyro.set_rng_seed(0)" ] diff --git a/tutorial/source/gmm.ipynb b/tutorial/source/gmm.ipynb index dfcc3b6402..fe957e981e 100644 --- a/tutorial/source/gmm.ipynb +++ b/tutorial/source/gmm.ipynb @@ -41,7 +41,7 @@ "from pyro.infer import SVI, TraceEnum_ELBO, config_enumerate, infer_discrete\n", "\n", "smoke_test = ('CI' in os.environ)\n", - "assert pyro.__version__.startswith('1.0.0')\n", + "assert pyro.__version__.startswith('1.1.0')\n", "pyro.enable_validation(True)" ] }, diff --git a/tutorial/source/gp.ipynb b/tutorial/source/gp.ipynb index 580167bcb1..2ad1374e9b 100644 --- a/tutorial/source/gp.ipynb +++ b/tutorial/source/gp.ipynb @@ -60,7 +60,7 @@ "import pyro.distributions as dist\n", "\n", "smoke_test = ('CI' in os.environ) # ignore; used to check code integrity in the Pyro repo\n", - "assert pyro.__version__.startswith('1.0.0')\n", + "assert pyro.__version__.startswith('1.1.0')\n", "pyro.enable_validation(True) # can help with debugging\n", "pyro.set_rng_seed(0)" ] diff --git a/tutorial/source/gplvm.ipynb b/tutorial/source/gplvm.ipynb index 69d8ca209b..4a4f34736c 100644 --- a/tutorial/source/gplvm.ipynb +++ b/tutorial/source/gplvm.ipynb @@ -39,7 +39,7 @@ "import pyro.ops.stats as stats\n", "\n", "smoke_test = ('CI' in os.environ) # ignore; used to check code integrity in the Pyro repo\n", - "assert pyro.__version__.startswith('1.0.0')\n", + "assert pyro.__version__.startswith('1.1.0')\n", "pyro.enable_validation(True) # can help with debugging\n", "pyro.set_rng_seed(1)" ] diff --git a/tutorial/source/jit.ipynb b/tutorial/source/jit.ipynb index 363c708283..42fecd3a1a 100644 --- a/tutorial/source/jit.ipynb +++ b/tutorial/source/jit.ipynb @@ -48,7 +48,7 @@ "from pyro.optim import Adam\n", "\n", "smoke_test = ('CI' in os.environ)\n", - "assert pyro.__version__.startswith('1.0.0')\n", + "assert pyro.__version__.startswith('1.1.0')\n", "pyro.enable_validation(True) # <---- This is always a good idea!" ] }, diff --git a/tutorial/source/svi_part_i.ipynb b/tutorial/source/svi_part_i.ipynb index 379d0afbfd..ac9489738f 100644 --- a/tutorial/source/svi_part_i.ipynb +++ b/tutorial/source/svi_part_i.ipynb @@ -265,7 +265,7 @@ "n_steps = 2 if smoke_test else 2000\n", "\n", "# enable validation (e.g. validate parameters of distributions)\n", - "assert pyro.__version__.startswith('1.0.0')\n", + "assert pyro.__version__.startswith('1.1.0')\n", "pyro.enable_validation(True)\n", "\n", "# clear the param store in case we're in a REPL\n", diff --git a/tutorial/source/svi_part_iii.ipynb b/tutorial/source/svi_part_iii.ipynb index 9be1fdb215..b0de1a8f15 100644 --- a/tutorial/source/svi_part_iii.ipynb +++ b/tutorial/source/svi_part_iii.ipynb @@ -284,7 +284,7 @@ "import sys\n", "\n", "# enable validation (e.g. validate parameters of distributions)\n", - "assert pyro.__version__.startswith('1.0.0')\n", + "assert pyro.__version__.startswith('1.1.0')\n", "pyro.enable_validation(True)\n", "\n", "# this is for running the notebook in our testing framework\n", diff --git a/tutorial/source/tensor_shapes.ipynb b/tutorial/source/tensor_shapes.ipynb index f06849d3c9..a0ab599be0 100644 --- a/tutorial/source/tensor_shapes.ipynb +++ b/tutorial/source/tensor_shapes.ipynb @@ -52,7 +52,7 @@ "from pyro.optim import Adam\n", "\n", "smoke_test = ('CI' in os.environ)\n", - "assert pyro.__version__.startswith('1.0.0')\n", + "assert pyro.__version__.startswith('1.1.0')\n", "pyro.enable_validation(True) # <---- This is always a good idea!\n", "\n", "# We'll ue this helper to check our models are correct.\n", diff --git a/tutorial/source/tracking_1d.ipynb b/tutorial/source/tracking_1d.ipynb index 73b679ca5e..f7c54bbb4c 100644 --- a/tutorial/source/tracking_1d.ipynb +++ b/tutorial/source/tracking_1d.ipynb @@ -30,7 +30,7 @@ "from pyro.optim import Adam\n", "\n", "%matplotlib inline\n", - "assert pyro.__version__.startswith('1.0.0')\n", + "assert pyro.__version__.startswith('1.1.0')\n", "pyro.enable_validation(True)\n", "smoke_test = ('CI' in os.environ)" ] diff --git a/tutorial/source/vae.ipynb b/tutorial/source/vae.ipynb index b7ce341319..cddae54b36 100644 --- a/tutorial/source/vae.ipynb +++ b/tutorial/source/vae.ipynb @@ -114,7 +114,7 @@ "metadata": {}, "outputs": [], "source": [ - "assert pyro.__version__.startswith('1.0.0')\n", + "assert pyro.__version__.startswith('1.1.0')\n", "pyro.enable_validation(True)\n", "pyro.distributions.enable_validation(False)\n", "pyro.set_rng_seed(0)\n",