diff --git a/wetb/flex/__init__.py b/wetb/flex/__init__.py index 4d88eaf1e090114e4fe440c509ee4f83e9605c60..9bdc6ddbf9c798d100054b06b33e5d956b703bcf 100644 --- a/wetb/flex/__init__.py +++ b/wetb/flex/__init__.py @@ -1,6 +1,7 @@ from ._io import load, read_sensor_info from wetb import gtsdf +import numpy as np class Dataset(gtsdf.Dataset): - def __init__(self, filename): - self.time, self.data, self.info = load(filename) \ No newline at end of file + def __init__(self, filename, name_stop=8,dtype=np.float): + self.time, self.data, self.info = load(filename, name_stop,dtype) \ No newline at end of file diff --git a/wetb/hawc2/st_file.py b/wetb/hawc2/st_file.py index 09ae72407aa278b03c2a5164187bfb82c9a406e4..b739bc979c194f6b8239cb4773b1e3f22b98462b 100644 --- a/wetb/hawc2/st_file.py +++ b/wetb/hawc2/st_file.py @@ -68,7 +68,8 @@ class StFile(object): def __init__(self, filename): with open (filename) as fid: txt = fid.read() - no_maindata_sets = int(txt.replace("#","").strip()[0]) +# Some files sat + no_maindata_sets = int(txt.strip()[0]) assert no_maindata_sets == txt.count("#") self.main_data_sets = {} for mset in txt.split("#")[1:]: