From aa829fb85c79bc0dee1c897ec82469aa20b555bb Mon Sep 17 00:00:00 2001 From: Du Phan Date: Thu, 6 Feb 2025 16:06:11 -0500 Subject: [PATCH] Bump to 0.17.0 (#1969) * Bump to 0.17.0 * add version asserts for various examples --- examples/annotation.py | 2 +- examples/ar2.py | 2 +- examples/baseball.py | 2 +- examples/bnn.py | 2 +- examples/capture_recapture.py | 1 + examples/covtype.py | 2 +- examples/cvae-flax/main.py | 2 ++ examples/dais_demo.py | 1 + examples/funnel.py | 2 +- examples/gaussian_shells.py | 2 +- examples/gp.py | 2 +- examples/hmcecs.py | 1 + examples/hmm.py | 2 +- examples/hmm_enum.py | 1 + examples/holt_winters.py | 2 +- examples/horseshoe_regression.py | 2 +- examples/hsgp.py | 1 + examples/minipyro.py | 2 +- examples/mortality.py | 2 +- examples/neutra.py | 2 +- examples/ode.py | 2 +- examples/prodlda.py | 2 +- examples/proportion_test.py | 2 +- examples/sparse_regression.py | 2 +- examples/ssbvm_mixture.py | 1 + examples/stein_bnn.py | 2 ++ examples/stochastic_volatility.py | 2 +- examples/thompson_sampling.py | 2 +- examples/toy_mixture_model_discrete_enumeration.py | 2 +- examples/ucbadmit.py | 2 +- examples/vae.py | 2 +- examples/var2.py | 1 + examples/zero_inflated_poisson.py | 1 + notebooks/source/bad_posterior_geometry.ipynb | 2 +- notebooks/source/bayesian_hierarchical_linear_regression.ipynb | 2 +- notebooks/source/bayesian_hierarchical_stacking.ipynb | 2 +- notebooks/source/bayesian_imputation.ipynb | 2 +- notebooks/source/bayesian_regression.ipynb | 2 +- notebooks/source/censoring.ipynb | 2 +- notebooks/source/gmm.ipynb | 2 +- notebooks/source/hsgp_example.ipynb | 2 +- notebooks/source/logistic_regression.ipynb | 2 +- notebooks/source/model_rendering.ipynb | 2 +- notebooks/source/ordinal_regression.ipynb | 2 +- notebooks/source/other_samplers.ipynb | 2 +- notebooks/source/time_series_forecasting.ipynb | 2 +- numpyro/version.py | 2 +- 47 files changed, 49 insertions(+), 37 deletions(-) diff --git a/examples/annotation.py b/examples/annotation.py index 8dfd8cd3d..9be7ebaca 100644 --- a/examples/annotation.py +++ b/examples/annotation.py @@ -320,7 +320,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="Bayesian Models of Annotation") parser.add_argument("-n", "--num-samples", nargs="?", default=1000, type=int) parser.add_argument("--num-warmup", nargs="?", default=1000, type=int) diff --git a/examples/ar2.py b/examples/ar2.py index 9d5bb6022..a78addec0 100644 --- a/examples/ar2.py +++ b/examples/ar2.py @@ -138,7 +138,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="AR2 example") parser.add_argument("--num-data", nargs="?", default=142, type=int) parser.add_argument("-n", "--num-samples", nargs="?", default=1000, type=int) diff --git a/examples/baseball.py b/examples/baseball.py index dc73eb588..5100cc9ea 100644 --- a/examples/baseball.py +++ b/examples/baseball.py @@ -210,7 +210,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="Baseball batting average using MCMC") parser.add_argument("-n", "--num-samples", nargs="?", default=3000, type=int) parser.add_argument("--num-warmup", nargs="?", default=1500, type=int) diff --git a/examples/bnn.py b/examples/bnn.py index 5ce4eb977..fb1f78551 100644 --- a/examples/bnn.py +++ b/examples/bnn.py @@ -160,7 +160,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="Bayesian neural network example") parser.add_argument("-n", "--num-samples", nargs="?", default=2000, type=int) parser.add_argument("--num-warmup", nargs="?", default=1000, type=int) diff --git a/examples/capture_recapture.py b/examples/capture_recapture.py index 87d78e214..1c3757223 100644 --- a/examples/capture_recapture.py +++ b/examples/capture_recapture.py @@ -348,6 +348,7 @@ def main(args): if __name__ == "__main__": + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser( description="CJS capture-recapture model for ecological data" ) diff --git a/examples/covtype.py b/examples/covtype.py index 36607d40a..66751988f 100644 --- a/examples/covtype.py +++ b/examples/covtype.py @@ -206,7 +206,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="parse args") parser.add_argument( "-n", "--num-samples", default=1000, type=int, help="number of samples" diff --git a/examples/cvae-flax/main.py b/examples/cvae-flax/main.py index 95e362e7d..05fcd02e4 100644 --- a/examples/cvae-flax/main.py +++ b/examples/cvae-flax/main.py @@ -8,6 +8,7 @@ from train_baseline import train_baseline from train_cvae import train_cvae +import numpyro from numpyro.examples.datasets import MNIST from models import BaselineNet, Decoder, Encoder, cvae_guide, cvae_model # isort:skip @@ -78,6 +79,7 @@ def main(args): if __name__ == "__main__": + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser( description="Conditional Variational Autoencoder on MNIST using Flax" ) diff --git a/examples/dais_demo.py b/examples/dais_demo.py index b4ac69152..bc218e94a 100644 --- a/examples/dais_demo.py +++ b/examples/dais_demo.py @@ -160,6 +160,7 @@ def add_fig(samples, title, ax): if __name__ == "__main__": + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser("Usage example for AutoDAIS guide.") parser.add_argument("--num-svi-steps", type=int, default=80 * 1000) parser.add_argument("--num-warmup", type=int, default=2000) diff --git a/examples/funnel.py b/examples/funnel.py index 71f3920fa..7a0484b4e 100644 --- a/examples/funnel.py +++ b/examples/funnel.py @@ -139,7 +139,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser( description="Non-centered reparameterization example" ) diff --git a/examples/gaussian_shells.py b/examples/gaussian_shells.py index 6e49a9a8a..3d91f22d4 100644 --- a/examples/gaussian_shells.py +++ b/examples/gaussian_shells.py @@ -120,7 +120,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="Nested sampler for Gaussian shells") parser.add_argument("-n", "--num-samples", nargs="?", default=10000, type=int) diff --git a/examples/gp.py b/examples/gp.py index ad39ade57..6a6414cef 100644 --- a/examples/gp.py +++ b/examples/gp.py @@ -180,7 +180,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="Gaussian Process example") parser.add_argument("-n", "--num-samples", nargs="?", default=1000, type=int) parser.add_argument("--num-warmup", nargs="?", default=1000, type=int) diff --git a/examples/hmcecs.py b/examples/hmcecs.py index 5e0abc696..5d6dfbbd5 100644 --- a/examples/hmcecs.py +++ b/examples/hmcecs.py @@ -139,6 +139,7 @@ def summary_plot(losses, hmc_samples, hmcecs_samples, hmc_runtime, hmcecs_runtim if __name__ == "__main__": + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser( "Hamiltonian Monte Carlo with Energy Conserving Subsampling" ) diff --git a/examples/hmm.py b/examples/hmm.py index f3adc1204..355e8d584 100644 --- a/examples/hmm.py +++ b/examples/hmm.py @@ -263,7 +263,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="Semi-supervised Hidden Markov Model") parser.add_argument("--num-categories", default=3, type=int) parser.add_argument("--num-words", default=10, type=int) diff --git a/examples/hmm_enum.py b/examples/hmm_enum.py index c6108ba37..fd4b3f5c5 100644 --- a/examples/hmm_enum.py +++ b/examples/hmm_enum.py @@ -370,6 +370,7 @@ def main(args): if __name__ == "__main__": + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="HMC for HMMs") parser.add_argument( "-m", diff --git a/examples/holt_winters.py b/examples/holt_winters.py index 653070823..2c3cb3095 100644 --- a/examples/holt_winters.py +++ b/examples/holt_winters.py @@ -180,7 +180,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="Holt-Winters") parser.add_argument("--T", nargs="?", default=6, type=int) parser.add_argument("--future", nargs="?", default=1, type=int) diff --git a/examples/horseshoe_regression.py b/examples/horseshoe_regression.py index b6562d1e6..e04c9030b 100644 --- a/examples/horseshoe_regression.py +++ b/examples/horseshoe_regression.py @@ -162,7 +162,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="Horseshoe regression example") parser.add_argument("-n", "--num-samples", nargs="?", default=2000, type=int) parser.add_argument("--num-warmup", nargs="?", default=1000, type=int) diff --git a/examples/hsgp.py b/examples/hsgp.py index 5b074f773..e26646914 100644 --- a/examples/hsgp.py +++ b/examples/hsgp.py @@ -568,6 +568,7 @@ def main(args): if __name__ == "__main__": + assert numpyro.__version__.startswith("0.17.0") args = parse_arguments() numpyro.enable_x64(args.x64) numpyro.set_platform(args.device) diff --git a/examples/minipyro.py b/examples/minipyro.py index 996cbd7f5..73849f7c6 100644 --- a/examples/minipyro.py +++ b/examples/minipyro.py @@ -58,7 +58,7 @@ def body_fn(i, val): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="Mini Pyro demo") parser.add_argument("-f", "--full-pyro", action="store_true", default=False) parser.add_argument("-n", "--num-steps", default=1001, type=int) diff --git a/examples/mortality.py b/examples/mortality.py index a420c441c..9674e189b 100644 --- a/examples/mortality.py +++ b/examples/mortality.py @@ -220,7 +220,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="Mortality regression model") parser.add_argument("-n", "--num-samples", nargs="?", default=500, type=int) diff --git a/examples/neutra.py b/examples/neutra.py index 6b0ada526..386fef87e 100644 --- a/examples/neutra.py +++ b/examples/neutra.py @@ -197,7 +197,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="NeuTra HMC") parser.add_argument("-n", "--num-samples", nargs="?", default=4000, type=int) parser.add_argument("--num-warmup", nargs="?", default=1000, type=int) diff --git a/examples/ode.py b/examples/ode.py index e50857b4e..75f4b6d40 100644 --- a/examples/ode.py +++ b/examples/ode.py @@ -117,7 +117,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="Predator-Prey Model") parser.add_argument("-n", "--num-samples", nargs="?", default=1000, type=int) parser.add_argument("--num-warmup", nargs="?", default=1000, type=int) diff --git a/examples/prodlda.py b/examples/prodlda.py index e55bb2308..6b1d100c3 100644 --- a/examples/prodlda.py +++ b/examples/prodlda.py @@ -315,7 +315,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser( description="Probabilistic topic modelling with Flax and Haiku" ) diff --git a/examples/proportion_test.py b/examples/proportion_test.py index f0f4dbe6b..d775395d3 100644 --- a/examples/proportion_test.py +++ b/examples/proportion_test.py @@ -158,7 +158,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="Testing whether ") parser.add_argument("-n", "--num-samples", nargs="?", default=500, type=int) parser.add_argument("--num-warmup", nargs="?", default=1500, type=int) diff --git a/examples/sparse_regression.py b/examples/sparse_regression.py index a1d6f6354..21d2f4c20 100644 --- a/examples/sparse_regression.py +++ b/examples/sparse_regression.py @@ -384,7 +384,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="Gaussian Process example") parser.add_argument("-n", "--num-samples", nargs="?", default=1000, type=int) parser.add_argument("--num-warmup", nargs="?", default=500, type=int) diff --git a/examples/ssbvm_mixture.py b/examples/ssbvm_mixture.py index 71b7243f2..de717f483 100644 --- a/examples/ssbvm_mixture.py +++ b/examples/ssbvm_mixture.py @@ -283,6 +283,7 @@ def main(args): if __name__ == "__main__": + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser( description="Sine-skewed sine (bivariate von mises) mixture model example" ) diff --git a/examples/stein_bnn.py b/examples/stein_bnn.py index de724b484..e7296e5bd 100644 --- a/examples/stein_bnn.py +++ b/examples/stein_bnn.py @@ -26,6 +26,7 @@ from jax import config, nn, numpy as jnp, random +import numpyro from numpyro import deterministic, plate, sample, set_platform, subsample from numpyro.contrib.einstein import MixtureGuidePredictive, RBFKernel, SteinVI from numpyro.distributions import Gamma, Normal @@ -186,6 +187,7 @@ def main(args): if __name__ == "__main__": + assert numpyro.__version__.startswith("0.17.0") config.update("jax_debug_nans", True) parser = argparse.ArgumentParser() diff --git a/examples/stochastic_volatility.py b/examples/stochastic_volatility.py index a91c9c3d5..b5ecf16bf 100644 --- a/examples/stochastic_volatility.py +++ b/examples/stochastic_volatility.py @@ -122,7 +122,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="Stochastic Volatility Model") parser.add_argument("-n", "--num-samples", nargs="?", default=600, type=int) parser.add_argument("--num-warmup", nargs="?", default=600, type=int) diff --git a/examples/thompson_sampling.py b/examples/thompson_sampling.py index 3d7bf552a..17cf99a31 100644 --- a/examples/thompson_sampling.py +++ b/examples/thompson_sampling.py @@ -292,7 +292,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="Thompson sampling example") parser.add_argument( "--num-random", nargs="?", default=2, type=int, help="number of random draws" diff --git a/examples/toy_mixture_model_discrete_enumeration.py b/examples/toy_mixture_model_discrete_enumeration.py index 46223d2e1..15413f052 100644 --- a/examples/toy_mixture_model_discrete_enumeration.py +++ b/examples/toy_mixture_model_discrete_enumeration.py @@ -126,7 +126,7 @@ def get_true_pred_CPDs(CPD, posterior_param): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") 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/ucbadmit.py b/examples/ucbadmit.py index 50a712461..7df358638 100644 --- a/examples/ucbadmit.py +++ b/examples/ucbadmit.py @@ -151,7 +151,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser( description="UCBadmit gender discrimination using HMC" ) diff --git a/examples/vae.py b/examples/vae.py index 8bbedabbb..bf5ba4851 100644 --- a/examples/vae.py +++ b/examples/vae.py @@ -160,7 +160,7 @@ def reconstruct_img(epoch, rng_key): if __name__ == "__main__": - assert numpyro.__version__.startswith("0.16.1") + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="parse args") parser.add_argument( "-n", "--num-epochs", default=15, type=int, help="number of training epochs" diff --git a/examples/var2.py b/examples/var2.py index 0e5e9864d..cdbde1fd6 100644 --- a/examples/var2.py +++ b/examples/var2.py @@ -202,6 +202,7 @@ def main(args): if __name__ == "__main__": + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser(description="VAR(2) example") parser.add_argument("--num-data", nargs="?", default=100, type=int) parser.add_argument("-n", "--num-samples", nargs="?", default=1000, type=int) diff --git a/examples/zero_inflated_poisson.py b/examples/zero_inflated_poisson.py index b21656334..9ecfbf39e 100644 --- a/examples/zero_inflated_poisson.py +++ b/examples/zero_inflated_poisson.py @@ -154,6 +154,7 @@ def add_fig(var_name, title, ax): if __name__ == "__main__": + assert numpyro.__version__.startswith("0.17.0") parser = argparse.ArgumentParser("Zero-Inflated Poisson Regression") parser.add_argument("--seed", nargs="?", default=42, type=int) parser.add_argument("-n", "--num-samples", nargs="?", default=2000, type=int) diff --git a/notebooks/source/bad_posterior_geometry.ipynb b/notebooks/source/bad_posterior_geometry.ipynb index af667ae9f..4bd529886 100644 --- a/notebooks/source/bad_posterior_geometry.ipynb +++ b/notebooks/source/bad_posterior_geometry.ipynb @@ -49,7 +49,7 @@ "import numpyro.distributions as dist\n", "from numpyro.infer import MCMC, NUTS\n", "\n", - "assert numpyro.__version__.startswith(\"0.16.1\")\n", + "assert numpyro.__version__.startswith(\"0.17.0\")\n", "\n", "# NB: replace cpu by gpu to run this notebook on gpu\n", "numpyro.set_platform(\"cpu\")" diff --git a/notebooks/source/bayesian_hierarchical_linear_regression.ipynb b/notebooks/source/bayesian_hierarchical_linear_regression.ipynb index 262ff060a..30b6baa10 100644 --- a/notebooks/source/bayesian_hierarchical_linear_regression.ipynb +++ b/notebooks/source/bayesian_hierarchical_linear_regression.ipynb @@ -246,7 +246,7 @@ "import numpyro.distributions as dist\n", "from numpyro.infer import MCMC, NUTS, Predictive\n", "\n", - "assert numpyro.__version__.startswith(\"0.16.1\")" + "assert numpyro.__version__.startswith(\"0.17.0\")" ] }, { diff --git a/notebooks/source/bayesian_hierarchical_stacking.ipynb b/notebooks/source/bayesian_hierarchical_stacking.ipynb index a9bee14f8..5825659ef 100644 --- a/notebooks/source/bayesian_hierarchical_stacking.ipynb +++ b/notebooks/source/bayesian_hierarchical_stacking.ipynb @@ -97,7 +97,7 @@ " set_matplotlib_formats(\"svg\")\n", "\n", "numpyro.set_host_device_count(4)\n", - "assert numpyro.__version__.startswith(\"0.16.1\")" + "assert numpyro.__version__.startswith(\"0.17.0\")" ] }, { diff --git a/notebooks/source/bayesian_imputation.ipynb b/notebooks/source/bayesian_imputation.ipynb index cdce24336..855c652d0 100644 --- a/notebooks/source/bayesian_imputation.ipynb +++ b/notebooks/source/bayesian_imputation.ipynb @@ -52,7 +52,7 @@ "if \"NUMPYRO_SPHINXBUILD\" in os.environ:\n", " set_matplotlib_formats(\"svg\")\n", "\n", - "assert numpyro.__version__.startswith(\"0.16.1\")" + "assert numpyro.__version__.startswith(\"0.17.0\")" ] }, { diff --git a/notebooks/source/bayesian_regression.ipynb b/notebooks/source/bayesian_regression.ipynb index 4f634a9a9..26dccf0c9 100644 --- a/notebooks/source/bayesian_regression.ipynb +++ b/notebooks/source/bayesian_regression.ipynb @@ -91,7 +91,7 @@ "if \"NUMPYRO_SPHINXBUILD\" in os.environ:\n", " set_matplotlib_formats(\"svg\")\n", "\n", - "assert numpyro.__version__.startswith(\"0.16.1\")" + "assert numpyro.__version__.startswith(\"0.17.0\")" ] }, { diff --git a/notebooks/source/censoring.ipynb b/notebooks/source/censoring.ipynb index 0f2827512..8c3a93efe 100644 --- a/notebooks/source/censoring.ipynb +++ b/notebooks/source/censoring.ipynb @@ -60,7 +60,7 @@ "\n", "rng_key = random.PRNGKey(seed=0)\n", "\n", - "assert numpyro.__version__.startswith(\"0.16.1\")\n", + "assert numpyro.__version__.startswith(\"0.17.0\")\n", "\n", "%load_ext autoreload\n", "%autoreload 2\n", diff --git a/notebooks/source/gmm.ipynb b/notebooks/source/gmm.ipynb index 9ca4ec8cd..e78a7c210 100644 --- a/notebooks/source/gmm.ipynb +++ b/notebooks/source/gmm.ipynb @@ -54,7 +54,7 @@ "%matplotlib inline\n", "\n", "smoke_test = \"CI\" in os.environ\n", - "assert numpyro.__version__.startswith(\"0.16.1\")" + "assert numpyro.__version__.startswith(\"0.17.0\")" ] }, { diff --git a/notebooks/source/hsgp_example.ipynb b/notebooks/source/hsgp_example.ipynb index 2f5b07c4e..ffd39efea 100644 --- a/notebooks/source/hsgp_example.ipynb +++ b/notebooks/source/hsgp_example.ipynb @@ -62,7 +62,7 @@ "\n", "rng_key = random.PRNGKey(seed=42)\n", "\n", - "assert numpyro.__version__.startswith(\"0.16.1\")\n", + "assert numpyro.__version__.startswith(\"0.17.0\")\n", "\n", "%load_ext autoreload\n", "%autoreload 2\n", diff --git a/notebooks/source/logistic_regression.ipynb b/notebooks/source/logistic_regression.ipynb index 8127586e6..75a606529 100644 --- a/notebooks/source/logistic_regression.ipynb +++ b/notebooks/source/logistic_regression.ipynb @@ -41,7 +41,7 @@ "from numpyro.examples.datasets import COVTYPE, load_dataset\n", "from numpyro.infer import HMC, MCMC, NUTS\n", "\n", - "assert numpyro.__version__.startswith(\"0.16.1\")\n", + "assert numpyro.__version__.startswith(\"0.17.0\")\n", "\n", "# NB: replace gpu by cpu to run this notebook in cpu\n", "numpyro.set_platform(\"gpu\")" diff --git a/notebooks/source/model_rendering.ipynb b/notebooks/source/model_rendering.ipynb index deaa88f3e..8d69d6712 100644 --- a/notebooks/source/model_rendering.ipynb +++ b/notebooks/source/model_rendering.ipynb @@ -38,7 +38,7 @@ "import numpyro.distributions as dist\n", "import numpyro.distributions.constraints as constraints\n", "\n", - "assert numpyro.__version__.startswith(\"0.16.1\")" + "assert numpyro.__version__.startswith(\"0.17.0\")" ] }, { diff --git a/notebooks/source/ordinal_regression.ipynb b/notebooks/source/ordinal_regression.ipynb index c2f00d181..7dd4f90fb 100644 --- a/notebooks/source/ordinal_regression.ipynb +++ b/notebooks/source/ordinal_regression.ipynb @@ -54,7 +54,7 @@ "from numpyro.infer import MCMC, NUTS\n", "from numpyro.infer.reparam import TransformReparam\n", "\n", - "assert numpyro.__version__.startswith(\"0.16.1\")" + "assert numpyro.__version__.startswith(\"0.17.0\")" ] }, { diff --git a/notebooks/source/other_samplers.ipynb b/notebooks/source/other_samplers.ipynb index 18518c72a..8dac0f357 100644 --- a/notebooks/source/other_samplers.ipynb +++ b/notebooks/source/other_samplers.ipynb @@ -67,7 +67,7 @@ "\n", "rng_key = random.PRNGKey(seed=42)\n", "\n", - "assert numpyro.__version__.startswith(\"0.16.1\")\n", + "assert numpyro.__version__.startswith(\"0.17.0\")\n", "\n", "%load_ext autoreload\n", "%autoreload 2\n", diff --git a/notebooks/source/time_series_forecasting.ipynb b/notebooks/source/time_series_forecasting.ipynb index 90cee1ed7..04e5cc01b 100644 --- a/notebooks/source/time_series_forecasting.ipynb +++ b/notebooks/source/time_series_forecasting.ipynb @@ -48,7 +48,7 @@ " set_matplotlib_formats(\"svg\")\n", "\n", "numpyro.set_host_device_count(4)\n", - "assert numpyro.__version__.startswith(\"0.16.1\")" + "assert numpyro.__version__.startswith(\"0.17.0\")" ] }, { diff --git a/numpyro/version.py b/numpyro/version.py index 8ff8ea390..2f5751530 100644 --- a/numpyro/version.py +++ b/numpyro/version.py @@ -1,4 +1,4 @@ # Copyright Contributors to the Pyro project. # SPDX-License-Identifier: Apache-2.0 -__version__ = "0.16.1" +__version__ = "0.17.0"