From 4331dfab7781f476400053e2535c047b51da8d0c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=A3o=20Faria?= <joao.faria@unige.ch>
Date: Thu, 6 Feb 2025 18:07:57 +0100
Subject: [PATCH] new instrument names; store priors

---
 src/kima/pykima/results.py | 9 +++++++++
 src/kima/pykima/utils.py   | 5 +++--
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/src/kima/pykima/results.py b/src/kima/pykima/results.py
index 4d0b146..765c642 100644
--- a/src/kima/pykima/results.py
+++ b/src/kima/pykima/results.py
@@ -1086,6 +1086,11 @@ def parameter_priors(self):
                     prior = self.priors[f'individual_offset_prior[{i}]']
                     priors[self.indices['inst_offsets']][i] = prior
 
+        if self.indicator_correlations:
+            prior = self.priors['beta_prior']
+            priors[self.indices['betas']] = prior
+
+
         if self.has_gp:
             if self.model == 'GPmodel':
                 priors[self.indices['GPpars']] = [
@@ -1161,6 +1166,9 @@ def parameter_priors(self):
                 planet_priors.append(self.priors['wprior'])
             priors[self.indices['planets']] = planet_priors
 
+        if self.studentt:
+            priors[self.indices['nu']] = self.priors['nu_prior']
+
         try:
             priors[self.indices['vsys']] = self.priors['Cprior']
         except KeyError:
@@ -1612,6 +1620,7 @@ def log_posterior(self, sample, separate=False):
             return logp + logl, logl, logp
         return logp + logl
 
+    @lru_cache
     def map_sample(self, Np=None, mask=None, printit=True,
                    from_posterior=False):
         """
diff --git a/src/kima/pykima/utils.py b/src/kima/pykima/utils.py
index cf4b038..1d5fa66 100644
--- a/src/kima/pykima/utils.py
+++ b/src/kima/pykima/utils.py
@@ -872,8 +872,9 @@ def get_instrument_name(data_file):
             r'SOPHIE',
             r'HAMILTON',
             r'NIRPS',
-            r'PFS',
-            r'UCLES'
+            r'PFS[\d+]',
+            r'UCLES',
+            r'MIKE',
         ])
         return re.findall(pattern, bn, re.IGNORECASE)[0]