Merge pull request #73 from danielward27/version_bump

Version bump
danielward27 · Jan 19, 2023 · 9752074 · 9752074
2 parents e29c80b + d92ed8f
commit 9752074
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/docs/conf.py b/docs/conf.py
@@ -13,7 +13,7 @@
 project = "FlowJax"
 copyright = "2022, Daniel Ward"
 author = "Daniel Ward"
-release = "v7.1.0"
+release = "v8.0.0"
 
 # -- General configuration ---------------------------------------------------
 # https://www.sphinx-doc.org/en/master/usage/configuration.html#general-configuration

diff --git a/flowjax/__init__.py b/flowjax/__init__.py
@@ -1,5 +1,5 @@
 """flowjax - Basic flowjax implementation in jax."""
 
-__version__ = "7.1.0"
+__version__ = "8.0.0"
 __author__ = "Daniel Ward <danielward27@outlook.com>"
 __all__ = []
diff --git a/flowjax/train/variational_fit.py b/flowjax/train/variational_fit.py
@@ -31,8 +31,8 @@ def fit_to_variational_target(
     show_progress: bool = True,
 ):
     """
-    Train a distribution (e.g. a flow) by variational inference.
-    By default this does nothing fancy, just runs SGD steps on the ELBO loss for a fixed number of epochs.
+    Train a distribution (e.g. a flow) by variational inference to a target
+    (e.g. an unnormalized density).
     
     Args:
         key (KeyArray): Jax PRNGKey.

diff --git a/setup.py b/setup.py
@@ -14,7 +14,7 @@ def read(filename):
 
 setup(
     name="flowjax",
-    version="7.1.0",
+    version="8.0.0",
     url="https://github.com/danielward27/flowjax.git",
     license="MIT",
     author="Daniel Ward",