import datajoint as dj
from database import Dataset, Repro
schema = dj.schema("fish_book", locals())


def get_baseline_data(cell_id=None, restriction={}):
    dsets = []
    if dataset_name:
        dsets = (Dataset & ("dataset_id like'%s'" % dataset_name) & restriction)
    else:
        dsets = (Dataset & restriction)

    if len(dsets) == 0:
        return None
    for d in dsets:
        d