From 5ba15e65fdb725e48ba9bb8a9eec90855e7ccb57 Mon Sep 17 00:00:00 2001 From: Jim Chiang Date: Thu, 16 Aug 2018 13:29:12 -0700 Subject: [PATCH] add logger to unpickled psfs --- bin.src/imsim.py | 2 +- python/desc/imsim/imSim.py | 8 +++++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/bin.src/imsim.py b/bin.src/imsim.py index 397fcdbb..6bb4e8ae 100755 --- 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 index 0764b026..fe486b9e 100644 --- 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