Bump to version 1.5.1 (#2698)

* Bump to version 1.5.1

* Bump torch version in docs, add instructions

docs/source/conf.py

@@ -213,5 +213,5 @@ def setup(app):
 # @jpchen's hack to get rtd builder to install latest pytorch
 # See similar line in the install section of .travis.yml
 if 'READTHEDOCS' in os.environ:
-    os.system('pip install torch==1.5.0+cpu torchvision==0.6.0+cpu '
+    os.system('pip install torch==1.6.0+cpu torchvision==0.7.0+cpu '
               '-f https://download.pytorch.org/whl/torch_stable.html')

examples/air/main.py

@@ -248,7 +248,7 @@ def per_param_optim_args(module_name, param_name):
 
 
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description="Pyro AIR example", argument_default=argparse.SUPPRESS)
     parser.add_argument('-n', '--num-steps', type=int, default=int(1e8),
                         help='number of optimization steps to take')

examples/baseball.py

@@ -330,7 +330,7 @@ def main(args):
 
 
 if __name__ == "__main__":
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     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)

examples/contrib/autoname/mixture.py

@@ -75,7 +75,7 @@ def main(args):
 
 
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description="parse args")
     parser.add_argument('-n', '--num-epochs', default=200, type=int)
     parser.add_argument('--jit', action='store_true')

examples/contrib/autoname/scoping_mixture.py

@@ -67,7 +67,7 @@ def main(args):
 
 
 if __name__ == "__main__":
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description="parse args")
     parser.add_argument('-n', '--num-epochs', default=200, type=int)
     args = parser.parse_args()

examples/contrib/autoname/tree_data.py

@@ -105,7 +105,7 @@ def main(args):
 
 
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description="parse args")
     parser.add_argument('-n', '--num-epochs', default=100, type=int)
     args = parser.parse_args()

examples/contrib/cevae/synthetic.py

@@ -81,7 +81,7 @@ def main(args):
 
 
 if __name__ == "__main__":
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description="Causal Effect Variational Autoencoder")
     parser.add_argument("--num-data", default=1000, type=int)
     parser.add_argument("--feature-dim", default=5, type=int)

examples/contrib/epidemiology/regional.py

@@ -144,7 +144,7 @@ def main(args):
 
 
 if __name__ == "__main__":
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(
         description="Regional compartmental epidemiology modeling using HMC")
     parser.add_argument("-p", "--population", default=1000, type=int)

examples/contrib/epidemiology/sir.py

@@ -295,7 +295,7 @@ def main(args):
 
 
 if __name__ == "__main__":
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(
         description="Compartmental epidemiology modeling using HMC")
     parser.add_argument("-p", "--population", default=1000, type=float)

examples/contrib/forecast/bart.py

@@ -156,7 +156,7 @@ def transform(pred, truth):
 
 
 if __name__ == "__main__":
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     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)

examples/contrib/funsor/hmm.py

@@ -620,7 +620,7 @@ def main(args):
 
 
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description="MAP Baum-Welch learning Bach Chorales")
     parser.add_argument("-m", "--model", default="1", type=str,
                         help="one of: {}".format(", ".join(sorted(models.keys()))))

examples/contrib/gp/sv-dkl.py

@@ -165,7 +165,7 @@ def main(args):
 
 
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description='Pyro GP MNIST Example')
     parser.add_argument('--data-dir', type=str, default=None, metavar='PATH',
                         help='default directory to cache MNIST data')

examples/contrib/oed/ab_test.py

@@ -115,7 +115,7 @@ def main(num_vi_steps, num_bo_steps, seed):
 
 
 if __name__ == "__main__":
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     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)

examples/contrib/timeseries/gp_models.py

@@ -149,7 +149,7 @@ def main(args):
 
 
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     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)

examples/cvae/main.py

@@ -82,7 +82,7 @@ def main(args):
 
 
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     # parse command line arguments
     parser = argparse.ArgumentParser(description="parse args")
     parser.add_argument('-nq', '--num-quadrant-inputs', metavar='N', type=int,

examples/dmm.py

@@ -453,7 +453,7 @@ def do_evaluation():
 
 # parse command-line arguments and execute the main method
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
 
     parser = argparse.ArgumentParser(description="parse args")
     parser.add_argument('-n', '--num-epochs', type=int, default=5000)

examples/eight_schools/mcmc.py

@@ -42,7 +42,7 @@ def main(args):
 
 
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description='Eight Schools MCMC')
     parser.add_argument('--num-samples', type=int, default=1000,
                         help='number of MCMC samples (default: 1000)')

examples/eight_schools/svi.py

@@ -75,7 +75,7 @@ def main(args):
 
 
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description='Eight Schools SVI')
     parser.add_argument('--lr', type=float, default=0.01,
                         help='learning rate (default: 0.01)')

examples/hmm.py

@@ -638,7 +638,7 @@ def main(args):
 
 
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description="MAP Baum-Welch learning Bach Chorales")
     parser.add_argument("-m", "--model", default="1", type=str,
                         help="one of: {}".format(", ".join(sorted(models.keys()))))

examples/inclined_plane.py

@@ -123,7 +123,7 @@ def main(args):
 
 
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description="parse args")
     parser.add_argument('-n', '--num-samples', default=500, type=int)
     args = parser.parse_args()

examples/lda.py

@@ -138,7 +138,7 @@ def main(args):
 
 
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description="Amortized Latent Dirichlet Allocation")
     parser.add_argument("-t", "--num-topics", default=8, type=int)
     parser.add_argument("-w", "--num-words", default=1024, type=int)

examples/lkj.py

@@ -49,7 +49,7 @@ def main(args):
 
 
 if __name__ == "__main__":
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     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)

examples/minipyro.py

@@ -65,7 +65,7 @@ def guide(data):
 
 
 if __name__ == "__main__":
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description="Mini Pyro demo")
     parser.add_argument("-b", "--backend", default="minipyro")
     parser.add_argument("-n", "--num-steps", default=1001, type=int)

examples/neutra.py

@@ -186,7 +186,7 @@ def main(args):
 
 
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description='Example illustrating NeuTra Reparametrizer')
     parser.add_argument('-n', '--num-steps', default=10000, type=int,
                         help='number of SVI steps')

examples/rsa/generics.py

@@ -157,7 +157,7 @@ def main(args):
 
 
 if __name__ == "__main__":
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description="parse args")
     parser.add_argument('-n', '--num-samples', default=10, type=int)
     args = parser.parse_args()

examples/rsa/hyperbole.py

@@ -154,7 +154,7 @@ def main(args):
 
 
 if __name__ == "__main__":
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description="parse args")
     parser.add_argument('-n', '--num-samples', default=10, type=int)
     parser.add_argument('--price', default=10000, type=int)

examples/rsa/schelling.py

@@ -76,7 +76,7 @@ def main(args):
 
 
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description="parse args")
     parser.add_argument('-n', '--num-samples', default=10, type=int)
     parser.add_argument('--depth', default=2, type=int)

examples/rsa/schelling_false.py

@@ -89,7 +89,7 @@ def main(args):
 
 
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description="parse args")
     parser.add_argument('-n', '--num-samples', default=10, type=int)
     parser.add_argument('--depth', default=3, type=int)

examples/rsa/semantic_parsing.py

@@ -340,7 +340,7 @@ def is_all_qud(world):
 
 
 if __name__ == "__main__":
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description="parse args")
     parser.add_argument('-n', '--num-samples', default=10, type=int)
     args = parser.parse_args()

examples/scanvi/scanvi.py

@@ -351,7 +351,7 @@ def main(args):
 
 
 if __name__ == "__main__":
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     # Parse command line arguments
     parser = argparse.ArgumentParser(description="single-cell ANnotation using Variational Inference")
     parser.add_argument('-s', '--seed', default=0, type=int, help='rng seed')

examples/sir_hmc.py

@@ -573,7 +573,7 @@ def main(args):
 
 
 if __name__ == "__main__":
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     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)

examples/sparse_gamma_def.py

@@ -234,7 +234,7 @@ def main(args):
 
 
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     # parse command line arguments
     parser = argparse.ArgumentParser(description="parse args")
     parser.add_argument('-n', '--num-epochs', default=1500, type=int, help='number of training epochs')

examples/sparse_regression.py

@@ -311,7 +311,7 @@ def main(args):
 
 
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description='Krylov KIT')
     parser.add_argument('--num-data', type=int, default=750)
     parser.add_argument('--num-steps', type=int, default=1000)

examples/svi_horovod.py

@@ -150,7 +150,7 @@ def main(args):
 
 
 if __name__ == "__main__":
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     parser = argparse.ArgumentParser(description="Distributed training via Horovod")
     parser.add_argument("-o", "--outfile")
     parser.add_argument("-s", "--size", default=1000000, type=int)

examples/toy_mixture_model_discrete_enumeration.py

@@ -127,7 +127,7 @@ def get_true_pred_CPDs(CPD, posterior_param):
 
 
 if __name__ == "__main__":
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     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)

examples/vae/ss_vae_M2.py

@@ -383,7 +383,7 @@ def main(args):
               "-sup 3000 -zd 50 -hl 500 -lr 0.00042 -b1 0.95 -bs 200 -log ./tmp.log"
 
 if __name__ == "__main__":
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
 
     parser = argparse.ArgumentParser(description="SS-VAE\n{}".format(EXAMPLE_RUN))
 

examples/vae/vae.py

@@ -201,7 +201,7 @@ def main(args):
 
 
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     # parse command line arguments
     parser = argparse.ArgumentParser(description="parse args")
     parser.add_argument('-n', '--num-epochs', default=101, type=int, help='number of training epochs')

examples/vae/vae_comparison.py

@@ -246,7 +246,7 @@ def main(args):
 
 
 if __name__ == '__main__':
-    assert pyro.__version__.startswith('1.5.0')
+    assert pyro.__version__.startswith('1.5.1')
     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)

pyro/__init__.py

@@ -9,7 +9,8 @@
                              validation_enabled)
 from pyro.util import set_rng_seed
 
-version_prefix = '1.5.0'
+# After changing this, run scripts/update_version.py
+version_prefix = '1.5.1'
 
 # Get the __version__ string from the auto-generated _version.py file, if exists.
 try:

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.5.0')\n",
+    "assert pyro.__version__.startswith('1.5.1')\n",
     "pyro.enable_validation(True)"
    ]
   },

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.5.0')\n",
+    "assert pyro.__version__.startswith('1.5.1')\n",
     "pyro.enable_validation(True)\n",
     "pyro.set_rng_seed(1)\n",
     "pyro.enable_validation(True)\n",

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.5.0')"
+    "assert pyro.__version__.startswith('1.5.1')"
    ]
   },
   {

tutorial/source/bo.ipynb

@@ -54,7 +54,7 @@
     "import pyro\n",
     "import pyro.contrib.gp as gp\n",
     "\n",
-    "assert pyro.__version__.startswith('1.5.0')\n",
+    "assert pyro.__version__.startswith('1.5.1')\n",
     "pyro.enable_validation(True)  # can help with debugging\n",
     "pyro.set_rng_seed(1)"
    ]

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.5.0')\n",
+    "assert pyro.__version__.startswith('1.5.1')\n",
     "pyro.enable_validation(True)       # can help with debugging\n",
     "pyro.set_rng_seed(0)"
    ]