diff options
Diffstat (limited to 'utils/dc')
-rwxr-xr-x | utils/dc | 12 |
1 files changed, 1 insertions, 11 deletions
@@ -19,7 +19,6 @@ import numpy as np from scipy.stats import iqr import nlopt from scipy.stats import poisson -import contextlib import sys from math import exp import emcee @@ -29,6 +28,7 @@ from matplotlib.lines import Line2D from sddm.plot import despine from sddm.dc import * from sddm.plot_energy import * +from sddm import printoptions try: from emcee import moves @@ -36,16 +36,6 @@ except ImportError: print("emcee version 2.2.1 is required",file=sys.stderr) sys.exit(1) -# from https://stackoverflow.com/questions/2891790/how-to-pretty-print-a-numpy-array-without-scientific-notation-and-with-given-pre -@contextlib.contextmanager -def printoptions(*args, **kwargs): - original = np.get_printoptions() - np.set_printoptions(*args, **kwargs) - try: - yield - finally: - np.set_printoptions(**original) - def radius_cut(ev): ev['radius_cut'] = np.digitize((ev.r/PSUP_RADIUS)**3,(0.9,)) return ev |