add version asserts for various examples

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"
     )

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"
     )

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)

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"
     )

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",

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)

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"
     )

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()

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)

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)