pspipe 0.7__tar.gz → 0.7.2__tar.gz

{pspipe-0.7 → pspipe-0.7.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: pspipe
-Version: 0.7
+Version: 0.7.2
 Summary: Power spectra pipeline for Cosmic Dawn, Epoch of Reionization radio interferometric experiments
 Author: Florent Mertens
 Author-email: florent.mertens@gmail.com

{pspipe-0.7 → pspipe-0.7.2}/pspipe/obsdata.py

@@ -4,10 +4,14 @@
 
 import os
 
-from ps_eor import pspec, datacube, flagger, fgfit, ml_gpr
+from ps_eor import pspec, datacube, flagger
 
 from . import database
 
+# Lazy loaded:
+# from ps_eor import fgfit
+# from ps_eor import ml_gpr
+
 
 class ObsData(object):
 
@@ -90,6 +94,8 @@ class ObsData(object):
         return stokes in self.data or os.path.exists(self.get_h5_file(stokes))
 
     def get_gpr_res(self, eor_bin, umin=50, umax=250, stokes='I', name=''):
+        from ps_eor import fgfit
+
         path = os.path.join(os.path.dirname(self.get_h5_file(stokes)), name, f'eor{eor_bin}_u{umin}-{umax}')
         i_res = fgfit.GprForegroundResult.load(path, f'gpr_res_{stokes}')
         i_noise = datacube.CartDataCube.load(os.path.join(path, 'gpr_res_I.noise.h5'))
@@ -101,6 +107,8 @@ class ObsData(object):
         return os.path.join(os.path.dirname(self.get_h5_file('I')), name, f'eor{eor_bin}_u{umin}-{umax}')
 
     def get_ml_gpr_res(self, eor_bin, umin=50, umax=250, name='', sampler_method='mcmc'):
+        from ps_eor import ml_gpr
+
         path = self.get_ml_gpr_path(eor_bin, umin=umin, umax=umax, name=name, sampler_method=sampler_method)
         return ml_gpr.MLGPRResult.load(path, 'ml_gpr_res_I', sampler_method=sampler_method)
 
@@ -109,6 +117,8 @@ class ObsData(object):
         return os.path.join(os.path.dirname(self.get_h5_file('I')), name, d)
 
     def get_ml_gpr_inj_res(self, eor_bin, var_21_over_noise, x1, x2, umin=50, umax=250, name='', sampler_method='mcmc'):
+        from ps_eor import ml_gpr
+
         path = self.get_ml_gpr_inj_path(eor_bin, var_21_over_noise=var_21_over_noise, x1=x1, x2=x2, umin=umin, umax=umax, name=name)
         data_inj = datacube.CartDataCube.load(os.path.join(path, 'c_injected.h5'))
         return ml_gpr.MLGPRInjResult.load(path, 'ml_gpr_res_I', data_inj, sampler_method=sampler_method)

{pspipe-0.7 → pspipe-0.7.2}/pspipe/utils.py

@@ -2,6 +2,8 @@
 #
 # Author: F. Mertens
 
+import os
+
 import itertools
 
 import astropy.time as at

{pspipe-0.7 → pspipe-0.7.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [project]
 name = "pspipe"
-version = "0.7"
+version = "0.7.2"
 description = "Power spectra pipeline for Cosmic Dawn, Epoch of Reionization radio interferometric experiments"
 readme = "README.md"
 authors = [