from environments.kitchen.spirl.utils.general_utils import AttrDict
from environments.kitchen.spirl.components.data_loader import GlobalSplitVideoDataset

data_spec = AttrDict(
    dataset_class=GlobalSplitVideoDataset,
    n_actions=3,
    state_dim=41,
    split=AttrDict(train=0.95, val=0.05, test=0.0),
    res=32,
    crop_rand_subseq=True,
)
data_spec.max_seq_len = 150
