add logger to unpickled psfs

LSSTDESC · Aug 16, 2018 · 5ba15e6 · 5ba15e6
1 parent 4146648
commit 5ba15e6
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/bin.src/imsim.py b/bin.src/imsim.py
@@ -55,7 +55,7 @@
     if args.psf_file is not None:
         desc.imsim.save_psf(psf, args.psf_file)
 else:
-    psf = desc.imsim.load_psf(args.psf_file)
+    psf = desc.imsim.load_psf(args.psf_file, log_level=args.log_level)
 
 sensor_list = args.sensors.split('^') if args.sensors is not None \
     else args.sensors

diff --git a/python/desc/imsim/imSim.py b/python/desc/imsim/imSim.py
@@ -917,10 +917,16 @@ def save_psf(psf, outfile):
     with open(outfile, 'wb') as output:
         pickle.dump(psf, output)
 
-def load_psf(psf_file):
+def load_psf(psf_file, log_level='INFO'):
     """
     Load a psf from a pickle file.
     """
     with open(psf_file, 'rb') as fd:
         psf = pickle.load(fd)
+
+    # Since save_psf sets any logger attribute to None, restore
+    # it here.
+    if hasattr(psf, 'logger'):
+        psf.logger = get_logger(log_level, 'psf')
+
     return psf