get_chain fix for integrated_time

PynPoint · Oct 2, 2019 · f4a4f61 · f4a4f61
1 parent 1cc07bb
commit f4a4f61
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pynpoint/processing/fluxposition.py b/pynpoint/processing/fluxposition.py
@@ -854,14 +854,13 @@ def run(self) -> None:
         history = f'walkers = {self.m_nwalkers}, steps = {self.m_nsteps}'
         self.m_chain_out_port.copy_attributes(self.m_image_in_port)
         self.m_chain_out_port.add_history('MCMCsamplingModule', history)
-        self.m_chain_out_port.close_port()
 
         mean_accept = np.mean(sampler.acceptance_fraction)
         print(f'Mean acceptance fraction: {mean_accept:.3f}')
         self.m_chain_out_port.add_attribute('ACCEPTANCE', mean_accept, static=True)
 
         try:
-            autocorr = emcee.autocorr.integrated_time(sampler.get_chain(flat=True))
+            autocorr = emcee.autocorr.integrated_time(sampler.get_chain())
             print(f'Integrated autocorrelation time ={autocorr}')
 
         except emcee.autocorr.AutocorrError:
@@ -872,6 +871,7 @@ def run(self) -> None:
         self.m_chain_out_port.add_attribute('AUTOCORR_1', autocorr[1], static=True)
         self.m_chain_out_port.add_attribute('AUTOCORR_2', autocorr[2], static=True)
 
+        self.m_chain_out_port.close_port()
 
 class AperturePhotometryModule(ProcessingModule):
     """