Renaming exo4jax -> jaxoplanet (#10)

* renaming

* supporting merge queue

Author: dfm

.github/workflows/tests.yml

@@ -7,6 +7,7 @@ on:
     tags:
       - "*"
   pull_request:
+  merge_group:
 
 jobs:
   tests:

docs/conf.py

@@ -4,7 +4,7 @@
 from pkg_resources import DistributionNotFound, get_distribution
 
 try:
-    __version__ = get_distribution("exo4jax").version
+    __version__ = get_distribution("jaxoplanet").version
 except DistributionNotFound:
     __version__ = "dev"
 
@@ -18,7 +18,7 @@
 ]
 autodoc_mock_imports = []
 
-project = "exo4jax"
+project = "jaxoplanet"
 copyright = "2021 Dan Foreman-Mackey"
 version = __version__
 release = __version__
@@ -28,10 +28,10 @@
 html_theme = "sphinx_book_theme"
 html_title = "exoplanet-jax"
 html_show_sourcelink = False
-html_baseurl = "https://exo4jax.readthedocs.io/en/latest/"
+html_baseurl = "https://jaxoplanet.readthedocs.io/en/latest/"
 html_theme_options = {
     "path_to_docs": "docs",
-    "repository_url": "https://github.com/exoplanet-dev/exo4jax",
+    "repository_url": "https://github.com/exoplanet-dev/jaxoplanet",
     "repository_branch": "main",
     "launch_buttons": {
         "binderhub_url": "https://mybinder.org",

pyproject.toml

@@ -14,13 +14,13 @@ include_trailing_comma = true
 force_grid_wrap = 0
 use_parentheses = true
 combine_as_imports = true
-known_first_party = ["exo4jax"]
+known_first_party = ["jaxoplanet"]
 
 [tool.coverage.run]
 parallel = true
 branch = true
-source = ["exo4jax"]
-omit = ["*_test.py", "*__init__*", "*/exo4jax/exo4jax.py"]
+source = ["jaxoplanet"]
+omit = ["*_test.py", "*__init__*", "*/jaxoplanet/jaxoplanet.py"]
 
 [tool.coverage.paths]
 source = ["src", "*/site-packages"]

setup.py

@@ -8,9 +8,9 @@
 
 # PROJECT SPECIFIC
 
-NAME = "exo4jax"
+NAME = "jaxoplanet"
 PACKAGES = find_packages(where="src")
-META_PATH = os.path.join("src", "exo4jax", "__init__.py")
+META_PATH = os.path.join("src", "jaxoplanet", "__init__.py")
 CLASSIFIERS = [
     "Development Status :: 4 - Beta",
     "Intended Audience :: Developers",
@@ -62,7 +62,9 @@ def find_meta(meta: str, meta_file: str = read(META_PATH)) -> str:
     setup(
         name=NAME,
         use_scm_version={
-            "write_to": os.path.join("src", "exo4jax", "exo4jax_version.py"),
+            "write_to": os.path.join(
+                "src", "jaxoplanet", "jaxoplanet_version.py"
+            ),
             "write_to_template": '__version__ = "{version}"\n',
         },
         author=find_meta("author"),

src/exo4jax/__init__.py

@@ -0,0 +1,10 @@
+# -*- coding: utf-8 -*-
+
+from jaxoplanet import core as core, orbits as orbits
+from jaxoplanet.jaxoplanet_version import __version__ as __version__
+
+__uri__ = "https://github.com/exoplanet-dev/jaxoplanet"
+__author__ = "Dan Foreman-Mackey"
+__email__ = "foreman.mackey@gmail.com"
+__description__ = "Exoplanet, but this time in JAX"
+__license__ = "MIT"

src/exo4jax/_src/experimental/starry/__init__.py

@@ -3,7 +3,7 @@
 import jax
 import jax.numpy as jnp
 
-from exo4jax._src.types import Array
+from jaxoplanet._src.types import Array
 
 
 def kepler(M: Array, ecc: Array) -> Tuple[Array, Array]:

src/exo4jax/core.py

@@ -6,7 +6,7 @@
 import numpy as np
 from scipy.special import binom, roots_legendre
 
-from exo4jax._src.types import Array
+from jaxoplanet._src.types import Array
 
 
 @partial(jax.jit, static_argnames=("order",))

src/exo4jax/light_curves.py

@@ -5,7 +5,7 @@
 import jax.numpy as jnp
 from scipy.special import roots_legendre
 
-from exo4jax._src.types import Array
+from jaxoplanet._src.types import Array
 
 
 @partial(jax.jit, static_argnames=("order",))

src/jaxoplanet/__init__.py

@@ -0,0 +1,3 @@
+from jaxoplanet._src.experimental.starry.solution import (
+    solution_vector as solution_vector,
+)

src/exo4jax/_src/__init__.py renamed to src/jaxoplanet/_src/__init__.py

@@ -6,8 +6,8 @@
 import numpy as np
 from scipy.special import roots_legendre
 
-from exo4jax._src.core.quad import kite_area
-from exo4jax._src.types import Array
+from jaxoplanet._src.core.quad import kite_area
+from jaxoplanet._src.types import Array
 
 
 def solution_vector(

src/exo4jax/_src/core/__init__.py renamed to src/jaxoplanet/_src/core/__init__.py

@@ -4,8 +4,8 @@
 import jax
 import jax.numpy as jnp
 
-from exo4jax._src.core.kepler import kepler
-from exo4jax._src.types import Array, Scalar
+from jaxoplanet._src.core.kepler import kepler
+from jaxoplanet._src.types import Array, Scalar
 
 # FIXME: Switch to constants from astropy
 GRAVITATIONAL_CONSTANT = 2942.2062175044193 / (4 * jnp.pi**2)

src/exo4jax/_src/core/kepler.py renamed to src/jaxoplanet/_src/core/kepler.py

@@ -3,9 +3,9 @@
 
 import jax.numpy as jnp
 
-from exo4jax._src.proto import LightCurveOrbit
-from exo4jax._src.core.quad import light_curve
-from exo4jax._src.types import Array
+from jaxoplanet._src.proto import LightCurveOrbit
+from jaxoplanet._src.core.quad import light_curve
+from jaxoplanet._src.types import Array
 
 
 class QuadLightCurve(NamedTuple):

src/exo4jax/_src/core/limb_dark.py renamed to src/jaxoplanet/_src/core/limb_dark.py

@@ -1,7 +1,7 @@
 from typing import Optional, Tuple
 from typing_extensions import Protocol
 
-from exo4jax._src.types import Array
+from jaxoplanet._src.types import Array
 
 
 class LightCurveBody(Protocol):

src/exo4jax/_src/core/py.typed renamed to src/jaxoplanet/_src/core/py.typed

@@ -3,7 +3,7 @@
 import jax
 import jax.numpy as jnp
 
-from exo4jax._src.types import Array
+from jaxoplanet._src.types import Array
 
 
 class TransitOrbit(NamedTuple):

src/exo4jax/_src/core/quad.py renamed to src/jaxoplanet/_src/core/quad.py

@@ -0,0 +1,2 @@
+from jaxoplanet._src.core.kepler import kepler as kepler
+from jaxoplanet._src.core.quad import light_curve as light_curve

src/exo4jax/_src/experimental/__init__.py renamed to src/jaxoplanet/_src/experimental/__init__.py

@@ -0,0 +1 @@
+from jaxoplanet._src.light_curves import QuadLightCurve as QuadLightCurve

src/jaxoplanet/_src/experimental/starry/__init__.py

@@ -1,6 +1,6 @@
-from exo4jax._src.keplerian import (
+from jaxoplanet._src.keplerian import (
     KeplerianBody as KeplerianBody,
     KeplerianCentral as KeplerianCentral,
     KeplerianOrbit as KeplerianOrbit,
 )
-from exo4jax._src.transit import TransitOrbit as TransitOrbit
+from jaxoplanet._src.transit import TransitOrbit as TransitOrbit

src/exo4jax/_src/experimental/starry/basis.py renamed to src/jaxoplanet/_src/experimental/starry/basis.py

@@ -5,8 +5,8 @@
 import pytest
 from jax.test_util import check_grads
 
-from exo4jax.core import kepler
-from exo4jax.test_utils import assert_allclose
+from jaxoplanet.core import kepler
+from jaxoplanet.test_utils import assert_allclose
 
 
 def get_mean_and_true_anomaly(eccentricity, eccentric_anomaly):

src/exo4jax/_src/experimental/starry/solution.py renamed to src/jaxoplanet/_src/experimental/starry/solution.py

@@ -8,8 +8,8 @@
 import pytest
 from jax.test_util import check_grads
 
-from exo4jax._src.core.limb_dark import light_curve
-from exo4jax.test_utils import assert_allclose
+from jaxoplanet._src.core.limb_dark import light_curve
+from jaxoplanet.test_utils import assert_allclose
 
 
 @pytest.mark.parametrize("r", [0.01, 0.1, 1.1, 2.0])

src/exo4jax/_src/keplerian.py renamed to src/jaxoplanet/_src/keplerian.py

@@ -4,8 +4,8 @@
 import jax.numpy as jnp
 import pytest
 
-from exo4jax._src.core.quad import quad_soln_impl
-from exo4jax.test_utils import assert_allclose
+from jaxoplanet._src.core.quad import quad_soln_impl
+from jaxoplanet.test_utils import assert_allclose
 
 
 @pytest.mark.skipif(

src/exo4jax/_src/light_curves.py renamed to src/jaxoplanet/_src/light_curves.py

@@ -2,7 +2,7 @@
 import pytest
 import warnings
 
-from exo4jax._src.experimental.starry.basis import A1, A2_inv
+from jaxoplanet._src.experimental.starry.basis import A1, A2_inv
 
 
 @pytest.mark.parametrize("lmax", [10, 7, 5, 4, 3, 2, 1, 0])

src/exo4jax/_src/proto.py renamed to src/jaxoplanet/_src/proto.py

@@ -6,8 +6,11 @@
 import numpy as np
 import pytest
 
-from exo4jax.test_utils import assert_allclose
-from exo4jax._src.experimental.starry.solution import kappas, solution_vector
+from jaxoplanet.test_utils import assert_allclose
+from jaxoplanet._src.experimental.starry.solution import (
+    kappas,
+    solution_vector,
+)
 
 
 def test_kappas():

src/exo4jax/_src/py.typed renamed to src/jaxoplanet/_src/py.typed

@@ -1,5 +1,5 @@
 import jax.numpy as jnp
-from exo4jax import orbits
+from jaxoplanet import orbits
 
 
 def test_keplerian_central_shape():

src/exo4jax/_src/transit.py renamed to src/jaxoplanet/_src/transit.py

@@ -1,7 +1,7 @@
 import jax.numpy as jnp
 
-from exo4jax import light_curves, orbits
-from exo4jax.test_utils import assert_allclose
+from jaxoplanet import light_curves, orbits
+from jaxoplanet.test_utils import assert_allclose
 
 
 def test_keplerian_basic():

src/exo4jax/_src/types.py renamed to src/jaxoplanet/_src/types.py

@@ -7,8 +7,8 @@
 import numpy as np
 import pytest
 
-from exo4jax.orbits import KeplerianBody, KeplerianCentral, KeplerianOrbit
-from exo4jax.test_utils import assert_allclose
+from jaxoplanet.orbits import KeplerianBody, KeplerianCentral, KeplerianOrbit
+from jaxoplanet.test_utils import assert_allclose
 
 
 def test_sky_coords():