plotastrodata 1.9.19__tar.gz → 1.9.20__tar.gz

{plotastrodata-1.9.19/plotastrodata.egg-info → plotastrodata-1.9.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plotastrodata
-Version: 1.9.19
+Version: 1.9.20
 Summary: plotastrodata is a tool for astronomers to create figures from FITS files and perform fundamental data analyses with ease.
 Home-page: https://github.com/yusukeaso-astron/plotastrodata
 Download-URL: https://github.com/yusukeaso-astron/plotastrodata
@@ -8,7 +8,7 @@ Author: yusukeaso-astron
 License: GNU General Public License Version 3
 Keywords: astronomy,FITS,plot
 Classifier: Programming Language :: Python :: 3
-Requires-Python: >=3.12
+Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: astropy>=7.2
@@ -16,7 +16,6 @@ Requires-Dist: corner
 Requires-Dist: dynesty
 Requires-Dist: emcee
 Requires-Dist: matplotlib
-Requires-Dist: multiprocess
 Requires-Dist: numpy>=2.0
 Requires-Dist: pillow
 Requires-Dist: plotly
@@ -81,14 +80,14 @@ plotastrodata can do the following things.
  
 ## Requirement
 
-* astropy
+* Python >= 3.11
+* astropy >= 7.2
 * corner (only for fitting)
 * dynesty (only for fitting)
-* emcee (only for fitting)
+* emcee >= 3.0 (only for fitting)
 * ffmpeg (only for movie)
 * matplotlib
-* multiprocess (only for fitting)
-* numpy
+* numpy >= 2.0
 * pillow (only for RGB figures)
 * plotly (only for html cube)
 * ptemcee (only for fitting)

{plotastrodata-1.9.19 → plotastrodata-1.9.20}/README.md

@@ -52,14 +52,14 @@ plotastrodata can do the following things.
  
 ## Requirement
 
-* astropy
+* Python >= 3.11
+* astropy >= 7.2
 * corner (only for fitting)
 * dynesty (only for fitting)
-* emcee (only for fitting)
+* emcee >= 3.0 (only for fitting)
 * ffmpeg (only for movie)
 * matplotlib
-* multiprocess (only for fitting)
-* numpy
+* numpy >= 2.0
 * pillow (only for RGB figures)
 * plotly (only for html cube)
 * ptemcee (only for fitting)

{plotastrodata-1.9.19 → plotastrodata-1.9.20}/plotastrodata/__init__.py

@@ -1,4 +1,4 @@
 import warnings
 
 warnings.simplefilter('ignore', FutureWarning)
-__version__ = '1.9.19'
+__version__ = '1.9.20'

{plotastrodata-1.9.19 → plotastrodata-1.9.20}/plotastrodata/analysis_utils.py

@@ -856,6 +856,6 @@ class AstroFrame():
         d.beam = _as_list(d.beam, d.n, isbeam=True)
         grid = [d.x, d.y, d.v]
         for i in range(d.n):
-            self._read_one(d, i, grid.copy())
+            self._read_one(d, i, grid.copy())  # .copy() not to updated d.x again ang again.
         for name in ASTRODATA_ARGS + ['beam']:
             setattr(d, name, _scalar_if_single(getattr(d, name), d.n))

{plotastrodata-1.9.19 → plotastrodata-1.9.20}/plotastrodata/fitting_utils.py

@@ -38,9 +38,9 @@ def logp(x: np.ndarray) -> float:
 def _get_GR(samples: np.ndarray, nwalkers: int, ndata: int, dim: int
             ) -> np.ndarray:
     """Calculate the Gelman-Rubin statistics."""
-    B = np.std(np.mean(samples, axis=1), axis=0)
-    W = np.mean(np.std(samples, axis=1), axis=0)
-    V = (len(samples[0]) - 1) / len(samples[0]) * W \
+    B = np.std(np.mean(samples, axis=0), axis=0)
+    W = np.mean(np.std(samples, axis=0), axis=0)
+    V = (len(samples) - 1) / len(samples) * W \
         + (nwalkers + 1) / (nwalkers - 1) * B
     d = ndata - dim - 1
     GR = np.sqrt((d + 3) / (d + 1) * V / W)
@@ -66,7 +66,7 @@ class EmceeCorner():
 
     This class wraps ``emcee`` and ``ptemcee`` for simple bounded-parameter fitting.  The likelihood can be supplied directly through ``logl``, or it can be constructed from ``model``, ``xdata``, ``ydata``, and ``sigma``. Parameters are sampled with a uniform prior inside ``bounds`` and zero prior probability outside them.
 
-    After calling :meth:`fit`, the main results are stored as attributes:``samples`` for the post-burn-in chain, ``popt`` for the maximum-likelihood parameter set, and ``plow``, ``pmid``, and ``phigh`` for posterior percentiles.  The samples can be visualized with :meth:`plotcorner` and :meth:`plotchain`.
+    After calling :meth:`fit`, the main results are stored as attributes:``samples`` for the post-burn-in chain, ``popt`` for the maximum-likelihood parameter set, and ``plow``, ``pmid``, and ``phigh`` for posterior percentiles. ``samples`` has the shape ``(steps, walkers, dimensions)``. The samples can be visualized with :meth:`plotcorner` and :meth:`plotchain`.
 
     Args:
         bounds (np.ndarray): Parameter bounds with shape ``(dim, 2)``. logl (Callable, optional): Log-likelihood function. Defaults to None.
@@ -146,22 +146,23 @@ class EmceeCorner():
                      pt: bool) -> np.ndarray:
         """Extract post-burn-in samples from sampler chain."""
         if pt:
-            return sampler.chain[0, :, nburnin:, :]  # temperatures, walkers, steps, dim
+            chain = sampler.chain[0]  # walkers, steps, dim
+            return np.swapaxes(chain, 0, 1)[nburnin:, :, :]  # steps, walkers, dim
         else:
-            return sampler.chain[:, nburnin:, :]  # walkers, steps, dim
+            return sampler.get_chain(discard=nburnin)  # steps, walkers, dim
 
     def _get_lnp_popt(self, sampler: Any, pt: bool, nburnin: int,
                       ) -> tuple[np.ndarray, np.ndarray]:
         """Get log probabilities and best-fit parameters from sampler."""
         if pt:
-            lnp = sampler.logprobability[0]  # 0th temperature chain
-            chain = sampler.chain[0]
+            lnp = np.swapaxes(sampler.logprobability[0], 0, 1)
+            chain = np.swapaxes(sampler.chain[0], 0, 1)
         else:
-            lnp = sampler.lnprobability
-            chain = sampler.chain
+            lnp = sampler.get_log_prob()  # steps, walkers
+            chain = sampler.get_chain()  # steps, walkers, dim
         idx_best = np.unravel_index(np.argmax(lnp), lnp.shape)
         popt = chain[idx_best]
-        lnp = lnp[:, nburnin:]
+        lnp = lnp[nburnin:, :]
         return lnp, popt
 
     def _get_percentiles(self, samples: np.ndarray
@@ -266,14 +267,14 @@ class EmceeCorner():
         if ylim is None:
             ylim = self.bounds
         fig = plt.figure(figsize=(4, 2 * self.dim))
-        x = np.arange(np.shape(self.samples)[1])
+        x = np.arange(np.shape(self.samples)[0])
         naverage = max(1, len(x) // 100)
         nend = len(x) - len(x) % 100 if naverage > 1 else len(x)
         x = x[:nend:naverage]
         for i in range(self.dim):
-            y = self.samples[:, :, i]  # walkers, steps, dim
+            y = self.samples[:, :, i]  # steps, walkers, dim
             plist = [self.percent[0], 50, self.percent[1]]
-            y = [np.percentile(y, p, axis=0) for p in plist]  # percent over the walkers, steps
+            y = [np.percentile(y, p, axis=1) for p in plist]  # percent over the walkers, steps
             y = [[np.percentile(np.reshape(yy[:nend], (naverage, -1)), p, axis=0)
                   for p in plist] for yy in y]  # percent over the walkers, percent over the steps
             ax = fig.add_subplot(self.dim, 1, i + 1)

{plotastrodata-1.9.19 → plotastrodata-1.9.20/plotastrodata.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: plotastrodata
-Version: 1.9.19
+Version: 1.9.20
 Summary: plotastrodata is a tool for astronomers to create figures from FITS files and perform fundamental data analyses with ease.
 Home-page: https://github.com/yusukeaso-astron/plotastrodata
 Download-URL: https://github.com/yusukeaso-astron/plotastrodata
@@ -8,7 +8,7 @@ Author: yusukeaso-astron
 License: GNU General Public License Version 3
 Keywords: astronomy,FITS,plot
 Classifier: Programming Language :: Python :: 3
-Requires-Python: >=3.12
+Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: astropy>=7.2
@@ -16,7 +16,6 @@ Requires-Dist: corner
 Requires-Dist: dynesty
 Requires-Dist: emcee
 Requires-Dist: matplotlib
-Requires-Dist: multiprocess
 Requires-Dist: numpy>=2.0
 Requires-Dist: pillow
 Requires-Dist: plotly
@@ -81,14 +80,14 @@ plotastrodata can do the following things.
  
 ## Requirement
 
-* astropy
+* Python >= 3.11
+* astropy >= 7.2
 * corner (only for fitting)
 * dynesty (only for fitting)
-* emcee (only for fitting)
+* emcee >= 3.0 (only for fitting)
 * ffmpeg (only for movie)
 * matplotlib
-* multiprocess (only for fitting)
-* numpy
+* numpy >= 2.0
 * pillow (only for RGB figures)
 * plotly (only for html cube)
 * ptemcee (only for fitting)

{plotastrodata-1.9.19 → plotastrodata-1.9.20}/plotastrodata.egg-info/requires.txt

@@ -3,7 +3,6 @@ corner
 dynesty
 emcee
 matplotlib
-multiprocess
 numpy>=2.0
 pillow
 plotly

{plotastrodata-1.9.19 → plotastrodata-1.9.20}/setup.cfg

@@ -14,14 +14,13 @@ classifiers = Programming Language :: Python :: 3
 [options]
 packages = find:
 zip_safe = False
-python_requires = >=3.12
+python_requires = >=3.11
 install_requires = 
 	astropy >= 7.2
 	corner
 	dynesty
 	emcee
 	matplotlib
-	multiprocess
 	numpy >= 2.0
 	pillow
 	plotly