aboutsummaryrefslogtreecommitdiff
path: root/utils/dc
diff options
context:
space:
mode:
Diffstat (limited to 'utils/dc')
-rwxr-xr-xutils/dc12
1 files changed, 1 insertions, 11 deletions
diff --git a/utils/dc b/utils/dc
index ba363cc..9366b6e 100755
--- a/utils/dc
+++ b/utils/dc
@@ -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