def __init__(self, num_envs=4096, env_size=(4, 4), steps=128, save=False, data_root='./data/', num_categories=None, verbose=False):
    self.num_envs = num_envs
    self.steps = steps
    self.env_size = env_size
    self.save = False
    self.data_root = data_root
    self.num_categories = num_categories
    self.generate_data(verbose=verbose)
    log.info('Generated Data: \n                    \t\t\t {} Environments...\n                    \t\t\t {} env size...\n                    \t\t\t {} steps in each... \n                    \t\t\t {} observable one hot categories... '.format(num_envs, env_size, steps, self.num_categories))
def __init__(self, num_envs=4096, env_size=(4, 4), steps=128, save=False, data_root='./data/', num_categories=None, verbose=False):
    self.num_envs = num_envs
    self.steps = steps
    self.env_size = env_size
    self.save = False
    self.data_root = data_root
    self.num_categories = num_categories
    self.generate_data(verbose=verbose)
    log.info('Generated Data: \n                    \t\t\t {} Environments...\n                    \t\t\t {} env size...\n                    \t\t\t {} steps in each... \n                    \t\t\t {} observable one hot categories... '.format(num_envs, env_size, steps, self.num_categories))