jaxspec 0.1.4__py3-none-any.whl → 0.2.1__py3-none-any.whl

jaxspec/model/background.py

@@ -10,7 +10,6 @@ from numpyro.distributions import Poisson
 from tinygp import GaussianProcess, kernels
 
 from .._fit._build_model import build_prior, forward_model
-from ..util.typing import PriorDictModel
 from .abc import SpectralModel
 
 
@@ -138,14 +137,14 @@ class GaussianProcessBackground(BackgroundModel):
         return count_rate
 
 
-class SpectralBackgroundModel(BackgroundModel):
+class SpectralModelBackground(BackgroundModel):
     def __init__(self, spectral_model: "SpectralModel", prior_distributions, sparse=False):
         self.spectral_model = spectral_model
-        self.prior = PriorDictModel.from_dict(prior_distributions).nested_dict
+        self.prior = prior_distributions
         self.sparse = sparse
 
     def numpyro_model(self, observation, name: str = "", observed=True):
-        params = build_prior(self.prior, prefix=f"_bkg_{name}")
+        params = build_prior(self.prior, prefix=f"_bkg_{name}_")
         bkg_model = jax.jit(
             lambda par: forward_model(self.spectral_model, par, observation, sparse=self.sparse)
         )

jaxspec/model/multiplicative.py

@@ -1,16 +1,31 @@
 from __future__ import annotations
 
-import haiku as hk
+import flax.nnx as nnx
 import jax.numpy as jnp
 import numpy as np
 
 from astropy.table import Table
-from haiku.initializers import Constant as HaikuConstant
 
 from ..util.online_storage import table_manager
 from .abc import MultiplicativeComponent
 
 
+class MultiplicativeConstant(MultiplicativeComponent):
+    r"""
+    A multiplicative constant
+
+    !!! abstract "Parameters"
+        * $K$ (`norm`) $\left[\text{dimensionless}\right]$: The multiplicative constant.
+
+    """
+
+    def __init__(self):
+        self.norm = nnx.Param(1.0)
+
+    def factor(self, energy):
+        return self.norm
+
+
 class Expfac(MultiplicativeComponent):
     r"""
     An exponential modification of a spectrum.
@@ -20,19 +35,20 @@ class Expfac(MultiplicativeComponent):
     \text{if $E>E_c$}\\1 & \text{if $E<E_c$}\end{cases}
     $$
 
-    ??? abstract "Parameters"
-        * $A$ : Amplitude of the modification $\left[\text{dimensionless}\right]$
-        * $f$ : Exponential factor $\left[\text{keV}^{-1}\right]$
-        * $E_c$ : Start energy of modification $\left[\text{keV}\right]$
+    !!! abstract "Parameters"
+        * $A$ (`A`) $\left[\text{dimensionless}\right]$ : Amplitude of the modification
+        * $f$ (`f`) $\left[\text{keV}^{-1}\right]$ : Exponential factor
+        * $E_c$ (`E_c`) $\left[\text{keV}\right]$: Start energy of modification
 
     """
 
-    def continuum(self, energy):
-        amplitude = hk.get_parameter("A", [], float, init=HaikuConstant(1))
-        factor = hk.get_parameter("f", [], float, init=HaikuConstant(1))
-        pivot = hk.get_parameter("E_c", [], float, init=HaikuConstant(1))
+    def __init__(self):
+        self.A = nnx.Param(1.0)
+        self.f = nnx.Param(1.0)
+        self.E_c = nnx.Param(1.0)
 
-        return jnp.where(energy >= pivot, 1.0 + amplitude * jnp.exp(-factor * energy), 1.0)
+    def factor(self, energy):
+        return jnp.where(energy >= self.E_c, 1.0 + self.A * jnp.exp(-self.f * energy), 1.0)
 
 
 class Tbabs(MultiplicativeComponent):
@@ -45,49 +61,47 @@ class Tbabs(MultiplicativeComponent):
         \mathcal{M}(E) = \exp^{-N_{\text{H}}\sigma(E)}
     $$
 
-    ??? abstract "Parameters"
-        * $N_{\text{H}}$ : Equivalent hydrogen column density
-            $\left[\frac{\text{atoms}~10^{22}}{\text{cm}^2}\right]$
+    !!! abstract "Parameters"
+        * $N_{\text{H}}$ (`nh`) $\left[\frac{\text{atoms}~10^{22}}{\text{cm}^2}\right]$ : Equivalent hydrogen column density
+
 
     !!! note
         Abundances and cross-sections $\sigma$ can be found in Wilms et al. (2000).
 
     """
 
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
+    def __init__(self):
         table = Table.read(table_manager.fetch("xsect_tbabs_wilm.fits"))
-        self.energy = jnp.asarray(np.array(table["ENERGY"]), dtype=np.float64)
-        self.sigma = jnp.asarray(np.array(table["SIGMA"]), dtype=np.float64)
+        self.energy = nnx.Variable(np.asarray(table["ENERGY"], dtype=np.float64))
+        self.sigma = nnx.Variable(np.asarray(table["SIGMA"], dtype=np.float64))
+        self.nh = nnx.Param(1.0)
 
-    def continuum(self, energy):
-        nh = hk.get_parameter("N_H", [], float, init=HaikuConstant(1))
+    def factor(self, energy):
         sigma = jnp.interp(energy, self.energy, self.sigma, left=1e9, right=0.0)
 
-        return jnp.exp(-nh * sigma)
+        return jnp.exp(-self.nh * sigma)
 
 
 class Phabs(MultiplicativeComponent):
     r"""
     A photoelectric absorption model.
 
-    ??? abstract "Parameters"
-        * $N_{\text{H}}$ : Equivalent hydrogen column density
-            $\left[\frac{\text{atoms}~10^{22}}{\text{cm}^2}\right]$
+    !!! abstract "Parameters"
+        * $N_{\text{H}}$ (`nh`) $\left[\frac{\text{atoms}~10^{22}}{\text{cm}^2}\right]$ : Equivalent hydrogen column density
+
 
     """
 
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
+    def __init__(self):
         table = Table.read(table_manager.fetch("xsect_phabs_aspl.fits"))
-        self.energy = jnp.asarray(np.array(table["ENERGY"]), dtype=np.float64)
-        self.sigma = jnp.asarray(np.array(table["SIGMA"]), dtype=np.float64)
+        self.energy = nnx.Variable(np.asarray(table["ENERGY"], dtype=np.float64))
+        self.sigma = nnx.Variable(np.asarray(table["SIGMA"], dtype=np.float64))
+        self.nh = nnx.Param(1.0)
 
-    def continuum(self, energy):
-        nh = hk.get_parameter("N_H", [], float, init=HaikuConstant(1))
-        sigma = jnp.interp(energy, self.energy, self.sigma, left=jnp.inf, right=0.0)
+    def factor(self, energy):
+        sigma = jnp.interp(energy, self.energy, self.sigma, left=1e9, right=0.0)
 
-        return jnp.exp(-nh * sigma)
+        return jnp.exp(-self.nh * sigma)
 
 
 class Wabs(MultiplicativeComponent):
@@ -95,22 +109,19 @@ class Wabs(MultiplicativeComponent):
     A photo-electric absorption using Wisconsin (Morrison & McCammon 1983) cross-sections.
 
     ??? abstract "Parameters"
-        * $N_{\text{H}}$ : Equivalent hydrogen column density
-            $\left[\frac{\text{atoms}~10^{22}}{\text{cm}^2}\right]$
-
+        * $N_{\text{H}}$ (`nh`) $\left[\frac{\text{atoms}~10^{22}}{\text{cm}^2}\right]$ : Equivalent hydrogen column density
     """
 
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
+    def __init__(self):
         table = Table.read(table_manager.fetch("xsect_wabs_angr.fits"))
-        self.energy = jnp.asarray(np.array(table["ENERGY"]), dtype=np.float64)
-        self.sigma = jnp.asarray(np.array(table["SIGMA"]), dtype=np.float64)
+        self.energy = nnx.Variable(np.asarray(table["ENERGY"], dtype=np.float64))
+        self.sigma = nnx.Variable(np.asarray(table["SIGMA"], dtype=np.float64))
+        self.nh = nnx.Param(1.0)
 
-    def continuum(self, energy):
-        nh = hk.get_parameter("N_H", [], float, init=HaikuConstant(1))
+    def factor(self, energy):
         sigma = jnp.interp(energy, self.energy, self.sigma, left=1e9, right=0.0)
 
-        return jnp.exp(-nh * sigma)
+        return jnp.exp(-self.nh * sigma)
 
 
 class Gabs(MultiplicativeComponent):
@@ -122,23 +133,26 @@ class Gabs(MultiplicativeComponent):
         \left( -\frac{\left(E-E_0\right)^2}{2 \sigma^2} \right) \right)
     $$
 
-    ??? abstract "Parameters"
-        * $\tau$ : Absorption strength $\left[\text{dimensionless}\right]$
-        * $\sigma$ : Absorption width $\left[\text{keV}\right]$
-        * $E_0$ : Absorption center $\left[\text{keV}\right]$
+    !!! abstract "Parameters"
+        * $\tau$ (`tau`) $\left[\text{dimensionless}\right]$ : Absorption strength
+        * $\sigma$ (`sigma`) $\left[\text{keV}\right]$ : Absorption width
+        * $E_0$ (`E0`) $\left[\text{keV}\right]$ : Absorption center
 
     !!! note
         The optical depth at line center is $\tau/(\sqrt{2 \pi} \sigma)$.
 
     """
 
-    def continuum(self, energy):
-        tau = hk.get_parameter("tau", [], float, init=HaikuConstant(1))
-        sigma = hk.get_parameter("sigma", [], float, init=HaikuConstant(1))
-        center = hk.get_parameter("E_0", [], float, init=HaikuConstant(1))
+    def __init__(self):
+        self.tau = nnx.Param(1.0)
+        self.sigma = nnx.Param(1.0)
+        self.E0 = nnx.Param(1.0)
 
+    def factor(self, energy):
         return jnp.exp(
-            -tau / (jnp.sqrt(2 * jnp.pi) * sigma) * jnp.exp(-0.5 * ((energy - center) / sigma) ** 2)
+            -self.tau
+            / (jnp.sqrt(2 * jnp.pi) * self.sigma)
+            * jnp.exp(-0.5 * ((energy - self.E0) / self.sigma) ** 2)
         )
 
 
@@ -151,16 +165,17 @@ class Highecut(MultiplicativeComponent):
         \left( \frac{E_c - E}{E_f} \right)& \text{if $E > E_c$}\\ 1 & \text{if $E < E_c$}\end{cases}
     $$
 
-    ??? abstract "Parameters"
-        * $E_c$ : Cutoff energy $\left[\text{keV}\right]$
-        * $E_f$ : Folding energy $\left[\text{keV}\right]$
+    !!! abstract "Parameters"
+        * $E_c$ (`Ec`) $\left[\text{keV}\right]$ : Cutoff energy
+        * $E_f$ (`Ef`) $\left[\text{keV}\right]$ : Folding energy
     """
 
-    def continuum(self, energy):
-        cutoff = hk.get_parameter("E_c", [], float, init=HaikuConstant(1))
-        folding = hk.get_parameter("E_f", [], float, init=HaikuConstant(1))
+    def __init__(self):
+        self.Ec = nnx.Param(1.0)
+        self.Ef = nnx.Param(1.0)
 
-        return jnp.where(energy <= cutoff, 1.0, jnp.exp((cutoff - energy) / folding))
+    def factor(self, energy):
+        return jnp.where(energy <= self.Ec, 1.0, jnp.exp((self.Ec - energy) / self.Ef))
 
 
 class Zedge(MultiplicativeComponent):
@@ -172,18 +187,21 @@ class Zedge(MultiplicativeComponent):
         & \text{if $E > E_c$}\\ 1 & \text{if $E < E_c$}\end{cases}
     $$
 
-    ??? abstract "Parameters"
-        * $E_c$ : Threshold energy
-        * $E_f$ : Absorption depth at the threshold
-        * $z$ : Redshift [dimensionless]
+    !!! abstract "Parameters"
+        * $E_c$ (`Ec`) $\left[\text{keV}\right]$ : Threshold energy
+        * $D$ (`D`) $\left[\text{dimensionless}\right]$ : Absorption depth at the threshold
+        * $z$ (`z`) $\left[\text{dimensionless}\right]$ : Redshift
     """
 
-    def continuum(self, energy):
-        E_c = hk.get_parameter("E_c", [], float, init=HaikuConstant(1))
-        D = hk.get_parameter("D", [], float, init=HaikuConstant(1))
-        z = hk.get_parameter("z", [], float, init=HaikuConstant(0))
+    def __init__(self):
+        self.Ec = nnx.Param(1.0)
+        self.D = nnx.Param(1.0)
+        self.z = nnx.Param(0.0)
 
-        return jnp.where(energy <= E_c, 1.0, jnp.exp(-D * (energy * (1 + z) / E_c) ** 3))
+    def factor(self, energy):
+        return jnp.where(
+            energy <= self.Ec, 1.0, jnp.exp(-self.D * (energy * (1 + self.z) / self.Ec) ** 3)
+        )
 
 
 class Tbpcf(MultiplicativeComponent):
@@ -194,28 +212,25 @@ class Tbpcf(MultiplicativeComponent):
         \mathcal{M}(E) = f \exp^{-N_{\text{H}}\sigma(E)} + (1-f)
     $$
 
-    ??? abstract "Parameters"
-        * $N_{\text{H}}$ : Equivalent hydrogen column density
-            $\left[\frac{\text{atoms}~10^{22}}{\text{cm}^2}\right]$
-        * $f$ : Partial covering fraction, ranges between 0 and 1 [dimensionless]
+    !!! abstract "Parameters"
+        * $N_{\text{H}}$ (`nh`) $\left[\frac{\text{atoms}~10^{22}}{\text{cm}^2}\right]$ : Equivalent hydrogen column density
+        * $f$ (`f`) $\left[\text{dimensionless}\right]$ : Partial covering fraction, ranges between 0 and 1
 
     !!! note
         Abundances and cross-sections $\sigma$ can be found in Wilms et al. (2000).
 
     """
 
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
+    def __init__(self):
         table = Table.read(table_manager.fetch("xsect_tbabs_wilm.fits"))
-        self.energy = jnp.asarray(np.array(table["ENERGY"]), dtype=np.float64)
-        self.sigma = jnp.asarray(np.array(table["SIGMA"]), dtype=np.float64)
+        self.energy = nnx.Variable(np.asarray(table["ENERGY"], dtype=np.float64))
+        self.sigma = nnx.Variable(np.asarray(table["SIGMA"], dtype=np.float64))
+        self.nh = nnx.Param(1.0)
+        self.f = nnx.Param(0.2)
 
-    def continuum(self, energy):
-        nh = hk.get_parameter("N_H", [], float, init=HaikuConstant(1))
-        f = hk.get_parameter("f", [], float, init=HaikuConstant(0.2))
+    def factor(self, energy):
         sigma = jnp.interp(energy, self.energy, self.sigma, left=1e9, right=0.0)
-
-        return f * jnp.exp(-nh * sigma) + (1 - f)
+        return self.f * jnp.exp(-self.nh * sigma) + (1.0 - self.f)
 
 
 class FDcut(MultiplicativeComponent):
@@ -227,12 +242,13 @@ class FDcut(MultiplicativeComponent):
     $$
 
     ??? abstract "Parameters"
-        * $E_c$ : Cutoff energy $\left[\text{keV}\right]$
-        * $E_f$ : Folding energy $\left[\text{keV}\right]$
+        * $E_c$ (`Ec`) $\left[\text{keV}\right]$ : Cutoff energy
+        * $E_f$ (`Ef`) $\left[\text{keV}\right]$ : Folding energy
     """
 
-    def continuum(self, energy):
-        cutoff = hk.get_parameter("E_c", [], init=HaikuConstant(1))
-        folding = hk.get_parameter("E_f", [], init=HaikuConstant(1))
+    def __init__(self):
+        self.Ec = nnx.Param(1.0)
+        self.Ef = nnx.Param(3.0)
 
-        return (1 + jnp.exp((energy - cutoff) / folding)) ** -1
+    def factor(self, energy):
+        return (1 + jnp.exp((energy - self.Ec) / self.Ef)) ** -1

jaxspec/scripts/debug.py

@@ -8,4 +8,4 @@ def debug_info():
 
     # Cimer CamilleTheBest pour l'idée
     print(watermark())
-    print(watermark(packages="jaxspec,jax,jaxlib,numpyro,haiku,numpy,scipy"))
+    print(watermark(packages="jaxspec,jax,jaxlib,numpyro,flax,numpy,scipy"))

jaxspec/util/__init__.py

@@ -1,45 +0,0 @@
-from collections.abc import Callable
-from contextlib import contextmanager
-from time import perf_counter
-
-import haiku as hk
-
-from jax.random import PRNGKey, split
-
-
-@contextmanager
-def catchtime(desc="Task", print_time=True) -> Callable[[], float]:
-    """
-    Context manager to measure time taken by a task.
-
-    Parameters
-    ----------
-        desc (str): Description of the task.
-        print_time (bool): Whether to print the time taken by the task.
-
-    Returns
-    -------
-        Callable[[], float]: Function to get the time taken by the task.
-    """
-
-    t1 = t2 = perf_counter()
-    yield lambda: t2 - t1
-    t2 = perf_counter()
-    if print_time:
-        print(f"{desc}: {t2 - t1:.3f} seconds")
-
-
-def sample_prior(dict_of_prior, key=PRNGKey(0), flat_parameters=False):
-    """
-    Sample the prior distribution from a dict of distributions
-    """
-
-    parameters = dict(hk.data_structures.to_haiku_dict(dict_of_prior))
-    parameters_flat = {}
-
-    for m, n, distribution in hk.data_structures.traverse(dict_of_prior):
-        key, subkey = split(key)
-        parameters[m][n] = distribution.sample(subkey)
-        parameters_flat[m + "_" + n] = distribution.sample(subkey)
-
-    return parameters if not flat_parameters else parameters_flat

jaxspec/util/misc.py

@@ -0,0 +1,25 @@
+from collections.abc import Callable
+from contextlib import contextmanager
+from time import perf_counter
+
+
+@contextmanager
+def catchtime(desc="Task", print_time=True) -> Callable[[], float]:
+    """
+    Context manager to measure time taken by a task.
+
+    Parameters
+    ----------
+        desc (str): Description of the task.
+        print_time (bool): Whether to print the time taken by the task.
+
+    Returns:
+    -------
+        Callable[[], float]: Function to get the time taken by the task.
+    """
+
+    t1 = t2 = perf_counter()
+    yield lambda: t2 - t1
+    t2 = perf_counter()
+    if print_time:
+        print(f"{desc}: {t2 - t1:.3f} seconds")

jaxspec/util/typing.py

@@ -1,68 +1,5 @@
-from typing import Any
-
 import numpyro.distributions as dist
 
-from jax import numpy as jnp
 from jax.typing import ArrayLike
-from pydantic import BaseModel, field_validator
 
 PriorDictType = dict[str, dict[str, dist.Distribution | ArrayLike]]
-
-
-def is_flat_dict(input_data: dict[str, Any]) -> bool:
-    """
-    Check if the input data is a flat dictionary with string keys and non-dictionary values.
-    """
-    return all(isinstance(k, str) and not isinstance(v, dict) for k, v in input_data.items())
-
-
-class PriorDictModel(BaseModel):
-    """
-    Pydantic model for a nested dictionary of NumPyro distributions or JAX arrays.
-    The top level keys are strings, and the values are dictionaries with string keys and values that are either
-    NumPyro distributions or JAX arrays (or convertible to JAX arrays).
-    """
-
-    nested_dict: PriorDictType
-
-    class Config:  # noqa D106
-        arbitrary_types_allowed = True
-
-    @classmethod
-    def from_dict(cls, input_prior: dict[str, Any]):
-        if is_flat_dict(input_prior):
-            nested_dict = {}
-
-            for key, obj in input_prior.items():
-                component, component_number, *parameter = key.split("_")
-
-                sub_dict = nested_dict.get(f"{component}_{component_number}", {})
-                sub_dict["_".join(parameter)] = obj
-
-                nested_dict[f"{component}_{component_number}"] = sub_dict
-
-            return cls(nested_dict=nested_dict)
-
-        return cls(nested_dict=input_prior)
-
-    @field_validator("nested_dict", mode="before")
-    def check_and_cast_nested_dict(cls, value: dict[str, Any]):
-        if not isinstance(value, dict):
-            raise ValueError("The top level must be a dictionary")
-
-        for key, inner_dict in value.items():
-            if not isinstance(inner_dict, dict):
-                raise ValueError(f'The value for key "{key}" must be a dictionary')
-
-            for inner_key, obj in inner_dict.items():
-                if not isinstance(obj, dist.Distribution):
-                    try:
-                        # Attempt to cast to JAX array
-                        value[key][inner_key] = jnp.array(obj, dtype=float)
-
-                    except Exception as e:
-                        raise ValueError(
-                            f'The value for key "{inner_key}" in {key} be a NumPyro '
-                            f"distribution or castable to JAX array. Error: {e}"
-                        )
-        return value

{jaxspec-0.1.4.dist-info → jaxspec-0.2.1.dist-info}/METADATA

@@ -1,43 +1,42 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: jaxspec
-Version: 0.1.4
+Version: 0.2.1
 Summary: jaxspec is a bayesian spectral fitting library for X-ray astronomy.
-Home-page: https://github.com/renecotyfanboy/jaxspec
 License: MIT
 Author: sdupourque
 Author-email: sdupourque@irap.omp.eu
-Requires-Python: >=3.10,<3.12
+Requires-Python: >=3.10,<3.13
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
 Requires-Dist: arviz (>=0.17.1,<0.21.0)
 Requires-Dist: astropy (>=6.0.0,<7.0.0)
+Requires-Dist: catppuccin (>=2.3.4,<3.0.0)
 Requires-Dist: chainconsumer (>=1.1.2,<2.0.0)
 Requires-Dist: cmasher (>=1.6.3,<2.0.0)
-Requires-Dist: dm-haiku (>=0.0.12,<0.0.13)
-Requires-Dist: gpjax (>=0.8.0,<0.9.0)
+Requires-Dist: flax (>=0.10.1,<0.11.0)
 Requires-Dist: interpax (>=0.3.3,<0.4.0)
-Requires-Dist: jax (>=0.4.33,<0.5.0)
-Requires-Dist: jaxlib (>=0.4.30,<0.5.0)
-Requires-Dist: jaxns (<2.6)
+Requires-Dist: jax (>=0.5.0,<0.6.0)
+Requires-Dist: jaxns (>=2.6.7,<3.0.0)
 Requires-Dist: jaxopt (>=0.8.1,<0.9.0)
 Requires-Dist: matplotlib (>=3.8.0,<4.0.0)
-Requires-Dist: mendeleev (>=0.15,<0.19)
+Requires-Dist: mendeleev (>=0.15,<0.20)
 Requires-Dist: networkx (>=3.1,<4.0)
 Requires-Dist: numpy (<2.0.0)
-Requires-Dist: numpyro (>=0.15.3,<0.16.0)
-Requires-Dist: optimistix (>=0.0.7,<0.0.10)
+Requires-Dist: numpyro (>=0.16.1,<0.17.0)
+Requires-Dist: optimistix (>=0.0.7,<0.0.11)
 Requires-Dist: pandas (>=2.2.0,<3.0.0)
 Requires-Dist: pooch (>=1.8.2,<2.0.0)
-Requires-Dist: pyzmq (<27)
 Requires-Dist: scipy (<1.15)
 Requires-Dist: seaborn (>=0.13.1,<0.14.0)
 Requires-Dist: simpleeval (>=0.9.13,<1.1.0)
-Requires-Dist: sparse (>=0.15.1,<0.16.0)
+Requires-Dist: sparse (>=0.15.4,<0.16.0)
 Requires-Dist: tinygp (>=0.3.0,<0.4.0)
 Requires-Dist: watermark (>=2.4.3,<3.0.0)
 Project-URL: Documentation, https://jaxspec.readthedocs.io/en/latest/
+Project-URL: Homepage, https://github.com/renecotyfanboy/jaxspec
 Description-Content-Type: text/markdown
 
 <p align="center">

jaxspec-0.2.1.dist-info/RECORD

@@ -0,0 +1,34 @@
+jaxspec/__init__.py,sha256=Sbn02lX6Y-zNXk17N8dec22c5jeypiS0LkHmGfz7lWA,126
+jaxspec/_fit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jaxspec/_fit/_build_model.py,sha256=NgLobtysyhCoiRguAazYGRIpG75iW-pG0Ss_hQGuCT4,2019
+jaxspec/analysis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jaxspec/analysis/_plot.py,sha256=1VbWentCLHR8XzM3jZtGyppbQRFWnChNztrwpZ2reA4,6185
+jaxspec/analysis/compare.py,sha256=g2UFhmR9Zt-7cz5gQFOB6lXuklXB3yTyUvjTypOzoSY,725
+jaxspec/analysis/results.py,sha256=0y10mnaTdmHJUV2h-xfKrAm6PE4S_Qscexknn3oClwo,25115
+jaxspec/data/__init__.py,sha256=aantcYKC9kZFvaE-V2SIwSuLhIld17Kjrd9CIUu___Y,415
+jaxspec/data/instrument.py,sha256=RDiG_LkucvnF2XE_ghTFME6d_2YirgQUcEY0gEle6dk,4775
+jaxspec/data/obsconf.py,sha256=r0deDgzpfKJZFMeiKQYUnkqQEmoDYTzl3eQFjb5xYjo,10263
+jaxspec/data/observation.py,sha256=dfb-hJFpeSPMS52oNmQl39CB9GD_MK1gG9vHlQXhhwU,7741
+jaxspec/data/ogip.py,sha256=57lfUgvMhM4j5tdfnOmwLI0ehx09pG7SvJx6VNIJFHE,9543
+jaxspec/data/util.py,sha256=FnIqrnwm29WsFV3FayeLXpjJDS10FcZ613B3SzGQCTk,9331
+jaxspec/fit.py,sha256=gO0H8CR76WP7lrG2X-Z3HLMndOSkCulzrp2jll1b4ow,24337
+jaxspec/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jaxspec/model/_graph_util.py,sha256=hPvHYmAxb7P3nyIecaZ7RqWOjwcZ1WvUByt_yNANiaY,4552
+jaxspec/model/abc.py,sha256=sHaFKOGfpD1ocXKlzEMdsmGLluikmBSmX8FAu7duFMQ,15489
+jaxspec/model/additive.py,sha256=Weipp5mrk-MrcPJdttBjR2mu5OHteV3BSo5Greq53I0,20208
+jaxspec/model/background.py,sha256=INRuuHIYgfu2E1a8Ddqpr_gF1wnhPtOpciWCteZfVVg,7687
+jaxspec/model/list.py,sha256=0RPAoscVz_zM1CWdx_Gd5wfrQWV5Nv4Kd4bSXu2ayUA,860
+jaxspec/model/multiplicative.py,sha256=5LHBAXCV9EQO1BGm1HzAI56dqjVbafBdV8Eq7_qXdbM,7890
+jaxspec/scripts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jaxspec/scripts/debug.py,sha256=qhyDtX4G5UdChmTLCM-5Wti4XZU-sU5S-wDb6TZjrvM,292
+jaxspec/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+jaxspec/util/abundance.py,sha256=fsC313taIlGzQsZNwbYsJupDWm7ZbqzGhY66Ku394Mw,8546
+jaxspec/util/integrate.py,sha256=_Ax_knpC7d4et2-QFkOUzVtNeQLX1-cwLvm-FRBxYcw,4505
+jaxspec/util/misc.py,sha256=O3qorCL1Y2X1BS2jdd36C1eDHK9QDXTSOr9kj3uqcJo,654
+jaxspec/util/online_storage.py,sha256=vm56RfcbFKpkRVfr0bXO7J9aQxuBq-I_oEgA26YIhCo,2469
+jaxspec/util/typing.py,sha256=ZQM_l68qyYnIBZPz_1mKvwPMx64jvVBD8Uj6bx9sHv0,140
+jaxspec-0.2.1.dist-info/LICENSE.md,sha256=2q5XoWzddts5IqzIcgYYMOL21puU3MfO8gvT3Ype1eQ,1073
+jaxspec-0.2.1.dist-info/METADATA,sha256=quIPdNWupFboPuiQqRQRMWHBy0TFrTwrpCkOAYNUWaM,4462
+jaxspec-0.2.1.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+jaxspec-0.2.1.dist-info/entry_points.txt,sha256=kzLG2mGlCWITRn4Q6zKG_idx-_RKAncvA0DMNYTgHAg,71
+jaxspec-0.2.1.dist-info/RECORD,,

{jaxspec-0.1.4.dist-info → jaxspec-0.2.1.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.1
+Generator: poetry-core 2.0.1
 Root-Is-Purelib: true
 Tag: py3-none-any

jaxspec/data/grouping.py

@@ -1,23 +0,0 @@
-import numpy as np
-from .observation import Observation
-
-
-def constant_rebin(obs: Observation, downfactor: int):
-    """
-    Create a grouping matrix for constant rebinning of the data.
-
-    Parameters:
-        obs: Observation object.
-        downfactor: Downfactor for the rebinning.
-
-    Returns:
-        A grouping matrix.
-    """
-    n_channels = len(obs.pha.channel)
-    n_bins = n_channels // downfactor + 1 * (n_channels % downfactor != 0)
-    grouping = np.zeros((n_bins, n_channels), dtype=bool)
-
-    for i in range(n_bins):
-        grouping[i, i * downfactor : (i + 1) * downfactor] = True
-
-    return grouping

jaxspec-0.1.4.dist-info/RECORD

@@ -1,33 +0,0 @@
-jaxspec/__init__.py,sha256=Sbn02lX6Y-zNXk17N8dec22c5jeypiS0LkHmGfz7lWA,126
-jaxspec/_fit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-jaxspec/_fit/_build_model.py,sha256=MvMgg0pzN-LuRz-5tw7GQ2WT3Go9r4rC4Y9MtzioCvs,4645
-jaxspec/analysis/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-jaxspec/analysis/_plot.py,sha256=C4XljmuzQz8xQur_jQddgInrBDmKgTn0eugSreLoD5k,862
-jaxspec/analysis/compare.py,sha256=g2UFhmR9Zt-7cz5gQFOB6lXuklXB3yTyUvjTypOzoSY,725
-jaxspec/analysis/results.py,sha256=apUy30N1hxswDnxA5w2d71C3EonKhU9m7Rsl0nI0UkA,28298
-jaxspec/data/__init__.py,sha256=aantcYKC9kZFvaE-V2SIwSuLhIld17Kjrd9CIUu___Y,415
-jaxspec/data/grouping.py,sha256=hhgBt-voiH0DDSyePacaIGsaMnrYbJM_-ZeU66keC7I,622
-jaxspec/data/instrument.py,sha256=0pSf1p82g7syDMmKm13eVbYih-Veiq5DnwsyZe6_b4g,3890
-jaxspec/data/obsconf.py,sha256=gv14sL6azK2avRiMCWuTbyLBPulzm4PwvoLY6iWPEVE,9833
-jaxspec/data/observation.py,sha256=oM2QcEYnnTyzw1WeuT6Wn-3GaFbbrdIbXGzoZKINJ8Q,7104
-jaxspec/data/ogip.py,sha256=sv9p00qHS5pzw61pzWyyF0nV-E-RXySdSFK2tUavokA,9545
-jaxspec/data/util.py,sha256=SKx9VgC9CL1Wler80FYqc3DiHsnAq1ulDioLzTjrVBE,9554
-jaxspec/fit.py,sha256=1wvflKoCVbnnMePAsI0EwvwmsqdbXF8bo50l3pZ_qRM,22968
-jaxspec/model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-jaxspec/model/abc.py,sha256=wbzpKYB2sB-qdP1B0YaUw7VHEYeTegeDhaoJ1EHPWaQ,20924
-jaxspec/model/additive.py,sha256=wjY2wL3Io3F45GJpz-UB8xYVnA-W1OFBnZMbj5pWPbQ,22449
-jaxspec/model/background.py,sha256=9gk6WdlaYytzB6ie9cllEYOQZs05A9QZCXp-gH-bPn0,7765
-jaxspec/model/list.py,sha256=0RPAoscVz_zM1CWdx_Gd5wfrQWV5Nv4Kd4bSXu2ayUA,860
-jaxspec/model/multiplicative.py,sha256=GCQ6JRz92QqbzDBFwWxGZ9SUqTJZQpD7B6ji9VEFXWo,8135
-jaxspec/scripts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-jaxspec/scripts/debug.py,sha256=RIRykBW_kzt8__PhOohQn2xtDW6oAz49E8rmuR5ewAU,293
-jaxspec/util/__init__.py,sha256=vKurfp7p2hxHptJjXhXqFAXAikAGXAqISMJUqPeiGTw,1259
-jaxspec/util/abundance.py,sha256=fsC313taIlGzQsZNwbYsJupDWm7ZbqzGhY66Ku394Mw,8546
-jaxspec/util/integrate.py,sha256=_Ax_knpC7d4et2-QFkOUzVtNeQLX1-cwLvm-FRBxYcw,4505
-jaxspec/util/online_storage.py,sha256=vm56RfcbFKpkRVfr0bXO7J9aQxuBq-I_oEgA26YIhCo,2469
-jaxspec/util/typing.py,sha256=8qK1aJlsqTcVKjYN-BxsDx20BTwtnS-wMw6Bdurpm-o,2459
-jaxspec-0.1.4.dist-info/LICENSE.md,sha256=2q5XoWzddts5IqzIcgYYMOL21puU3MfO8gvT3Ype1eQ,1073
-jaxspec-0.1.4.dist-info/METADATA,sha256=z-k8NXwz_PKDffLgdxlAu0OPUT0Vy0jig8ystiFwKOs,4456
-jaxspec-0.1.4.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-jaxspec-0.1.4.dist-info/entry_points.txt,sha256=kzLG2mGlCWITRn4Q6zKG_idx-_RKAncvA0DMNYTgHAg,71
-jaxspec-0.1.4.dist-info/RECORD,,