Commit 757a5e4d authored by matthmey's avatar matthmey
Browse files

addded warning

parent 74f5259c
...@@ -364,10 +364,10 @@ class SeismicSource(DataSource): ...@@ -364,10 +364,10 @@ class SeismicSource(DataSource):
del x.attrs["stats"] del x.attrs["stats"]
# x.rename({'seed_id':'channels'}) #TODO: rename seed_id to channels # x.rename({'seed_id':'channels'}) #TODO: rename seed_id to channels
# print(x.shape) # print(x.shape)
# TODO: same check for obspy # TODO: same check for obspy
if len(x['seed_id']) != len(config['channel']): if len(x["seed_id"]) != len(config["channel"]):
warnings.warn('Inconsistent data: Not all channels could be loaded') warnings.warn("Inconsistent data: Not all channels could be loaded")
return x return x
def process_seismic_data( def process_seismic_data(
...@@ -1422,7 +1422,7 @@ class SegmentedDataset(Dataset): ...@@ -1422,7 +1422,7 @@ class SegmentedDataset(Dataset):
warnings.simplefilter("default") warnings.simplefilter("default")
except Exception as e: except Exception as e:
warnings.simplefilter("default") warnings.simplefilter("default")
print('Ignoring file',e) print("Ignoring file", e)
continue continue
# TODO: maybe this can be done faster (and cleaner) # TODO: maybe this can be done faster (and cleaner)
i = o[0] i = o[0]
...@@ -1433,7 +1433,7 @@ class SegmentedDataset(Dataset): ...@@ -1433,7 +1433,7 @@ class SegmentedDataset(Dataset):
label = str(label) label = str(label)
if label not in self.classes: if label not in self.classes:
self.classes.append(label) self.classes.append(label)
if i not in label_dict: if i not in label_dict:
label_dict[i] = {"indexers": slices[i], "labels": [label]} label_dict[i] = {"indexers": slices[i], "labels": [label]}
elif label not in label_dict[i]["labels"]: elif label not in label_dict[i]["labels"]:
...@@ -1610,7 +1610,7 @@ class SegmentedDataset(Dataset): ...@@ -1610,7 +1610,7 @@ class SegmentedDataset(Dataset):
# class PytorchDataset(DataSource): # TODO: extends pytorch dataset # class PytorchDataset(DataSource): # TODO: extends pytorch dataset
# def __init__(self, source=None): # def __init__(self, source=None):
# """ Creates a pytorch like dataset from a data source and a label source. # """ Creates a pytorch like dataset from a data source and a label source.
# Arguments: # Arguments:
# DataSource {[type]} -- [description] # DataSource {[type]} -- [description]
# config {dict} -- configuration for labels # config {dict} -- configuration for labels
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment