load mcmc samples if requested

adrn · Feb 28, 2021 · 2383da7 · 2383da7
1 parent d65177d
commit 2383da7
Showing 1 changed file with 11 additions and 4 deletions.
diff --git a/hq/config.py b/hq/config.py
@@ -162,16 +162,23 @@ def get_source_data(self, source_id):
         rv_err = visits[self.rv_error_colname].astype('f8') * u.km/u.s
         return RVData(t=t, rv=rv, rv_err=rv_err)
 
-    def get_data_samples(self, source_id):
+    def get_data_samples(self, source_id, mcmc=False):
         import h5py
         import thejoker as tj
 
         data = self.get_source_data(source_id)
 
-        with h5py.File(self.joker_results_file, 'r') as results_f:
-            samples = tj.JokerSamples.read(results_f[source_id])
+        if not mcmc:
+            with h5py.File(self.joker_results_file, 'r') as results_f:
+                samples = tj.JokerSamples.read(results_f[source_id])
 
-        MAP_sample = samples[samples['ln_likelihood'].argmax()]
+            MAP_sample = samples[samples['ln_likelihood'].argmax()]
+
+        else:
+            with h5py.File(self.mcmc_results_file, 'r') as results_f:
+                samples = tj.JokerSamples.read(results_f[source_id])
+
+            MAP_sample = samples[samples['ln_likelihood'].argmax()]
 
         return data, samples, MAP_sample