Disallow pickle loading in npz files

keras-team · Jan 14, 2025 · 57c94f3 · 57c94f3
1 parent f70fd8a
commit 57c94f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/keras/src/saving/saving_lib.py b/keras/src/saving/saving_lib.py
@@ -1073,7 +1073,7 @@ def __init__(self, root_path, archive=None, mode="r"):
                 self.f = archive.open(root_path, mode="r")
             else:
                 self.f = open(root_path, mode="rb")
-            self.contents = np.load(self.f, allow_pickle=True)
+            self.contents = np.load(self.f)
 
     def make(self, path, metadata=None):
         if not path: