Skip to content

Commit fddb664

Browse files
Jammy2211Jammy2211
authored andcommitted
Revert "fix: restore Grid1D support in decorators — used by workspace scripts"
This reverts commit 8f22034.
1 parent 8f22034 commit fddb664

7 files changed

Lines changed: 4 additions & 84 deletions

File tree

autoarray/structures/decorators/abstract.py

Lines changed: 1 addition & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,6 @@
1-
from typing import Union
2-
31
import numpy as np
42

5-
from autoarray.mask.mask_1d import Mask1D
63
from autoarray.mask.mask_2d import Mask2D
7-
from autoarray.structures.grids.uniform_1d import Grid1D
84
from autoarray.structures.grids.irregular_2d import Grid2DIrregular
95
from autoarray.structures.grids.uniform_2d import Grid2D
106

@@ -62,7 +58,7 @@ def _xp(self):
6258
return np
6359

6460
@property
65-
def mask(self) -> Union[Mask1D, Mask2D]:
61+
def mask(self) -> Mask2D:
6662
return self.grid.mask
6763

6864
@property
@@ -75,14 +71,8 @@ def via_grid_2d(self, result):
7571
def via_grid_2d_irr(self, result):
7672
raise NotImplementedError
7773

78-
def via_grid_1d(self, result):
79-
raise NotImplementedError
80-
8174
@property
8275
def evaluate_func(self):
83-
if isinstance(self.grid, Grid1D):
84-
grid = self.grid.grid_2d_radial_projected_from()
85-
return self.func(self.obj, grid, self._xp, *self.args, **self.kwargs)
8676
return self.func(self.obj, self.grid, self._xp, *self.args, **self.kwargs)
8777

8878
@property
@@ -103,7 +93,5 @@ def result(self):
10393
return self.via_grid_2d(self.evaluate_func)
10494
elif isinstance(self.grid, Grid2DIrregular):
10595
return self.via_grid_2d_irr(self.evaluate_func)
106-
elif isinstance(self.grid, Grid1D):
107-
return self.via_grid_1d(self.evaluate_func)
10896

10997
return self.evaluate_func

autoarray/structures/decorators/to_array.py

Lines changed: 2 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,8 @@
33
from typing import List, Union
44

55
from autoarray.structures.arrays.irregular import ArrayIrregular
6-
from autoarray.structures.arrays.uniform_1d import Array1D
76
from autoarray.structures.arrays.uniform_2d import Array2D
87
from autoarray.structures.decorators.abstract import AbstractMaker
9-
from autoarray.structures.grids.uniform_1d import Grid1D
108
from autoarray.structures.grids.irregular_2d import Grid2DIrregular
119
from autoarray.structures.grids.uniform_2d import Grid2D
1210

@@ -22,11 +20,6 @@ def via_grid_2d_irr(self, result) -> Union[ArrayIrregular, List[ArrayIrregular]]
2220
return ArrayIrregular(values=result)
2321
return [ArrayIrregular(values=res) for res in result]
2422

25-
def via_grid_1d(self, result) -> Union[Array1D, List[Array1D]]:
26-
if not isinstance(result, list):
27-
return Array1D(values=result, mask=self.mask)
28-
return [Array1D(values=res, mask=self.mask) for res in result]
29-
3023

3124
def to_array(func):
3225
"""
@@ -46,11 +39,11 @@ def to_array(func):
4639
@wraps(func)
4740
def wrapper(
4841
obj: object,
49-
grid: Union[np.ndarray, Grid2D, Grid2DIrregular, Grid1D],
42+
grid: Union[np.ndarray, Grid2D, Grid2DIrregular],
5043
xp=np,
5144
*args,
5245
**kwargs,
53-
) -> Union[np.ndarray, Array1D, Array2D, ArrayIrregular, List]:
46+
) -> Union[np.ndarray, Array2D, ArrayIrregular, List]:
5447
return ArrayMaker(func=func, obj=obj, grid=grid, xp=xp, *args, **kwargs).result
5548

5649
return wrapper

autoarray/structures/decorators/to_grid.py

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@
33
from typing import List, Union
44

55
from autoarray.structures.decorators.abstract import AbstractMaker
6-
from autoarray.structures.grids.uniform_1d import Grid1D
76
from autoarray.structures.grids.irregular_2d import Grid2DIrregular
87
from autoarray.structures.grids.uniform_2d import Grid2D
98

@@ -35,13 +34,6 @@ def via_grid_2d_irr(self, result) -> Union[Grid2DIrregular, List[Grid2DIrregular
3534
return Grid2DIrregular(values=result)
3635
return [Grid2DIrregular(values=res) for res in result]
3736

38-
def via_grid_1d(self, result) -> Union[Grid2D, List[Grid2D]]:
39-
if not isinstance(result, list):
40-
return Grid2D(values=result, mask=self.mask.derive_mask.to_mask_2d)
41-
return [
42-
Grid2D(values=res, mask=self.mask.derive_mask.to_mask_2d) for res in result
43-
]
44-
4537

4638
def to_grid(func):
4739
"""
@@ -61,7 +53,7 @@ def to_grid(func):
6153
@wraps(func)
6254
def wrapper(
6355
obj: object,
64-
grid: Union[np.ndarray, Grid2D, Grid2DIrregular, Grid1D],
56+
grid: Union[np.ndarray, Grid2D, Grid2DIrregular],
6557
xp=np,
6658
*args,
6759
**kwargs,

autoarray/structures/decorators/to_vector_yx.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,6 @@
33
from typing import List, Union
44

55
from autoarray.structures.decorators.abstract import AbstractMaker
6-
from autoarray.structures.grids.uniform_1d import Grid1D
76
from autoarray.structures.grids.irregular_2d import Grid2DIrregular
87
from autoarray.structures.grids.uniform_2d import Grid2D
98
from autoarray.structures.vectors.irregular import VectorYX2DIrregular

autoarray/structures/decorators/transform.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
import numpy as np
33
from typing import Union
44

5-
from autoarray.structures.grids.uniform_1d import Grid1D
65
from autoarray.structures.grids.irregular_2d import Grid2DIrregular
76
from autoarray.structures.grids.uniform_2d import Grid2D
87

test_autoarray/structures/decorators/test_to_array.py

Lines changed: 0 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -3,22 +3,6 @@
33
import autoarray as aa
44

55

6-
def test__in_grid_1d__out_ndarray_1d_list():
7-
mask = aa.Mask1D(mask=[True, False, False, True], pixel_scales=(1.0,))
8-
9-
grid_1d = aa.Grid1D.from_mask(mask=mask)
10-
11-
obj = aa.m.MockGrid2DLikeObj()
12-
13-
ndarray_1d_list = obj.ndarray_1d_list_from(grid=grid_1d)
14-
15-
assert isinstance(ndarray_1d_list[0], aa.Array1D)
16-
assert (ndarray_1d_list[0].native == np.array([[0.0, 1.0, 1.0, 0.0]])).all()
17-
18-
assert isinstance(ndarray_1d_list[1], aa.Array1D)
19-
assert (ndarray_1d_list[1].native == np.array([[0.0, 2.0, 2.0, 0.0]])).all()
20-
21-
226
def test__in_grid_2d__out_ndarray_1d_list():
237
mask = aa.Mask2D(
248
mask=[

test_autoarray/structures/decorators/test_to_grid.py

Lines changed: 0 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -4,41 +4,6 @@
44
import autoarray as aa
55

66

7-
def test__in_grid_1d__out_ndarray_2d():
8-
mask_1d = aa.Mask1D(mask=[True, False, False, True], pixel_scales=(1.0,))
9-
10-
grid_1d = aa.Grid1D.from_mask(mask=mask_1d)
11-
12-
obj = aa.m.MockGrid2DLikeObj()
13-
14-
ndarray_2d = obj.ndarray_2d_from(grid=grid_1d)
15-
16-
assert isinstance(ndarray_2d, aa.Grid2D)
17-
assert ndarray_2d.native == pytest.approx(
18-
np.array([[[0.0, 0.0], [0.0, -1.0], [0.0, 1.0], [0.0, 0.0]]]), abs=1.0e-4
19-
)
20-
21-
22-
def test__in_dgrid_1d__out_ndarray_2d_list():
23-
mask = aa.Mask1D(mask=[True, False, False, True], pixel_scales=(1.0,))
24-
25-
grid_1d = aa.Grid1D.from_mask(mask=mask)
26-
27-
obj = aa.m.MockGrid2DLikeObj()
28-
29-
ndarray_2d_list = obj.ndarray_2d_list_from(grid=grid_1d)
30-
31-
assert isinstance(ndarray_2d_list[0], aa.Grid2D)
32-
assert ndarray_2d_list[0].native == pytest.approx(
33-
np.array([[[0.0, 0.0], [0.0, -0.5], [0.0, 0.5], [0.0, 0.0]]]), abs=1.0e-4
34-
)
35-
36-
assert isinstance(ndarray_2d_list[1], aa.Grid2D)
37-
assert ndarray_2d_list[1].native == pytest.approx(
38-
np.array([[[0.0, 0.0], [0.0, -1.0], [0.0, 1.0], [0.0, 0.0]]]), abs=1.0e-4
39-
)
40-
41-
427
def test__in_grid_2d__out_ndarray_2d():
438
mask = aa.Mask2D(
449
mask=[

0 commit comments

Comments
 (0)