33 lines
708 B
Python
33 lines
708 B
Python
"""
|
|
This file contains functionality related to data.
|
|
"""
|
|
|
|
import seisbench.data as sbd
|
|
|
|
|
|
def get_dataset_by_name(name):
|
|
"""
|
|
Resolve dataset name to class from seisbench.data.
|
|
|
|
:param name: Name of dataset as defined in seisbench.data.
|
|
:return: Dataset class from seisbench.data
|
|
"""
|
|
try:
|
|
return sbd.__getattribute__(name)
|
|
except AttributeError:
|
|
raise ValueError(f"Unknown dataset '{name}'.")
|
|
|
|
|
|
def get_custom_dataset(path):
|
|
"""
|
|
Return custom dataset in seisbench format
|
|
:param path:
|
|
:return: Dataset class
|
|
"""
|
|
|
|
try:
|
|
return sbd.WaveformDataset(path)
|
|
except AttributeError:
|
|
raise ValueError(f"Unknown dataset '{path}'.")
|
|
|