Let normalize.py load pickles with np.load()

Author: felker

plasma/preprocessor/normalize.py

@@ -185,7 +185,7 @@ def previously_saved_stats(self):
         if not os.path.isfile(self.path):
             return False, set([])
         else:
-            dat = np.load(self.path, encoding="latin1", allow_pickle=False)
+            dat = np.load(self.path, encoding="latin1", allow_pickle=True)
             machines = dat['machines'][()]
             ret = all(
                 [m in machines for m in self.conf['paths']['all_machines']])
@@ -292,7 +292,7 @@ def save_stats(self):
 
     def load_stats(self):
         assert self.previously_saved_stats()[0], "stats not saved before"
-        dat = np.load(self.path, encoding="latin1", allow_pickle=False)
+        dat = np.load(self.path, encoding="latin1", allow_pickle=True)
         self.means = dat['means'][()]
         self.stds = dat['stds'][()]
         self.num_processed = dat['num_processed'][()]
@@ -448,7 +448,7 @@ def save_stats(self):
 
     def load_stats(self):
         assert(self.previously_saved_stats()[0])
-        dat = np.load(self.path, encoding="latin1", allow_pickle=False)
+        dat = np.load(self.path, encoding="latin1", allow_pickle=True)
         self.minimums = dat['minimums'][()]
         self.maximums = dat['maximums'][()]
         self.num_processed = dat['num_processed'][()]