Bump to 0.17.0 (#1969)

fehiepsi · web-flow · commit aa829fb85c79 · 2025-02-06T16:06:11.000-05:00
* Bump to 0.17.0

* add version asserts for various examples
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
diff --git a/notebooks/source/other_samplers.ipynb b/notebooks/source/other_samplers.ipynb
diff --git a/notebooks/source/time_series_forecasting.ipynb b/notebooks/source/time_series_forecasting.ipynb
diff --git a/numpyro/version.py b/numpyro/version.py

Original file line number	Diff line number	Diff line change
`@@ -348,6 +348,7 @@ def main(args):`
`348`	`348`
`349`	`349`
`350`	`350`	`if __name__ == "__main__":`
	`351`	`+ assert numpyro.__version__.startswith("0.17.0")`
`351`	`352`	`parser = argparse.ArgumentParser(`
`352`	`353`	`description="CJS capture-recapture model for ecological data"`
`353`	`354`	`)`
Original file line number	Diff line number	Diff line change
`@@ -139,7 +139,7 @@ def main(args):`
`139`	`139`
`140`	`140`
`141`	`141`	`if __name__ == "__main__":`
`142`		`- assert numpyro.__version__.startswith("0.16.1")`
	`142`	`+ assert numpyro.__version__.startswith("0.17.0")`
`143`	`143`	`parser = argparse.ArgumentParser(`
`144`	`144`	`description="Non-centered reparameterization example"`
`145`	`145`	`)`