Allow pickle when loading personas
@@ -64,7 +64,8 @@ class PersonasGenerator(object):
self.add_idx_stack()
idx = self.idx_stack.pop()
data = np.load(
- os.path.join(self.personas_path, self.personas_name_list[int(idx)])
+ os.path.join(self.personas_path, self.personas_name_list[int(idx)]),
+ allow_pickle=True,
)
return (idx, data)
