if __name__ == "__main__":
from argparse import ArgumentParser
from collections import namedtuple
from ptsa.data.readers import JsonIndexReader
os.environ['RHINO_ROOT'] = "~/mnt/rhino"
Dataset = namedtuple("Dataset", "subject,experiment,session")
parser = ArgumentParser()
parser.add_argument("outfile", type=str, help="path to output HDF5 file")
args = parser.parse_args()
reader = JsonIndexReader('/Users/depalati/mnt/rhino/protocols/r1.json')
df = reader.as_dataframe().reset_index()
subjects = df[df.experiment.isin(['FR5', 'catFR5'])].subject.unique()
datasets = [
# Dataset('R1395M', "catFR5", 4),
]
for subject in subjects:
for session in range(4):
datasets.append(Dataset(subject, 'FR5', session))
datasets.append(Dataset(subject, 'catFR5', session))
with h5py.File(args.outfile, 'w') as hfile:
count = 0
for dataset in datasets:
def test_as_dataframe(self):
index = osp.join(get_rhino_root(), "protocols", "r1.json")
reader = JsonIndexReader(index)
df = reader.as_dataframe()
assert isinstance(df, pd.DataFrame)