From d5beb1759e04b2a7b7e1409b72dde794d27573ed Mon Sep 17 00:00:00 2001 From: Neeraj Pradhan Date: Thu, 23 Jan 2020 11:24:17 -0800 Subject: [PATCH] Bump to version 0.2.4 (#530) --- examples/baseball.py | 2 +- examples/bnn.py | 2 +- examples/covtype.py | 2 +- examples/funnel.py | 2 +- examples/gp.py | 2 +- examples/hmm.py | 2 +- examples/minipyro.py | 2 +- examples/neutra.py | 2 +- examples/ode.py | 2 +- examples/sparse_regression.py | 2 +- examples/stochastic_volatility.py | 2 +- examples/ucbadmit.py | 2 +- examples/vae.py | 2 +- notebooks/source/bayesian_regression.ipynb | 2 +- notebooks/source/logistic_regression.ipynb | 2 +- notebooks/source/time_series_forecasting.ipynb | 2 +- numpyro/version.py | 2 +- setup.py | 6 +++--- 18 files changed, 20 insertions(+), 20 deletions(-) diff --git a/examples/baseball.py b/examples/baseball.py index 079c624aa..f6dc41e69 100644 --- a/examples/baseball.py +++ b/examples/baseball.py @@ -185,7 +185,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith('0.2.3') + assert numpyro.__version__.startswith('0.2.4') parser = argparse.ArgumentParser(description="Baseball batting average using HMC") 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 5a09ede99..7765d74ec 100644 --- a/examples/bnn.py +++ b/examples/bnn.py @@ -136,7 +136,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith('0.2.3') + assert numpyro.__version__.startswith('0.2.4') 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/covtype.py b/examples/covtype.py index 9b3ebbc17..0e13246e2 100644 --- a/examples/covtype.py +++ b/examples/covtype.py @@ -60,7 +60,7 @@ def main(args): if __name__ == '__main__': - assert numpyro.__version__.startswith('0.2.3') + assert numpyro.__version__.startswith('0.2.4') parser = argparse.ArgumentParser(description="parse args") parser.add_argument('-n', '--num-samples', default=100, type=int, help='number of samples') parser.add_argument('--num-steps', default=10, type=int, help='number of steps (for "HMC")') diff --git a/examples/funnel.py b/examples/funnel.py index 43e30e239..fc5309b13 100644 --- a/examples/funnel.py +++ b/examples/funnel.py @@ -87,7 +87,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith('0.2.3') + assert numpyro.__version__.startswith('0.2.4') parser = argparse.ArgumentParser(description="Non-centered reparameterization 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/gp.py b/examples/gp.py index 9e4541571..2a4571552 100644 --- a/examples/gp.py +++ b/examples/gp.py @@ -129,7 +129,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith('0.2.3') + assert numpyro.__version__.startswith('0.2.4') 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/hmm.py b/examples/hmm.py index e331c3ee0..47576ce31 100644 --- a/examples/hmm.py +++ b/examples/hmm.py @@ -184,7 +184,7 @@ def main(args): if __name__ == '__main__': - assert numpyro.__version__.startswith('0.2.3') + assert numpyro.__version__.startswith('0.2.4') 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/minipyro.py b/examples/minipyro.py index e30df4418..0784e30d2 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.2.3') + assert numpyro.__version__.startswith('0.2.4') 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/neutra.py b/examples/neutra.py index eac5806cb..ccae859d6 100644 --- a/examples/neutra.py +++ b/examples/neutra.py @@ -154,7 +154,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith('0.2.3') + assert numpyro.__version__.startswith('0.2.4') 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 3c15f8690..339dd1505 100644 --- a/examples/ode.py +++ b/examples/ode.py @@ -100,7 +100,7 @@ def main(args): if __name__ == '__main__': - assert numpyro.__version__.startswith('0.2.3') + assert numpyro.__version__.startswith('0.2.4') 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/sparse_regression.py b/examples/sparse_regression.py index 43d01929f..7a4c102c9 100644 --- a/examples/sparse_regression.py +++ b/examples/sparse_regression.py @@ -322,7 +322,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith('0.2.3') + assert numpyro.__version__.startswith('0.2.4') 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/stochastic_volatility.py b/examples/stochastic_volatility.py index 4dd9e698f..e4744bcda 100644 --- a/examples/stochastic_volatility.py +++ b/examples/stochastic_volatility.py @@ -111,7 +111,7 @@ def main(args): if __name__ == "__main__": - assert numpyro.__version__.startswith('0.2.3') + assert numpyro.__version__.startswith('0.2.4') 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/ucbadmit.py b/examples/ucbadmit.py index 1e11160c2..4d5a4707d 100644 --- a/examples/ucbadmit.py +++ b/examples/ucbadmit.py @@ -129,7 +129,7 @@ def main(args): if __name__ == '__main__': - assert numpyro.__version__.startswith('0.2.3') + assert numpyro.__version__.startswith('0.2.4') parser = argparse.ArgumentParser(description='UCBadmit gender discrimination using HMC') parser.add_argument('-n', '--num-samples', nargs='?', default=2000, type=int) parser.add_argument('--num-warmup', nargs='?', default=500, type=int) diff --git a/examples/vae.py b/examples/vae.py index 6d258d265..1fd97da94 100644 --- a/examples/vae.py +++ b/examples/vae.py @@ -131,7 +131,7 @@ def reconstruct_img(epoch, rng_key): if __name__ == '__main__': - assert numpyro.__version__.startswith('0.2.3') + assert numpyro.__version__.startswith('0.2.4') parser = argparse.ArgumentParser(description="parse args") parser.add_argument('-n', '--num-epochs', default=15, type=int, help='number of training epochs') parser.add_argument('-lr', '--learning-rate', default=1.0e-3, type=float, help='learning rate') diff --git a/notebooks/source/bayesian_regression.ipynb b/notebooks/source/bayesian_regression.ipynb index cac3fdb6f..78266f1dd 100644 --- a/notebooks/source/bayesian_regression.ipynb +++ b/notebooks/source/bayesian_regression.ipynb @@ -66,7 +66,7 @@ "if \"NUMPYRO_SPHINXBUILD\" in os.environ:\n", " set_matplotlib_formats('svg')\n", "\n", - "assert numpyro.__version__.startswith('0.2.3')" + "assert numpyro.__version__.startswith('0.2.4')" ] }, { diff --git a/notebooks/source/logistic_regression.ipynb b/notebooks/source/logistic_regression.ipynb index 67640e3d0..650585e6d 100644 --- a/notebooks/source/logistic_regression.ipynb +++ b/notebooks/source/logistic_regression.ipynb @@ -31,7 +31,7 @@ "import numpyro.distributions as dist\n", "from numpyro.examples.datasets import COVTYPE, load_dataset\n", "from numpyro.infer import HMC, MCMC, NUTS\n", - "assert numpyro.__version__.startswith('0.2.3')\n", + "assert numpyro.__version__.startswith('0.2.4')\n", "\n", "# NB: replace gpu by cpu to run this notebook in cpu\n", "numpyro.set_platform(\"gpu\")" diff --git a/notebooks/source/time_series_forecasting.ipynb b/notebooks/source/time_series_forecasting.ipynb index 3ccd28624..a1b1ddd72 100644 --- a/notebooks/source/time_series_forecasting.ipynb +++ b/notebooks/source/time_series_forecasting.ipynb @@ -39,7 +39,7 @@ "if \"NUMPYRO_SPHINXBUILD\" in os.environ:\n", " set_matplotlib_formats('svg')\n", "\n", - "assert numpyro.__version__.startswith('0.2.3')" + "assert numpyro.__version__.startswith('0.2.4')" ] }, { diff --git a/numpyro/version.py b/numpyro/version.py index 5bdc40b7e..4a0b039a3 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.2.3' +__version__ = '0.2.4' diff --git a/setup.py b/setup.py index 99483d046..665e3a1d7 100644 --- a/setup.py +++ b/setup.py @@ -33,9 +33,9 @@ author='Uber AI Labs', author_email='npradhan@uber.com', install_requires=[ - # TODO: pin to a specific version for the release (unless JAX's API becomes stable) - 'jax>=0.1.57', - 'jaxlib>=0.1.37', + # TODO: pin to a specific version for the release (until JAX's API becomes stable) + 'jax==0.1.57', + 'jaxlib==0.1.37', 'tqdm', ], extras_require={