From ad53c727fea60e878a7527c9c8fdaee704a2841d Mon Sep 17 00:00:00 2001 From: Fritz Obermeyer Date: Tue, 6 Sep 2022 15:17:24 -0400 Subject: [PATCH] Bump version to 1.8.2 (#3135) * Bump version to 1.8.2 * Run scripts/update_version.py * Update to newer seaborn.kdeplot args * Fix sns.scatterplot --- examples/air/main.py | 2 +- examples/baseball.py | 2 +- examples/contrib/autoname/mixture.py | 2 +- examples/contrib/autoname/scoping_mixture.py | 2 +- examples/contrib/autoname/tree_data.py | 2 +- examples/contrib/cevae/synthetic.py | 2 +- examples/contrib/epidemiology/regional.py | 2 +- examples/contrib/epidemiology/sir.py | 2 +- examples/contrib/forecast/bart.py | 2 +- examples/contrib/funsor/hmm.py | 2 +- examples/contrib/gp/sv-dkl.py | 2 +- examples/contrib/oed/ab_test.py | 2 +- examples/contrib/timeseries/gp_models.py | 2 +- examples/cvae/main.py | 2 +- examples/dmm.py | 2 +- examples/eight_schools/mcmc.py | 2 +- examples/eight_schools/svi.py | 2 +- examples/hmm.py | 2 +- examples/inclined_plane.py | 2 +- examples/lda.py | 2 +- examples/lkj.py | 2 +- examples/minipyro.py | 2 +- examples/neutra.py | 16 ++++++------ examples/rsa/generics.py | 2 +- examples/rsa/hyperbole.py | 2 +- examples/rsa/schelling.py | 2 +- examples/rsa/schelling_false.py | 2 +- examples/rsa/semantic_parsing.py | 2 +- examples/scanvi/scanvi.py | 2 +- examples/sir_hmc.py | 2 +- examples/sparse_gamma_def.py | 2 +- examples/sparse_regression.py | 2 +- examples/svi_horovod.py | 2 +- .../toy_mixture_model_discrete_enumeration.py | 2 +- examples/vae/ss_vae_M2.py | 2 +- examples/vae/vae.py | 2 +- examples/vae/vae_comparison.py | 2 +- pyro/__init__.py | 2 +- setup.py | 2 +- tutorial/source/air.ipynb | 2 +- tutorial/source/bayesian_regression.ipynb | 12 ++++----- tutorial/source/bayesian_regression_ii.ipynb | 26 +++++++++---------- tutorial/source/bo.ipynb | 2 +- .../source/dirichlet_process_mixture.ipynb | 2 +- tutorial/source/easyguide.ipynb | 2 +- tutorial/source/ekf.ipynb | 2 +- tutorial/source/enumeration.ipynb | 2 +- tutorial/source/epi_intro.ipynb | 2 +- tutorial/source/forecasting_dlm.ipynb | 2 +- tutorial/source/forecasting_i.ipynb | 2 +- tutorial/source/forecasting_ii.ipynb | 2 +- tutorial/source/forecasting_iii.ipynb | 2 +- tutorial/source/gmm.ipynb | 2 +- tutorial/source/gp.ipynb | 2 +- tutorial/source/gplvm.ipynb | 2 +- tutorial/source/intro_long.ipynb | 10 +++---- tutorial/source/jit.ipynb | 2 +- tutorial/source/model_rendering.ipynb | 2 +- tutorial/source/modules.ipynb | 2 +- tutorial/source/prior_predictive.ipynb | 2 +- tutorial/source/prodlda.ipynb | 2 +- tutorial/source/stable.ipynb | 2 +- tutorial/source/svi_part_i.ipynb | 2 +- tutorial/source/svi_part_iii.ipynb | 2 +- tutorial/source/tensor_shapes.ipynb | 2 +- tutorial/source/tracking_1d.ipynb | 2 +- tutorial/source/vae.ipynb | 2 +- 67 files changed, 95 insertions(+), 95 deletions(-) diff --git a/examples/air/main.py b/examples/air/main.py index 9aae6aa465..3725bd4aea 100644 --- a/examples/air/main.py +++ b/examples/air/main.py @@ -272,7 +272,7 @@ def per_param_optim_args(param_name): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") parser = argparse.ArgumentParser( description="Pyro AIR example", argument_default=argparse.SUPPRESS ) diff --git a/examples/baseball.py b/examples/baseball.py index 02da563027..10a6806b8d 100644 --- a/examples/baseball.py +++ b/examples/baseball.py @@ -392,7 +392,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") 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 dab2a10b40..96c68ee44d 100644 --- a/examples/contrib/autoname/mixture.py +++ b/examples/contrib/autoname/mixture.py @@ -74,7 +74,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") 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 0701ff00ae..717c4f4b51 100644 --- a/examples/contrib/autoname/scoping_mixture.py +++ b/examples/contrib/autoname/scoping_mixture.py @@ -71,7 +71,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") 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 88495e430c..1a0f4440f8 100644 --- a/examples/contrib/autoname/tree_data.py +++ b/examples/contrib/autoname/tree_data.py @@ -104,7 +104,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") 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 5b49e57824..919ebd61e3 100644 --- a/examples/contrib/cevae/synthetic.py +++ b/examples/contrib/cevae/synthetic.py @@ -86,7 +86,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") parser = argparse.ArgumentParser( description="Causal Effect Variational Autoencoder" ) diff --git a/examples/contrib/epidemiology/regional.py b/examples/contrib/epidemiology/regional.py index 46566fd561..02205028c9 100644 --- a/examples/contrib/epidemiology/regional.py +++ b/examples/contrib/epidemiology/regional.py @@ -166,7 +166,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") parser = argparse.ArgumentParser( description="Regional compartmental epidemiology modeling using HMC" ) diff --git a/examples/contrib/epidemiology/sir.py b/examples/contrib/epidemiology/sir.py index 5891cac1f7..7cc817a811 100644 --- a/examples/contrib/epidemiology/sir.py +++ b/examples/contrib/epidemiology/sir.py @@ -334,7 +334,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") parser = argparse.ArgumentParser( description="Compartmental epidemiology modeling using HMC" ) diff --git a/examples/contrib/forecast/bart.py b/examples/contrib/forecast/bart.py index ba65a70745..a75c26de8d 100644 --- a/examples/contrib/forecast/bart.py +++ b/examples/contrib/forecast/bart.py @@ -165,7 +165,7 @@ def transform(pred, truth): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") parser = argparse.ArgumentParser(description="Bart Ridership Forecasting Example") parser.add_argument("--train-window", default=2160, type=int) parser.add_argument("--test-window", default=336, type=int) diff --git a/examples/contrib/funsor/hmm.py b/examples/contrib/funsor/hmm.py index ce8d26a7c0..dfe930e056 100644 --- a/examples/contrib/funsor/hmm.py +++ b/examples/contrib/funsor/hmm.py @@ -820,7 +820,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") parser = argparse.ArgumentParser( description="MAP Baum-Welch learning Bach Chorales" ) diff --git a/examples/contrib/gp/sv-dkl.py b/examples/contrib/gp/sv-dkl.py index 222827a8a1..ea49069981 100644 --- a/examples/contrib/gp/sv-dkl.py +++ b/examples/contrib/gp/sv-dkl.py @@ -193,7 +193,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") parser = argparse.ArgumentParser(description="Pyro GP MNIST Example") parser.add_argument( "--data-dir", diff --git a/examples/contrib/oed/ab_test.py b/examples/contrib/oed/ab_test.py index 67115d75ad..32ad730987 100644 --- a/examples/contrib/oed/ab_test.py +++ b/examples/contrib/oed/ab_test.py @@ -125,7 +125,7 @@ def main(num_vi_steps, num_bo_steps, seed): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") 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 5e55019c98..4d95613b8f 100644 --- a/examples/contrib/timeseries/gp_models.py +++ b/examples/contrib/timeseries/gp_models.py @@ -186,7 +186,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") 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/cvae/main.py b/examples/cvae/main.py index 0f4acc6e9e..18cdcaa959 100644 --- a/examples/cvae/main.py +++ b/examples/cvae/main.py @@ -87,7 +87,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") # parse command line arguments parser = argparse.ArgumentParser(description="parse args") parser.add_argument( diff --git a/examples/dmm.py b/examples/dmm.py index 9c6ab2ff75..9731530849 100644 --- a/examples/dmm.py +++ b/examples/dmm.py @@ -571,7 +571,7 @@ def do_evaluation(): # parse command-line arguments and execute the main method if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") 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 f16ae063fc..0f5e61159a 100644 --- a/examples/eight_schools/mcmc.py +++ b/examples/eight_schools/mcmc.py @@ -43,7 +43,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") parser = argparse.ArgumentParser(description="Eight Schools MCMC") parser.add_argument( "--num-samples", diff --git a/examples/eight_schools/svi.py b/examples/eight_schools/svi.py index af14d4db09..ace2c24a52 100644 --- a/examples/eight_schools/svi.py +++ b/examples/eight_schools/svi.py @@ -81,7 +81,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") 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 3cbbfdc758..0881cb2c91 100644 --- a/examples/hmm.py +++ b/examples/hmm.py @@ -734,7 +734,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") parser = argparse.ArgumentParser( description="MAP Baum-Welch learning Bach Chorales" ) diff --git a/examples/inclined_plane.py b/examples/inclined_plane.py index c2cfb71b41..11f66dbe94 100644 --- a/examples/inclined_plane.py +++ b/examples/inclined_plane.py @@ -145,7 +145,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") 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 88230e5398..3796757d69 100644 --- a/examples/lda.py +++ b/examples/lda.py @@ -149,7 +149,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") parser = argparse.ArgumentParser( description="Amortized Latent Dirichlet Allocation" ) diff --git a/examples/lkj.py b/examples/lkj.py index 0363041ae6..04b4700089 100644 --- a/examples/lkj.py +++ b/examples/lkj.py @@ -56,7 +56,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") 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 086267ece4..524ff96130 100644 --- a/examples/minipyro.py +++ b/examples/minipyro.py @@ -65,7 +65,7 @@ def guide(data): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") 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/neutra.py b/examples/neutra.py index cbd2071e62..4eaa8e78bc 100644 --- a/examples/neutra.py +++ b/examples/neutra.py @@ -140,7 +140,7 @@ def main(args): ylim=ylim, title="Posterior \n(vanilla HMC)", ) - sns.kdeplot(vanilla_samples[:, 0], vanilla_samples[:, 1], ax=ax2) + sns.kdeplot(x=vanilla_samples[:, 0], y=vanilla_samples[:, 1], ax=ax2) # 3(a). Fit a diagonal normal autoguide logging.info("\nFitting a DiagNormal autoguide ...") @@ -157,7 +157,7 @@ def main(args): ylim=ylim, title="Posterior \n(DiagNormal autoguide)", ) - sns.kdeplot(guide_samples[:, 0], guide_samples[:, 1], ax=ax3) + sns.kdeplot(x=guide_samples[:, 0], y=guide_samples[:, 1], ax=ax3) # 3(b). Draw samples using NeuTra HMC logging.info("\nDrawing samples using DiagNormal autoguide + NeuTra HMC ...") @@ -165,7 +165,7 @@ def main(args): neutra_model = poutine.reparam(model, config=lambda _: neutra) mcmc = run_hmc(args, neutra_model) zs = mcmc.get_samples()["x_shared_latent"] - sns.scatterplot(zs[:, 0], zs[:, 1], alpha=0.2, ax=ax4) + sns.scatterplot(x=zs[:, 0], y=zs[:, 1], alpha=0.2, ax=ax4) ax4.set( xlabel="x0", ylabel="x1", @@ -182,7 +182,7 @@ def main(args): ylim=ylim, title="Posterior (transformed) \n(DiagNormal + NeuTra HMC)", ) - sns.kdeplot(samples[:, 0], samples[:, 1], ax=ax5) + sns.kdeplot(x=samples[:, 0], y=samples[:, 1], ax=ax5) # 4(a). Fit a BNAF autoguide logging.info("\nFitting a BNAF autoguide ...") @@ -201,7 +201,7 @@ def main(args): ylim=ylim, title="Posterior \n(BNAF autoguide)", ) - sns.kdeplot(guide_samples[:, 0], guide_samples[:, 1], ax=ax6) + sns.kdeplot(x=guide_samples[:, 0], y=guide_samples[:, 1], ax=ax6) # 4(b). Draw samples using NeuTra HMC logging.info("\nDrawing samples using BNAF autoguide + NeuTra HMC ...") @@ -209,7 +209,7 @@ def main(args): neutra_model = poutine.reparam(model, config=lambda _: neutra) mcmc = run_hmc(args, neutra_model) zs = mcmc.get_samples()["x_shared_latent"] - sns.scatterplot(zs[:, 0], zs[:, 1], alpha=0.2, ax=ax7) + sns.scatterplot(x=zs[:, 0], y=zs[:, 1], alpha=0.2, ax=ax7) ax7.set( xlabel="x0", ylabel="x1", @@ -226,13 +226,13 @@ def main(args): ylim=ylim, title="Posterior (transformed) \n(BNAF + NeuTra HMC)", ) - sns.kdeplot(samples[:, 0], samples[:, 1], ax=ax8) + sns.kdeplot(x=samples[:, 0], y=samples[:, 1], ax=ax8) plt.savefig(os.path.join(os.path.dirname(__file__), "neutra.pdf")) if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") parser = argparse.ArgumentParser( description="Example illustrating NeuTra Reparametrizer" ) diff --git a/examples/rsa/generics.py b/examples/rsa/generics.py index 7a1fd7da85..c443d01ac5 100644 --- a/examples/rsa/generics.py +++ b/examples/rsa/generics.py @@ -177,7 +177,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") 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 c1270e0ac1..859e135929 100644 --- a/examples/rsa/hyperbole.py +++ b/examples/rsa/hyperbole.py @@ -217,7 +217,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") 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 de3877ad74..37a44fc8e3 100644 --- a/examples/rsa/schelling.py +++ b/examples/rsa/schelling.py @@ -79,7 +79,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") 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 cdab585bc2..e43c6b1b23 100644 --- a/examples/rsa/schelling_false.py +++ b/examples/rsa/schelling_false.py @@ -96,7 +96,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") 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 092380349e..4c4e7dfadb 100644 --- a/examples/rsa/semantic_parsing.py +++ b/examples/rsa/semantic_parsing.py @@ -351,7 +351,7 @@ def is_all_qud(world): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") parser = argparse.ArgumentParser(description="parse args") parser.add_argument("-n", "--num-samples", default=10, type=int) args = parser.parse_args() diff --git a/examples/scanvi/scanvi.py b/examples/scanvi/scanvi.py index a0b57b75bb..6679f49d27 100644 --- a/examples/scanvi/scanvi.py +++ b/examples/scanvi/scanvi.py @@ -407,7 +407,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") # Parse command line arguments parser = argparse.ArgumentParser( description="single-cell ANnotation using Variational Inference" diff --git a/examples/sir_hmc.py b/examples/sir_hmc.py index 690ab142fb..11851ec784 100644 --- a/examples/sir_hmc.py +++ b/examples/sir_hmc.py @@ -633,7 +633,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") parser = argparse.ArgumentParser(description="SIR epidemiology modeling using HMC") parser.add_argument("-p", "--population", default=10, type=int) parser.add_argument("-m", "--min-observations", default=3, type=int) diff --git a/examples/sparse_gamma_def.py b/examples/sparse_gamma_def.py index 2e7163f10d..dd9c935e3e 100644 --- a/examples/sparse_gamma_def.py +++ b/examples/sparse_gamma_def.py @@ -269,7 +269,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") # parse command line arguments parser = argparse.ArgumentParser(description="parse args") parser.add_argument( diff --git a/examples/sparse_regression.py b/examples/sparse_regression.py index 09988feff8..0d5d2b0094 100644 --- a/examples/sparse_regression.py +++ b/examples/sparse_regression.py @@ -364,7 +364,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") 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/svi_horovod.py b/examples/svi_horovod.py index 393d944f15..5a05955a7d 100644 --- a/examples/svi_horovod.py +++ b/examples/svi_horovod.py @@ -154,7 +154,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") parser = argparse.ArgumentParser(description="Distributed training via Horovod") parser.add_argument("-o", "--outfile") parser.add_argument("-s", "--size", default=1000000, type=int) diff --git a/examples/toy_mixture_model_discrete_enumeration.py b/examples/toy_mixture_model_discrete_enumeration.py index 656803663a..26ac1452fb 100644 --- a/examples/toy_mixture_model_discrete_enumeration.py +++ b/examples/toy_mixture_model_discrete_enumeration.py @@ -133,7 +133,7 @@ def get_true_pred_CPDs(CPD, posterior_param): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") parser = argparse.ArgumentParser(description="Toy mixture model") parser.add_argument("-n", "--num-steps", default=4000, type=int) parser.add_argument("-o", "--num-obs", default=10000, type=int) diff --git a/examples/vae/ss_vae_M2.py b/examples/vae/ss_vae_M2.py index 7fdf1f5ae2..82ba77fab4 100644 --- a/examples/vae/ss_vae_M2.py +++ b/examples/vae/ss_vae_M2.py @@ -433,7 +433,7 @@ def main(args): ) if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") parser = argparse.ArgumentParser(description="SS-VAE\n{}".format(EXAMPLE_RUN)) diff --git a/examples/vae/vae.py b/examples/vae/vae.py index d1e9410348..fce421b113 100644 --- a/examples/vae/vae.py +++ b/examples/vae/vae.py @@ -216,7 +216,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") # parse command line arguments parser = argparse.ArgumentParser(description="parse args") parser.add_argument( diff --git a/examples/vae/vae_comparison.py b/examples/vae/vae_comparison.py index c804e0acb7..6c0333b70e 100644 --- a/examples/vae/vae_comparison.py +++ b/examples/vae/vae_comparison.py @@ -262,7 +262,7 @@ def main(args): if __name__ == "__main__": - assert pyro.__version__.startswith("1.8.1") + assert pyro.__version__.startswith("1.8.2") 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 3d1a9fb9d3..c8c07e6eb2 100644 --- a/pyro/__init__.py +++ b/pyro/__init__.py @@ -26,7 +26,7 @@ from pyro.util import set_rng_seed # After changing this, run scripts/update_version.py -version_prefix = "1.8.1" +version_prefix = "1.8.2" # Get the __version__ string from the auto-generated _version.py file, if exists. try: diff --git a/setup.py b/setup.py index 7b44fa7522..990d7f9402 100644 --- a/setup.py +++ b/setup.py @@ -73,7 +73,7 @@ "pandas", "pillow==8.2.0", # https://github.com/pytorch/pytorch/issues/61125 "scikit-learn", - "seaborn", + "seaborn>=0.11.0", "wget", "lap", # 'biopython>=1.54', diff --git a/tutorial/source/air.ipynb b/tutorial/source/air.ipynb index 7432ac1f29..ad5a613d9b 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.8.1')" + "assert pyro.__version__.startswith('1.8.2')" ] }, { diff --git a/tutorial/source/bayesian_regression.ipynb b/tutorial/source/bayesian_regression.ipynb index b8c7ee2195..2a7140e34a 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.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "pyro.set_rng_seed(1)\n", "\n", "\n", @@ -126,14 +126,14 @@ "fig, ax = plt.subplots(nrows=1, ncols=2, figsize=(12, 6), sharey=True)\n", "african_nations = df[df[\"cont_africa\"] == 1]\n", "non_african_nations = df[df[\"cont_africa\"] == 0]\n", - "sns.scatterplot(non_african_nations[\"rugged\"], \n", - " non_african_nations[\"rgdppc_2000\"], \n", - " ax=ax[0])\n", + "sns.scatterplot(x=non_african_nations[\"rugged\"],\n", + " y=non_african_nations[\"rgdppc_2000\"],\n", + " ax=ax[0])\n", "ax[0].set(xlabel=\"Terrain Ruggedness Index\",\n", " ylabel=\"log GDP (2000)\",\n", " title=\"Non African Nations\")\n", - "sns.scatterplot(african_nations[\"rugged\"], \n", - " african_nations[\"rgdppc_2000\"], \n", + "sns.scatterplot(x=african_nations[\"rugged\"],\n", + " y=african_nations[\"rgdppc_2000\"],\n", " ax=ax[1])\n", "ax[1].set(xlabel=\"Terrain Ruggedness Index\",\n", " ylabel=\"log GDP (2000)\",\n", diff --git a/tutorial/source/bayesian_regression_ii.ipynb b/tutorial/source/bayesian_regression_ii.ipynb index ab42a317b6..02827db6b3 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.8.1')" + "assert pyro.__version__.startswith('1.8.2')" ] }, { @@ -380,11 +380,11 @@ "source": [ "fig, axs = plt.subplots(nrows=1, ncols=2, figsize=(12, 6))\n", "fig.suptitle(\"Cross-section of the Posterior Distribution\", fontsize=16)\n", - "sns.kdeplot(hmc_samples[\"bA\"], hmc_samples[\"bR\"], ax=axs[0], shade=True, label=\"HMC\")\n", - "sns.kdeplot(svi_samples[\"bA\"], svi_samples[\"bR\"], ax=axs[0], label=\"SVI (DiagNormal)\")\n", + "sns.kdeplot(x=hmc_samples[\"bA\"], y=hmc_samples[\"bR\"], ax=axs[0], shade=True, label=\"HMC\")\n", + "sns.kdeplot(x=svi_samples[\"bA\"], y=svi_samples[\"bR\"], ax=axs[0], label=\"SVI (DiagNormal)\")\n", "axs[0].set(xlabel=\"bA\", ylabel=\"bR\", xlim=(-2.5, -1.2), ylim=(-0.5, 0.1))\n", - "sns.kdeplot(hmc_samples[\"bR\"], hmc_samples[\"bAR\"], ax=axs[1], shade=True, label=\"HMC\")\n", - "sns.kdeplot(svi_samples[\"bR\"], svi_samples[\"bAR\"], ax=axs[1], label=\"SVI (DiagNormal)\")\n", + "sns.kdeplot(x=hmc_samples[\"bR\"], y=hmc_samples[\"bAR\"], ax=axs[1], shade=True, label=\"HMC\")\n", + "sns.kdeplot(x=svi_samples[\"bR\"], y=svi_samples[\"bAR\"], ax=axs[1], label=\"SVI (DiagNormal)\")\n", "axs[1].set(xlabel=\"bR\", ylabel=\"bAR\", xlim=(-0.45, 0.05), ylim=(-0.15, 0.8))\n", "handles, labels = axs[1].get_legend_handles_labels()\n", "fig.legend(handles, labels, loc='upper right');" @@ -505,11 +505,11 @@ "source": [ "fig, axs = plt.subplots(nrows=1, ncols=2, figsize=(12, 6))\n", "fig.suptitle(\"Cross-sections of the Posterior Distribution\", fontsize=16)\n", - "sns.kdeplot(svi_samples[\"bA\"], svi_samples[\"bR\"], ax=axs[0], label=\"SVI (Diagonal Normal)\")\n", - "sns.kdeplot(svi_mvn_samples[\"bA\"], svi_mvn_samples[\"bR\"], ax=axs[0], shade=True, label=\"SVI (Multivariate Normal)\")\n", + "sns.kdeplot(x=svi_samples[\"bA\"], y=svi_samples[\"bR\"], ax=axs[0], label=\"SVI (Diagonal Normal)\")\n", + "sns.kdeplot(x=svi_mvn_samples[\"bA\"], y=svi_mvn_samples[\"bR\"], ax=axs[0], shade=True, label=\"SVI (Multivariate Normal)\")\n", "axs[0].set(xlabel=\"bA\", ylabel=\"bR\", xlim=(-2.5, -1.2), ylim=(-0.5, 0.1))\n", - "sns.kdeplot(svi_samples[\"bR\"], svi_samples[\"bAR\"], ax=axs[1], label=\"SVI (Diagonal Normal)\")\n", - "sns.kdeplot(svi_mvn_samples[\"bR\"], svi_mvn_samples[\"bAR\"], ax=axs[1], shade=True, label=\"SVI (Multivariate Normal)\")\n", + "sns.kdeplot(x=svi_samples[\"bR\"], y=svi_samples[\"bAR\"], ax=axs[1], label=\"SVI (Diagonal Normal)\")\n", + "sns.kdeplot(x=svi_mvn_samples[\"bR\"], y=svi_mvn_samples[\"bAR\"], ax=axs[1], shade=True, label=\"SVI (Multivariate Normal)\")\n", "axs[1].set(xlabel=\"bR\", ylabel=\"bAR\", xlim=(-0.45, 0.05), ylim=(-0.15, 0.8))\n", "handles, labels = axs[1].get_legend_handles_labels()\n", "fig.legend(handles, labels, loc='upper right');" @@ -541,11 +541,11 @@ "source": [ "fig, axs = plt.subplots(nrows=1, ncols=2, figsize=(12, 6))\n", "fig.suptitle(\"Cross-sections of the Posterior Distribution\", fontsize=16)\n", - "sns.kdeplot(hmc_samples[\"bA\"], hmc_samples[\"bR\"], ax=axs[0], shade=True, label=\"HMC\")\n", - "sns.kdeplot(svi_mvn_samples[\"bA\"], svi_mvn_samples[\"bR\"], ax=axs[0], label=\"SVI (Multivariate Normal)\")\n", + "sns.kdeplot(x=hmc_samples[\"bA\"], y=hmc_samples[\"bR\"], ax=axs[0], shade=True, label=\"HMC\")\n", + "sns.kdeplot(x=svi_mvn_samples[\"bA\"], y=svi_mvn_samples[\"bR\"], ax=axs[0], label=\"SVI (Multivariate Normal)\")\n", "axs[0].set(xlabel=\"bA\", ylabel=\"bR\", xlim=(-2.5, -1.2), ylim=(-0.5, 0.1))\n", - "sns.kdeplot(hmc_samples[\"bR\"], hmc_samples[\"bAR\"], ax=axs[1], shade=True, label=\"HMC\")\n", - "sns.kdeplot(svi_mvn_samples[\"bR\"], svi_mvn_samples[\"bAR\"], ax=axs[1], label=\"SVI (Multivariate Normal)\")\n", + "sns.kdeplot(x=hmc_samples[\"bR\"], y=hmc_samples[\"bAR\"], ax=axs[1], shade=True, label=\"HMC\")\n", + "sns.kdeplot(x=svi_mvn_samples[\"bR\"], y=svi_mvn_samples[\"bAR\"], ax=axs[1], label=\"SVI (Multivariate Normal)\")\n", "axs[1].set(xlabel=\"bR\", ylabel=\"bAR\", xlim=(-0.45, 0.05), ylim=(-0.15, 0.8))\n", "handles, labels = axs[1].get_legend_handles_labels()\n", "fig.legend(handles, labels, loc='upper right');" diff --git a/tutorial/source/bo.ipynb b/tutorial/source/bo.ipynb index 52dcedd73d..0db806931b 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.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "pyro.set_rng_seed(1)" ] }, diff --git a/tutorial/source/dirichlet_process_mixture.ipynb b/tutorial/source/dirichlet_process_mixture.ipynb index a3e8f3b19c..3266dc9d6e 100644 --- a/tutorial/source/dirichlet_process_mixture.ipynb +++ b/tutorial/source/dirichlet_process_mixture.ipynb @@ -76,7 +76,7 @@ "from pyro.infer import Predictive, SVI, Trace_ELBO\n", "from pyro.optim import Adam\n", "\n", - "assert pyro.__version__.startswith('1.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "pyro.set_rng_seed(0)" ] }, diff --git a/tutorial/source/easyguide.ipynb b/tutorial/source/easyguide.ipynb index 999d2de174..9661244484 100644 --- a/tutorial/source/easyguide.ipynb +++ b/tutorial/source/easyguide.ipynb @@ -44,7 +44,7 @@ "from torch.distributions import constraints\n", "\n", "smoke_test = ('CI' in os.environ)\n", - "assert pyro.__version__.startswith('1.8.1')" + "assert pyro.__version__.startswith('1.8.2')" ] }, { diff --git a/tutorial/source/ekf.ipynb b/tutorial/source/ekf.ipynb index 5ece56dfe8..ce44e2e75b 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.8.1')" + "assert pyro.__version__.startswith('1.8.2')" ] }, { diff --git a/tutorial/source/enumeration.ipynb b/tutorial/source/enumeration.ipynb index e4f1dcc7fc..58a033ecdc 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.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "pyro.set_rng_seed(0)" ] }, diff --git a/tutorial/source/epi_intro.ipynb b/tutorial/source/epi_intro.ipynb index 44a2c3f3b2..b5c51db19d 100644 --- a/tutorial/source/epi_intro.ipynb +++ b/tutorial/source/epi_intro.ipynb @@ -58,7 +58,7 @@ "from pyro.contrib.epidemiology import CompartmentalModel, binomial_dist, infection_dist\n", "\n", "%matplotlib inline\n", - "assert pyro.__version__.startswith('1.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "torch.set_default_dtype(torch.double) # Required for MCMC inference.\n", "smoke_test = ('CI' in os.environ)" ] diff --git a/tutorial/source/forecasting_dlm.ipynb b/tutorial/source/forecasting_dlm.ipynb index 17c947dbef..68f7388c15 100644 --- a/tutorial/source/forecasting_dlm.ipynb +++ b/tutorial/source/forecasting_dlm.ipynb @@ -46,7 +46,7 @@ "from pyro.ops.stats import quantile\n", "\n", "%matplotlib inline\n", - "assert pyro.__version__.startswith('1.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "\n", "pyro.set_rng_seed(20200928)\n", "\n", diff --git a/tutorial/source/forecasting_i.ipynb b/tutorial/source/forecasting_i.ipynb index e62b53fb14..c55689aba6 100644 --- a/tutorial/source/forecasting_i.ipynb +++ b/tutorial/source/forecasting_i.ipynb @@ -47,7 +47,7 @@ "import matplotlib.pyplot as plt\n", "\n", "%matplotlib inline\n", - "assert pyro.__version__.startswith('1.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "pyro.set_rng_seed(20200221)" ] }, diff --git a/tutorial/source/forecasting_ii.ipynb b/tutorial/source/forecasting_ii.ipynb index f9d8d33cd3..df8592f217 100644 --- a/tutorial/source/forecasting_ii.ipynb +++ b/tutorial/source/forecasting_ii.ipynb @@ -40,7 +40,7 @@ "import matplotlib.pyplot as plt\n", "\n", "%matplotlib inline\n", - "assert pyro.__version__.startswith('1.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "pyro.set_rng_seed(20200305)" ] }, diff --git a/tutorial/source/forecasting_iii.ipynb b/tutorial/source/forecasting_iii.ipynb index 98494c5b77..f218ab4fe0 100644 --- a/tutorial/source/forecasting_iii.ipynb +++ b/tutorial/source/forecasting_iii.ipynb @@ -40,7 +40,7 @@ "import matplotlib.pyplot as plt\n", "\n", "%matplotlib inline\n", - "assert pyro.__version__.startswith('1.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "pyro.set_rng_seed(20200305)" ] }, diff --git a/tutorial/source/gmm.ipynb b/tutorial/source/gmm.ipynb index 4b197ea9a8..301ea4c719 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.8.1')" + "assert pyro.__version__.startswith('1.8.2')" ] }, { diff --git a/tutorial/source/gp.ipynb b/tutorial/source/gp.ipynb index 687156a189..5c1b61ac1e 100644 --- a/tutorial/source/gp.ipynb +++ b/tutorial/source/gp.ipynb @@ -69,7 +69,7 @@ "\n", "\n", "smoke_test = \"CI\" in os.environ # ignore; used to check code integrity in the Pyro repo\n", - "assert pyro.__version__.startswith('1.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "pyro.set_rng_seed(0)" ] }, diff --git a/tutorial/source/gplvm.ipynb b/tutorial/source/gplvm.ipynb index 1a42a5be39..fd3a6e14d3 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.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "pyro.set_rng_seed(1)" ] }, diff --git a/tutorial/source/intro_long.ipynb b/tutorial/source/intro_long.ipynb index 022941b331..1c842a5f48 100644 --- a/tutorial/source/intro_long.ipynb +++ b/tutorial/source/intro_long.ipynb @@ -108,7 +108,7 @@ "outputs": [], "source": [ "smoke_test = ('CI' in os.environ)\n", - "assert pyro.__version__.startswith('1.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "\n", "pyro.enable_validation(True)\n", "pyro.set_rng_seed(1)\n", @@ -1835,12 +1835,12 @@ "\n", "fig, axs = plt.subplots(nrows=1, ncols=2, figsize=(12, 6))\n", "fig.suptitle(\"Cross-sections of the Posterior Distribution\", fontsize=16)\n", - "sns.kdeplot(svi_samples[\"bA\"], y=svi_samples[\"bR\"], ax=axs[0], bw_adjust=4 )\n", - "sns.kdeplot(svi_mvn_samples[\"bA\"], y=svi_mvn_samples[\"bR\"], ax=axs[0], shade=True, bw_adjust=4)\n", + "sns.kdeplot(x=svi_samples[\"bA\"], y=svi_samples[\"bR\"], ax=axs[0], bw_adjust=4 )\n", + "sns.kdeplot(x=svi_mvn_samples[\"bA\"], y=svi_mvn_samples[\"bR\"], ax=axs[0], shade=True, bw_adjust=4)\n", "axs[0].set(xlabel=\"bA\", ylabel=\"bR\", xlim=(-2.8, -0.9), ylim=(-0.6, 0.2))\n", "\n", - "sns.kdeplot(svi_samples[\"bR\"], y=svi_samples[\"bAR\"], ax=axs[1],bw_adjust=4 )\n", - "sns.kdeplot(svi_mvn_samples[\"bR\"], y=svi_mvn_samples[\"bAR\"], ax=axs[1], shade=True, bw_adjust=4)\n", + "sns.kdeplot(x=svi_samples[\"bR\"], y=svi_samples[\"bAR\"], ax=axs[1],bw_adjust=4 )\n", + "sns.kdeplot(x=svi_mvn_samples[\"bR\"], y=svi_mvn_samples[\"bAR\"], ax=axs[1], shade=True, bw_adjust=4)\n", "axs[1].set(xlabel=\"bR\", ylabel=\"bAR\", xlim=(-0.55, 0.2), ylim=(-0.15, 0.85))\n", "\n", "\n", diff --git a/tutorial/source/jit.ipynb b/tutorial/source/jit.ipynb index 1643eba442..4404b82cbb 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.8.1')" + "assert pyro.__version__.startswith('1.8.2')" ] }, { diff --git a/tutorial/source/model_rendering.ipynb b/tutorial/source/model_rendering.ipynb index 861a947469..f34d83b4b7 100644 --- a/tutorial/source/model_rendering.ipynb +++ b/tutorial/source/model_rendering.ipynb @@ -25,7 +25,7 @@ "import pyro.distributions.constraints as constraints\n", "\n", "smoke_test = ('CI' in os.environ)\n", - "assert pyro.__version__.startswith('1.8.1')" + "assert pyro.__version__.startswith('1.8.2')" ] }, { diff --git a/tutorial/source/modules.ipynb b/tutorial/source/modules.ipynb index af6bb29081..f52874bac6 100644 --- a/tutorial/source/modules.ipynb +++ b/tutorial/source/modules.ipynb @@ -61,7 +61,7 @@ "from pyro.optim import Adam\n", "\n", "smoke_test = ('CI' in os.environ)\n", - "assert pyro.__version__.startswith('1.8.1')" + "assert pyro.__version__.startswith('1.8.2')" ] }, { diff --git a/tutorial/source/prior_predictive.ipynb b/tutorial/source/prior_predictive.ipynb index 134f127e9a..d01db9cb36 100644 --- a/tutorial/source/prior_predictive.ipynb +++ b/tutorial/source/prior_predictive.ipynb @@ -46,7 +46,7 @@ "import pyro.poutine as poutine\n", "from pyro.infer.resampler import Resampler\n", "\n", - "assert pyro.__version__.startswith('1.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "smoke_test = ('CI' in os.environ) # for CI testing only" ] }, diff --git a/tutorial/source/prodlda.ipynb b/tutorial/source/prodlda.ipynb index 5856315b01..868c9fa99d 100644 --- a/tutorial/source/prodlda.ipynb +++ b/tutorial/source/prodlda.ipynb @@ -70,7 +70,7 @@ "from pyro.infer import MCMC, NUTS\n", "import torch\n", "\n", - "assert pyro.__version__.startswith('1.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "# Enable smoke test - run the notebook cells on CI.\n", "smoke_test = 'CI' in os.environ" ] diff --git a/tutorial/source/stable.ipynb b/tutorial/source/stable.ipynb index f2f47fc6c5..6abe840bff 100644 --- a/tutorial/source/stable.ipynb +++ b/tutorial/source/stable.ipynb @@ -62,7 +62,7 @@ "from pyro.ops.tensor_utils import convolve\n", "\n", "%matplotlib inline\n", - "assert pyro.__version__.startswith('1.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "smoke_test = ('CI' in os.environ)" ] }, diff --git a/tutorial/source/svi_part_i.ipynb b/tutorial/source/svi_part_i.ipynb index ad458124d7..4fdd533fbe 100644 --- a/tutorial/source/svi_part_i.ipynb +++ b/tutorial/source/svi_part_i.ipynb @@ -260,7 +260,7 @@ "smoke_test = ('CI' in os.environ)\n", "n_steps = 2 if smoke_test else 2000\n", "\n", - "assert pyro.__version__.startswith('1.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "\n", "# clear the param store in case we're in a REPL\n", "pyro.clear_param_store()\n", diff --git a/tutorial/source/svi_part_iii.ipynb b/tutorial/source/svi_part_iii.ipynb index 9d96af9259..b78b464e58 100644 --- a/tutorial/source/svi_part_iii.ipynb +++ b/tutorial/source/svi_part_iii.ipynb @@ -283,7 +283,7 @@ "from pyro.infer import SVI, TraceGraph_ELBO\n", "import sys\n", "\n", - "assert pyro.__version__.startswith('1.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "\n", "# this is for running the notebook in our testing framework\n", "smoke_test = ('CI' in os.environ)\n", diff --git a/tutorial/source/tensor_shapes.ipynb b/tutorial/source/tensor_shapes.ipynb index 3ec5479c37..d5cbfe8f3f 100644 --- a/tutorial/source/tensor_shapes.ipynb +++ b/tutorial/source/tensor_shapes.ipynb @@ -59,7 +59,7 @@ "from pyro.optim import Adam\n", "\n", "smoke_test = ('CI' in os.environ)\n", - "assert pyro.__version__.startswith('1.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "\n", "# We'll ue this helper to check our models are correct.\n", "def test_model(model, guide, loss):\n", diff --git a/tutorial/source/tracking_1d.ipynb b/tutorial/source/tracking_1d.ipynb index 8789aeb1b4..ec11f0a4ef 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.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "smoke_test = ('CI' in os.environ)" ] }, diff --git a/tutorial/source/vae.ipynb b/tutorial/source/vae.ipynb index 2aabb77748..d3d570ab28 100644 --- a/tutorial/source/vae.ipynb +++ b/tutorial/source/vae.ipynb @@ -115,7 +115,7 @@ "metadata": {}, "outputs": [], "source": [ - "assert pyro.__version__.startswith('1.8.1')\n", + "assert pyro.__version__.startswith('1.8.2')\n", "pyro.distributions.enable_validation(False)\n", "pyro.set_rng_seed(0)\n", "# Enable smoke test - run the notebook cells on CI.\n",