Skip to content

Commit

Permalink
Updated run.py
Browse files Browse the repository at this point in the history
  • Loading branch information
Deeban Ramalingam committed May 30, 2024
1 parent 2276866 commit 6c5f35e
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions btx/diagnostics/run.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@
import holoviews as hv
from holoviews import opts
import pandas as pd
import h5py as h5
from btx.interfaces.ipsana import *
from btx.interfaces.ischeduler import JobScheduler

Expand Down Expand Up @@ -114,10 +113,8 @@ def save_traces(self, outdir, raw_img=False):
suffix=""
if raw_img:
suffix = "_raw"
hf = h5.File(os.path.join(outdir, f"r{self.psi.run:04}.h5"), 'w')
for key in self.stats_final.keys():
hf.create_dataset(f"r{self.psi.run:04}_{key}{suffix}", data=self.stats_final[key])
hf.close()
np.save(os.path.join(outdir, f"r{self.psi.run:04}_trace_{key}{suffix}.npy"), self.stats_final[key])

def load_traces(self, outdir, raw_img=False):
"""
Expand Down

0 comments on commit 6c5f35e

Please sign in to comment.