Bump to version 0.2 (#328)

* Bump to version 0.2

* address comments

* add docs

* fix docs

Author: neerajprad

docs/source/distributions.rst

@@ -117,6 +117,14 @@ LogNormal
     :show-inheritance:
     :member-order: bysource
 
+MultivariateNormal
+------------------
+.. autoclass:: numpyro.distributions.continuous.MultivariateNormal
+    :members:
+    :undoc-members:
+    :show-inheritance:
+    :member-order: bysource
+
 Normal
 ------
 .. autoclass:: numpyro.distributions.continuous.Normal
@@ -265,6 +273,13 @@ Poisson
     :show-inheritance:
     :member-order: bysource
 
+PRNGIdentity
+------------
+.. autoclass:: numpyro.distributions.discrete.PRNGIdentity
+    :members:
+    :undoc-members:
+    :show-inheritance:
+    :member-order: bysource
 
 Constraints
 ===========

examples/baseball.py

@@ -197,6 +197,7 @@ def main(args):
 
 
 if __name__ == "__main__":
+    assert numpyro.__version__.startswith('0.2.0')
     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)

examples/bnn.py

@@ -132,6 +132,7 @@ def main(args):
 
 
 if __name__ == "__main__":
+    assert numpyro.__version__.startswith('0.2.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)

examples/covtype.py

@@ -59,6 +59,7 @@ def main(args):
 
 
 if __name__ == '__main__':
+    assert numpyro.__version__.startswith('0.2.0')
     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")')

examples/funnel.py

@@ -81,6 +81,7 @@ def main(args):
 
 
 if __name__ == "__main__":
+    assert numpyro.__version__.startswith('0.2.0')
     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)

examples/gp.py

@@ -123,6 +123,7 @@ def main(args):
 
 
 if __name__ == "__main__":
+    assert numpyro.__version__.startswith('0.2.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)

examples/hmm.py

@@ -161,6 +161,7 @@ def main(args):
 
 
 if __name__ == '__main__':
+    assert numpyro.__version__.startswith('0.2.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)

examples/minipyro.py

@@ -54,6 +54,7 @@ def body_fn(i, val):
 
 
 if __name__ == "__main__":
+    assert numpyro.__version__.startswith('0.2.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)

examples/neutra.py

@@ -149,6 +149,7 @@ def main(args):
 
 
 if __name__ == "__main__":
+    assert numpyro.__version__.startswith('0.2.0')
     parser = argparse.ArgumentParser(description="NeuTra HMC")
     parser.add_argument('-n', '--num-samples', nargs='?', default=20000, type=int)
     parser.add_argument('--num-warmup', nargs='?', default=0, type=int)

examples/sparse_regression.py

@@ -315,6 +315,7 @@ def main(args):
 
 
 if __name__ == "__main__":
+    assert numpyro.__version__.startswith('0.2.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)