From bf0f4fa48358fac0a9246a3f12575767d1d14b6f Mon Sep 17 00:00:00 2001
From: mmckerns <mmckerns@caltech.edu>
Date: Thu, 11 Aug 2022 13:31:57 -0400
Subject: [PATCH] move numpy special register to save

---
 dill/_dill.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/dill/_dill.py b/dill/_dill.py
index 614a522c..8b1f1b64 100644
--- a/dill/_dill.py
+++ b/dill/_dill.py
@@ -367,7 +367,7 @@ def __init__(self, file, *args, **kwds):
         self._postproc = OrderedDict()
         self._file = file
 
-    def dump(self, obj): #NOTE: if settings change, need to update attributes
+    def save(self, obj, save_persistent_id=True):
         # register if the object is a numpy ufunc
         # thanks to Paul Kienzle for pointing out ufuncs didn't pickle
         if NumpyUfuncType and numpyufunc(obj):
@@ -410,6 +410,11 @@ def save_numpy_array(pickler, obj):
         if GENERATOR_FAIL and type(obj) == GeneratorType:
             msg = "Can't pickle %s: attribute lookup builtins.generator failed" % GeneratorType
             raise PicklingError(msg)
+        StockPickler.save(self, obj, save_persistent_id)
+
+    save.__doc__ = StockPickler.save.__doc__
+
+    def dump(self, obj): #NOTE: if settings change, need to update attributes
         logger.trace_setup(self)
         StockPickler.dump(self, obj)