Fix jax.config, remove astropy import

Fix `jax.config`, remove astropy import
lgrcia · May 20, 2024 · bb804fa · bb804fa
2 parents 6a44557 + de1f26a
commit bb804fa
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/nuance/__init__.py b/nuance/__init__.py
@@ -1,7 +1,8 @@
 import jax
-from jax.config import config
 
 DEVICES_COUNT = jax.device_count()
+
+config = jax.config
 config.update("jax_enable_x64", True)
 
 from nuance.combined import CombinedNuance

diff --git a/nuance/star.py b/nuance/star.py
@@ -1,6 +1,5 @@
 from dataclasses import dataclass
 
-import astropy.units as u
 import numpy as np
 
 G = 2942.2062175044193

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "nuance"
-version = "0.6.0"
+version = "0.6.1"
 description = "Transit signals detection among correlated noises"
 authors = ["Lionel Garcia"]
 license = "MIT"