Added return type annotations for create_pv_population and lnlikelihood.

Author: hpparvi

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