diff --git a/.travis.yml b/.travis.yml index eb2effbcfd..c020b511b7 100644 --- a/.travis.yml +++ b/.travis.yml @@ -24,6 +24,8 @@ cache: install: - pip install -U pip + # Keep track of pyro-api master branch + - pip install https://github.com/pyro-ppl/pyro-api/archive/master.zip - pip install torch==1.3.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html - pip install .[test] - pip freeze diff --git a/docs/requirements.txt b/docs/requirements.txt index 50ad60dd0e..2ec8ba2801 100644 --- a/docs/requirements.txt +++ b/docs/requirements.txt @@ -5,4 +5,5 @@ graphviz>=0.8 numpy>=1.7 observations>=0.1.4 opt_einsum>=2.3.2 +pyro-api>=0.1.0 tqdm>=4.31 diff --git a/docs/source/pyro.nn.txt b/docs/source/pyro.nn.txt index 4d617d16d4..2b3651ef3f 100644 --- a/docs/source/pyro.nn.txt +++ b/docs/source/pyro.nn.txt @@ -1,7 +1,15 @@ AutoRegressiveNN ---------------- -.. automodule:: pyro.nn.auto_reg_nn +.. autoclass:: pyro.nn.auto_reg_nn.AutoRegressiveNN + :members: + :undoc-members: + :show-inheritance: + +ConditionalAutoRegressiveNN +--------------------------- + +.. autoclass:: pyro.nn.auto_reg_nn.ConditionalAutoRegressiveNN :members: :undoc-members: :show-inheritance: diff --git a/examples/air/main.py b/examples/air/main.py index 92b0bbae67..b34527b1d7 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('0.4.1') + assert pyro.__version__.startswith('0.5.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 1132237102..281a1b1163 100644 --- a/examples/baseball.py +++ b/examples/baseball.py @@ -359,7 +359,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith('0.4.1') + assert pyro.__version__.startswith('0.5.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/bayesian_regression.py b/examples/bayesian_regression.py index 6db00f3681..8e2a78b7c2 100644 --- a/examples/bayesian_regression.py +++ b/examples/bayesian_regression.py @@ -137,7 +137,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('0.4.1') + assert pyro.__version__.startswith('0.5.0') parser = argparse.ArgumentParser(description="parse args") parser.add_argument('-n', '--num-epochs', default=1000, type=int) parser.add_argument('-b', '--batch-size', default=N, type=int) diff --git a/examples/contrib/autoname/mixture.py b/examples/contrib/autoname/mixture.py index 261c02e209..ca5c622b25 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('0.4.1') + assert pyro.__version__.startswith('0.5.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 292427a926..85e95f107f 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('0.4.1') + assert pyro.__version__.startswith('0.5.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 fd5acb54ca..b2e836dcc5 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('0.4.1') + assert pyro.__version__.startswith('0.5.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/gp/sv-dkl.py b/examples/contrib/gp/sv-dkl.py index c85d3f7263..87d58417ca 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('0.4.1') + assert pyro.__version__.startswith('0.5.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 16241fab8f..987c667eaa 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('0.4.1') + assert pyro.__version__.startswith('0.5.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/dmm/dmm.py b/examples/dmm/dmm.py index 0b1182437b..9650244af8 100644 --- a/examples/dmm/dmm.py +++ b/examples/dmm/dmm.py @@ -433,7 +433,7 @@ def do_evaluation(): # parse command-line arguments and execute the main method if __name__ == '__main__': - assert pyro.__version__.startswith('0.4.1') + assert pyro.__version__.startswith('0.5.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 45f708b1fc..06239d86a0 100644 --- a/examples/eight_schools/mcmc.py +++ b/examples/eight_schools/mcmc.py @@ -40,7 +40,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('0.4.1') + assert pyro.__version__.startswith('0.5.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 c65921d6f4..263e7f11e8 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('0.4.1') + assert pyro.__version__.startswith('0.5.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 37c0b3bb83..3c7c5266f9 100644 --- a/examples/hmm.py +++ b/examples/hmm.py @@ -626,7 +626,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('0.4.1') + assert pyro.__version__.startswith('0.5.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 1419b5395b..a92f9a28d1 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('0.4.1') + assert pyro.__version__.startswith('0.5.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 fdbfce2bcf..6ebb83cade 100644 --- a/examples/lda.py +++ b/examples/lda.py @@ -135,7 +135,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('0.4.1') + assert pyro.__version__.startswith('0.5.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 1f74d1a7c9..13262b266b 100644 --- a/examples/lkj.py +++ b/examples/lkj.py @@ -46,7 +46,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith('0.4.1') + assert pyro.__version__.startswith('0.5.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 9224bd4447..8934bc280a 100644 --- a/examples/minipyro.py +++ b/examples/minipyro.py @@ -62,7 +62,7 @@ def guide(data): if __name__ == "__main__": - assert pyro.__version__.startswith('0.4.1') + assert pyro.__version__.startswith('0.5.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 071301f1e5..12aadef94a 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('0.4.1') + assert pyro.__version__.startswith('0.5.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 468d8f0344..379e542ac0 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('0.4.1') + assert pyro.__version__.startswith('0.5.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 acb6d0bcf7..0cf610409b 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('0.4.1') + assert pyro.__version__.startswith('0.5.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 8f23907881..377b9bf176 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('0.4.1') + assert pyro.__version__.startswith('0.5.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 aef24d82e5..368c09493a 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('0.4.1') + assert pyro.__version__.startswith('0.5.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 8dc028f49c..20a572d0ed 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('0.4.1') + assert pyro.__version__.startswith('0.5.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 a2ebcb375a..5c94663a50 100644 --- a/examples/sparse_regression.py +++ b/examples/sparse_regression.py @@ -306,7 +306,7 @@ def main(args): if __name__ == '__main__': - assert pyro.__version__.startswith('0.4.1') + assert pyro.__version__.startswith('0.5.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 5367238bdd..8cf0031be8 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('0.4.1') + assert pyro.__version__.startswith('0.5.0') parser = argparse.ArgumentParser(description="SS-VAE\n{}".format(EXAMPLE_RUN)) diff --git a/examples/vae/vae.py b/examples/vae/vae.py index b5a988fdb3..1b550c3385 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('0.4.1') + assert pyro.__version__.startswith('0.5.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 2171decb3f..ab7fff5162 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('0.4.1') + assert pyro.__version__.startswith('0.5.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 067b7527c2..8422e32ff2 100644 --- a/pyro/__init__.py +++ b/pyro/__init__.py @@ -5,7 +5,7 @@ param, plate, plate_stack, random_module, sample, validation_enabled) from pyro.util import set_rng_seed -version_prefix = '0.4.1' +version_prefix = '0.5.0' # Get the __version__ string from the auto-generated _version.py file, if exists. try: diff --git a/pyro/infer/abstract_infer.py b/pyro/infer/abstract_infer.py index b525051bda..02f2b51c5b 100644 --- a/pyro/infer/abstract_infer.py +++ b/pyro/infer/abstract_infer.py @@ -289,9 +289,8 @@ def __init__(self, model, posterior, num_samples, keep_sites=None): self.num_samples = num_samples self.keep_sites = keep_sites super(TracePredictive, self).__init__() - warnings.warn('This class will be deprecated in the next release. For MCMC, ' - 'please use the `pyro.infer.mcmc.util.predictive` function ' - 'instead.', PendingDeprecationWarning) + warnings.warn('This class will be removed in the next release. Use the ' + '`pyro.infer.Predictive` class instead.', DeprecationWarning) def _traces(self, *args, **kwargs): if not self.posterior.exec_traces: diff --git a/pyro/infer/elbo.py b/pyro/infer/elbo.py index fef5596618..d4fabd6a6e 100644 --- a/pyro/infer/elbo.py +++ b/pyro/infer/elbo.py @@ -39,7 +39,7 @@ class ELBO(object, metaclass=ABCMeta): tracer. When this is True, all :class:`torch.jit.TracerWarning` will be ignored. Defaults to False. :param bool jit_options: Optional dict of options to pass to - :func:`torch.jit.trace` , e.g. ``{"optimize": False}``. + :func:`torch.jit.trace` , e.g. ``{"check_trace": True}``. :param bool retain_graph: Whether to retain autograd graph during an SVI step. Defaults to None (False). :param float tail_adaptive_beta: Exponent beta with ``-1.0 <= beta < 0.0`` for diff --git a/pyro/infer/svi.py b/pyro/infer/svi.py index f1e81f8843..5f52d9bd86 100644 --- a/pyro/infer/svi.py +++ b/pyro/infer/svi.py @@ -1,3 +1,5 @@ +import warnings + import torch import pyro @@ -62,6 +64,9 @@ def _loss_and_grads(*args, **kwargs): self.loss_and_grads = loss_and_grads def run(self, *args, **kwargs): + warnings.warn('SVI will not derive from TracePosterior, and this method might be ' + 'unavailable in future releases. For predictions, use the ' + '`pyro.infer.Predictive` class directly.', DeprecationWarning) if self.num_steps > 0: with poutine.block(): for i in range(self.num_steps): diff --git a/setup.py b/setup.py index a10cb5061f..426c94f6a2 100644 --- a/setup.py +++ b/setup.py @@ -84,7 +84,7 @@ # numpy is necessary for some functionality of PyTorch 'numpy>=1.7', 'opt_einsum>=2.3.2', - 'pyro-api@https://api.github.com/repos/pyro-ppl/pyro-api/tarball/master', + 'pyro-api>=0.1.0', 'torch>=1.2.0', 'tqdm>=4.31', ], @@ -114,7 +114,6 @@ 'yapf', ], }, - tests_require=['flake8', 'pytest>=4.1'], keywords='machine learning statistics probabilistic programming bayesian modeling pytorch', license='MIT License', classifiers=[ diff --git a/tutorial/source/air.ipynb b/tutorial/source/air.ipynb index 0749e6ae86..08ee768af6 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('0.4.1')\n", + "assert pyro.__version__.startswith('0.5.0')\n", "pyro.enable_validation(True)" ] }, diff --git a/tutorial/source/bayesian_regression.ipynb b/tutorial/source/bayesian_regression.ipynb index ce082e77d5..b64b470b56 100644 --- a/tutorial/source/bayesian_regression.ipynb +++ b/tutorial/source/bayesian_regression.ipynb @@ -55,7 +55,7 @@ "\n", "# for CI testing\n", "smoke_test = ('CI' in os.environ)\n", - "assert pyro.__version__.startswith('0.4.1')\n", + "assert pyro.__version__.startswith('0.5.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 7731dcf284..58db56ba99 100644 --- a/tutorial/source/bayesian_regression_ii.ipynb +++ b/tutorial/source/bayesian_regression_ii.ipynb @@ -39,7 +39,7 @@ "import pyro.optim as optim\n", "\n", "pyro.set_rng_seed(1)\n", - "assert pyro.__version__.startswith('0.4.1')" + "assert pyro.__version__.startswith('0.5.0')" ] }, { diff --git a/tutorial/source/bo.ipynb b/tutorial/source/bo.ipynb index 66d50ca170..b4acdf5889 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('0.4.1')\n", + "assert pyro.__version__.startswith('0.5.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 aa086cb1e1..ac011f5103 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('0.4.1')" + "assert pyro.__version__.startswith('0.5.0')" ] }, { diff --git a/tutorial/source/ekf.ipynb b/tutorial/source/ekf.ipynb index 5a42b00983..5e394392bf 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('0.4.1')\n", + "assert pyro.__version__.startswith('0.5.0')\n", "pyro.enable_validation(True)" ] }, diff --git a/tutorial/source/enumeration.ipynb b/tutorial/source/enumeration.ipynb index 7535d11daf..8b4f9341c7 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('0.4.1')\n", + "assert pyro.__version__.startswith('0.5.0')\n", "pyro.enable_validation()\n", "pyro.set_rng_seed(0)" ] diff --git a/tutorial/source/gmm.ipynb b/tutorial/source/gmm.ipynb index 9b0f1c79d7..2d43ad110f 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('0.4.1')\n", + "assert pyro.__version__.startswith('0.5.0')\n", "pyro.enable_validation(True)" ] }, diff --git a/tutorial/source/gp.ipynb b/tutorial/source/gp.ipynb index 0c8c4e6158..b5569ebe08 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('0.4.1')\n", + "assert pyro.__version__.startswith('0.5.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 67039df9df..a13e533a84 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('0.4.1')\n", + "assert pyro.__version__.startswith('0.5.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 5e97790105..8c532d717f 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('0.4.1')\n", + "assert pyro.__version__.startswith('0.5.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 9a5441e0e4..b759e4c9e8 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('0.4.1')\n", + "assert pyro.__version__.startswith('0.5.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 b445d7304c..e16f7f6ad3 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('0.4.1')\n", + "assert pyro.__version__.startswith('0.5.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 2a64b49c9f..85bb8c7a7a 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('0.4.1')\n", + "assert pyro.__version__.startswith('0.5.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 848b3b1808..53c2d35ac5 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('0.4.1')\n", + "assert pyro.__version__.startswith('0.5.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 73ccfcf2b3..e578706376 100644 --- a/tutorial/source/vae.ipynb +++ b/tutorial/source/vae.ipynb @@ -114,7 +114,7 @@ "metadata": {}, "outputs": [], "source": [ - "assert pyro.__version__.startswith('0.4.1')\n", + "assert pyro.__version__.startswith('0.5.0')\n", "pyro.enable_validation(True)\n", "pyro.distributions.enable_validation(False)\n", "pyro.set_rng_seed(0)\n",