Update setup.py

dfm · web-flow · commit 6d351e58b866 · 2022-05-10T16:49:50.000-04:00
diff --git a/setup.py b/setup.py
@@ -23,7 +23,7 @@
     "Programming Language :: Python",
     "Programming Language :: Python :: 3",
 ]
-INSTALL_REQUIRES = ["tensorflow-probability", "numpyro", "jax<=0.2.20"]
+INSTALL_REQUIRES = ["tensorflow-probability", "numpyro", "jax"]
 EXTRA_REQUIRE = {
     "test": ["pytest>=3.6"],
 }

Original file line number	Diff line number	Diff line change
`@@ -23,7 +23,7 @@`
`23`	`23`	`"Programming Language :: Python",`
`24`	`24`	`"Programming Language :: Python :: 3",`
`25`	`25`	`]`
`26`		`-INSTALL_REQUIRES = ["tensorflow-probability", "numpyro", "jax<=0.2.20"]`
	`26`	`+INSTALL_REQUIRES = ["tensorflow-probability", "numpyro", "jax"]`
`27`	`27`	`EXTRA_REQUIRE = {`
`28`	`28`	`"test": ["pytest>=3.6"],`
`29`	`29`	`}`