Cleanse

jaxutils/__init__.py

@@ -16,9 +16,7 @@
 
 from .pytree import PyTree
 from .data import Dataset, verify_dataset
-from . import _version
 
-__version__ = _version.get_versions()["version"]
 __authors__ = "Thomas Pinder, Daniel Dodd"
 __license__ = "MIT"
 __emails__ = "[email protected], [email protected]"
@@ -35,3 +33,7 @@
     "Dataset",
     "verify_dataset",
 ]
+
+from . import _version
+
+__version__ = _version.get_versions()["version"]

setup.cfg

@@ -3,7 +3,7 @@ VCS = git
 style = pep440
 versionfile_source = jaxutils/_version.py
 versionfile_build = jaxutils/_version.py
-tag_prefix = ''
+tag_prefix = v
 
 [flake8]
 exclude =

setup.py

@@ -22,9 +22,9 @@
 
 
 # Handle builds of nightly release - adapted from BlackJax.
-if "BUILD_JAXUTILS_NIGHTLY" in os.environ:
-    if os.environ["BUILD_JAXUTILS_NIGHTLY"] == "nightly":
-        NAME += "-nightly"
+# if "BUILD_JAXUTILS_NIGHTLY" in os.environ:
+#     if os.environ["BUILD_JAXUTILS_NIGHTLY"] == "nightly":
+#         NAME += "-nightly"
 
 REQUIRES = ["jax>=0.4.0", "jaxlib>=0.4.0", "jaxtyping"]
 
@@ -42,7 +42,7 @@
 
 
 setup(
-    name="JaxUtils",
+    name=NAME,
     version=versioneer.get_version(),
     cmdclass=versioneer.get_cmdclass(),
     author="Daniel Dodd and Thomas Pinder",