add replacement wrapper for numpy's .data access

Author: MartinBubel

paramz/core/parameter_core.py

@@ -38,6 +38,8 @@
 import re
 import logging
 
+from paramz.util import _set_mem_addr
+
 from ..transformations import __fixed__, FIXED
 from .constrainable import Constrainable
 from .nameable import adjust_name_for_printing
@@ -287,8 +289,8 @@ def _propagate_param_grad(self, parray, garray):
             self.param_array[pislice] = pi.param_array.flat  # , requirements=['C', 'W']).flat
             self.gradient_full[pislice] = pi.gradient_full.flat  # , requirements=['C', 'W']).flat
 
-            pi.param_array.data = parray[pislice].data
-            pi.gradient_full.data = garray[pislice].data
+            _set_mem_addr(pi.param_array, parray[pislice])
+            _set_mem_addr(pi.gradient_full, garray[pislice])
 
             pi._propagate_param_grad(parray[pislice], garray[pislice])
             pi_old_size += pi.size

paramz/parameterized.py

@@ -28,7 +28,10 @@
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 #===============================================================================
 
-import numpy; np = numpy
+import ctypes
+import numpy
+
+from paramz.util import _set_mem_addr; np = numpy
 from re import compile
 try:
     from re import _pattern_type
@@ -268,8 +271,8 @@ def _connect_parameters(self, ignore_added_names=False):
             self.param_array[pslice] = p.param_array.flat  # , requirements=['C', 'W']).ravel(order='C')
             self.gradient_full[pslice] = p.gradient_full.flat  # , requirements=['C', 'W']).ravel(order='C')
 
-            p.param_array.data = self.param_array[pslice].data
-            p.gradient_full.data = self.gradient_full[pslice].data
+            _set_mem_addr(p.param_array, self.param_array[pslice])
+            _set_mem_addr(p.gradient_full, self.gradient_full[pslice])
 
             self._param_slices_.append(pslice)
 

paramz/util.py

@@ -27,6 +27,7 @@
 # OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 #===============================================================================
+import warnings
 
 def _inherit_doc(fromclass, done_classes = None):
     inherited = ''
@@ -44,4 +45,23 @@ def _inherit_doc(fromclass, done_classes = None):
                 pass
         inherited += _inherit_doc(c, done_classes=done_classes)        
 
-    return inherited
+    return inherited
+
+
+def _set_mem_addr(dest, src) -> None:
+    """
+    This function serves to replace the `.data` getter/setter that existed in
+    `numpy<2` and got removed in `numpy>=2`.
+    The original behavior was setting the memory address of dest to that of src.
+    However, directly setting the memory address of a numpy array to the data of
+    another one seems to be unwanted in `numpy>=2`, which is causing some major 
+    problems here.
+    """
+    with warnings.catch_warnings():
+        warnings.simplefilter("ignore", DeprecationWarning)
+
+        # original
+        # dest.data = src.data
+        
+        # take 1
+        dest.data = memoryview(src)