astro-nest 0.7.2.dev2__tar.gz → 0.7.2.dev3__tar.gz

{astro_nest-0.7.2.dev2 → astro_nest-0.7.2.dev3}/NEST/core.py

@@ -407,6 +407,9 @@ class AgeModel:
             ages = self.ages
 
         ages = np.array([age for age in ages if np.any(np.isnan(age)) == False])
+
+        if len(ages) == 0:
+            raise ValueError('No valid age samples available to compute population age (maybe all stars are out of domain ?)')
         
         if ages.shape[1] == 1:
             return np.median(ages)
@@ -415,7 +418,7 @@ class AgeModel:
         loop = range(n_mc)
         if self.use_tqdm:
             loop = tqdm(loop)
-        min_offset = 0.1
+        epsilon = np.clip(1/ages.shape[0],0.001,0.1)
         for i in loop:
             random_star_indices = np.random.choice(list(range(ages.shape[0])), size=ages.shape[0], replace=True)
             random_star_indices = random_star_indices.astype(int)
@@ -425,12 +428,15 @@ class AgeModel:
                 age = random_star_ages[j]
                 pdf, bins = np.histogram(age,bins=nbins,range=(min_age,max_age))
                 pdf = pdf.astype(float)
-                pdf += min_offset
+                pdf += epsilon
                 global_pdf *= pdf
             best_age = bins[np.argmax(global_pdf)] + (bins[1]-bins[0])/2
             if best_age != bins[0]+(bins[1]-bins[0])/2:
                 pdfs.append(bins[np.argmax(global_pdf)] + (bins[1]-bins[0])/2)
 
+        if len(pdfs) == 0:
+            raise ValueError('No valid population age could be computed from the samples')
+
         population_age = np.median(pdfs)
 
         return population_age
@@ -633,7 +639,7 @@ class AgeModel:
                     #raise ValueError('Plotting individual isochrones is only available for the BaSTI model at the moment')
                 if type(plot_isochrone) is bool and plot_isochrone:
                     if self.ages is not None and self.ages.shape[1] > 1:
-                        isochrone_ages = [self.population_age()]
+                        isochrone_ages = [self.population_age(check_domain=check_domain)]
                     else:
                         isochrone_ages = [np.median(age)]
         if type(plot_isochrone) is float or type(plot_isochrone) is int:

{astro_nest-0.7.2.dev2 → astro_nest-0.7.2.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: astro-nest
-Version: 0.7.2.dev2
+Version: 0.7.2.dev3
 Summary: A python package that provides neural networks to estimate ages of stars.
 Author-email: Tristan Boin <tristan.boin@obspm.fr>
 License-Expression: MIT

{astro_nest-0.7.2.dev2 → astro_nest-0.7.2.dev3}/astro_nest.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: astro-nest
-Version: 0.7.2.dev2
+Version: 0.7.2.dev3
 Summary: A python package that provides neural networks to estimate ages of stars.
 Author-email: Tristan Boin <tristan.boin@obspm.fr>
 License-Expression: MIT

{astro_nest-0.7.2.dev2 → astro_nest-0.7.2.dev3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "astro-nest"
-version = "0.7.2dev2"
+version = "0.7.2dev3"
 authors = [
   { name="Tristan Boin", email="tristan.boin@obspm.fr" },
 ]