def get_data(): sequence_obs, sequence_actions = buda.get_sequence( n_timesteps=viz_timesteps, obs_height=obs_height, obs_width=obs_width) return np.array(sequence_obs), np.array(sequence_actions)
def get_data(): sequence_obs, sequence_actions = buda.get_sequence(n_timesteps=30, obs_height=obs_height, obs_width=obs_width) return np.array(sequence_obs), np.array(sequence_actions)