Skip to content

Commit 0af7df2

Browse files
committed
[FEAT] Squeeze function returns itself
1 parent 877d010 commit 0af7df2

1 file changed

Lines changed: 4 additions & 5 deletions

File tree

src/blosc2/ndarray.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1845,14 +1845,14 @@ def slice(self, key: int | slice | Sequence[slice], **kwargs: Any) -> NDArray:
18451845

18461846
return ndslice
18471847

1848-
def squeeze(self) -> None:
1848+
def squeeze(self) -> NDArray:
18491849
"""Remove single-dimensional entries from the shape of the array.
18501850
18511851
This method modifies the array in-place, removing any dimensions with size 1.
18521852
18531853
Returns
18541854
-------
1855-
out: None
1855+
out: NDArray
18561856
18571857
Examples
18581858
--------
@@ -1868,6 +1868,7 @@ def squeeze(self) -> None:
18681868
(23, 11)
18691869
"""
18701870
super().squeeze()
1871+
return self
18711872

18721873
def indices(self, order: str | list[str] | None = None, **kwargs: Any) -> NDArray:
18731874
"""
@@ -3744,9 +3745,7 @@ def matmul(x1: NDArray, x2: NDArray, **kwargs: Any) -> NDArray:
37443745
if x1_is_vector and x2_is_vector:
37453746
return result[0][0]
37463747

3747-
result.squeeze()
3748-
3749-
return result
3748+
return result.squeeze()
37503749

37513750

37523751
# Class for dealing with fields in an NDArray

0 commit comments

Comments
 (0)