Fix support for numpy 2.4 (#472)

* Fix support for numpy 2.4

---------

Co-authored-by: Danny Dunlavy <[email protected]>

Author: ntjohnson1

pyproject.toml

@@ -10,7 +10,7 @@ readme = "README.md"
 requires-python = ">=3.10"
 
 dependencies = [
-    "numpy<2.4,>=1.24",
+    "numpy<2.5,>=1.24",
     "numpy_groupies>0.11",
     "scipy<1.17,>1.9",
     "matplotlib>3.7",

pyttb/sptensor.py

@@ -2030,7 +2030,9 @@ def ttv(
             )
             if np.count_nonzero(c) <= 0.5 * newsiz:
                 return ttb.sptensor.from_aggregator(
-                    np.arange(0, newsiz)[:, None], c.reshape((len(c), 1)), tuple(newsiz)
+                    np.arange(0, newsiz.item())[:, None],
+                    c.reshape((len(c), 1)),
+                    tuple(newsiz),
                 )
             return ttb.tensor(c, tuple(newsiz), copy=False)
 
@@ -2574,7 +2576,7 @@ def _set_subtensor(self, key, value):  # noqa: PLR0912, PLR0915
             # Generate appropriately sized ones vectors
             o = []
             for n in range(N):
-                o.append(np.ones((int(nssubs[n]), 1)))
+                o.append(np.ones((int(nssubs[n].item()), 1)))
 
             # Generate each column of the subscripts in turn
             for n in range(N):

pyttb/tensor.py

@@ -2172,6 +2172,8 @@ def _set_linear(self, key, value):
             idx = np.array(range(prod(self.shape))[key])
         idx = tt_ind2sub(self.shape, idx)
         if idx.shape[0] == 1:
+            if isinstance(value, np.ndarray) and value.size == 1:
+                value = value.item()
             self.data[tuple(idx[0, :])] = value
         else:
             actualIdx = tuple(idx.transpose())