diff --git a/requirements.txt b/requirements.txt index 59690d8..75b1893 100644 --- a/requirements.txt +++ b/requirements.txt @@ -4,6 +4,7 @@ celerite >= 0.4.2 colorama >= 0.4.6 corner >= 2.2.1 emcee >= 3.1.3 +h5py <= 3.12.1 linmix @ git+https://github.com/jmeyers314/linmix.git matplotlib >= 3.6.2 numba >= 0.56.4 diff --git a/src/pypetal/pyroa/utils.py b/src/pypetal/pyroa/utils.py index ed31637..8588890 100644 --- a/src/pypetal/pyroa/utils.py +++ b/src/pypetal/pyroa/utils.py @@ -602,6 +602,7 @@ def handler(signum, frame): for i in range(len(fnames)-1): filters = [line_names[0], line_names[i+1]] + cwd = os.getcwd() args = (lc_dir, objname, filters, prior_arr[i,:,:],) kwargs = {'add_var':add_var[i], 'init_tau':[init_tau[i]], 'init_delta':init_delta, 'sig_level':sig_level, @@ -637,6 +638,7 @@ def handler(signum, frame): signal.alarm(0) + shutil.move(cwd + '/Fit.h5', line_dir[i] + '/Fit.h5') except Exception as e: proc.terminate() @@ -645,6 +647,7 @@ def handler(signum, frame): print_error('Skipping and continuing to next line') fit_arr.append(None) + shutil.move(cwd + '/Fit.h5', line_dir[i] + '/Fit.h5') continue return fit_arr @@ -684,6 +687,7 @@ def handler(signum, frame): fit = MyFit(line_dir) signal.alarm(0) + shutil.move(cwd + '/Fit.h5', line_dir + '/Fit.h5') except Exception as e: proc.terminate() @@ -691,6 +695,7 @@ def handler(signum, frame): print_error('PyROA timed out'.format(line_names[i+1])) fit = None + shutil.move(cwd + '/Fit.h5', line_dir + '/Fit.h5') return fit