Added return type annotations for create_pv_population and lnlikelihood.

hpparvi · Jan 27, 2025 · ed8e24b · ed8e24b
1 parent b793f2f
commit ed8e24b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/exoiris/tslpf.py b/exoiris/tslpf.py
@@ -586,7 +586,7 @@ def flux_model(self, pv):
             transit_models[i][:, :, :] *= baseline_models[i][:, :, newaxis]
         return transit_models
 
-    def create_pv_population(self, npop: int = 50):
+    def create_pv_population(self, npop: int = 50) -> ndarray:
         """ Crate a parameter vector population.
         Parameters
         ----------
@@ -604,7 +604,7 @@ def set_radius_ratio_limits(self, kmin, kmax):
         for ipb in range(self.nk):
             self.set_prior(f'k_{ipb + 1:03d}', 'UP', kmin, kmax)
 
-    def lnlikelihood(self, pv):
+    def lnlikelihood(self, pv) -> ndarray | float :
         """Log likelihood for parameter vector pv.
 
         Parameters