Skip to content

Commit 04e0043

Browse files
committed
use CoordinateArrayTriangles by default
1 parent 70e96ec commit 04e0043

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

autolens/point/solver/shape_solver.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -4,24 +4,27 @@
44
from typing import Tuple, List, Iterator, Type, Optional
55

66
import autoarray as aa
7-
from autoarray.structures.triangles.coordinate_array import CoordinateArrayTriangles
87

98
from autoarray.structures.triangles.shape import Shape
109
from autofit.jax_wrapper import jit, use_jax, numpy as np, register_pytree_node_class
1110

1211
try:
1312
if use_jax:
14-
from autoarray.structures.triangles.array.jax_array import (
15-
ArrayTriangles,
13+
from autoarray.structures.triangles.coordinate_array.jax_coordinate_array import (
14+
CoordinateArrayTriangles,
1615
MAX_CONTAINING_SIZE,
1716
)
1817
else:
19-
from autoarray.structures.triangles.array import ArrayTriangles
18+
from autoarray.structures.triangles.coordinate_array.coordinate_array import (
19+
CoordinateArrayTriangles,
20+
)
2021

2122
MAX_CONTAINING_SIZE = None
2223

2324
except ImportError:
24-
from autoarray.structures.triangles.array import ArrayTriangles
25+
from autoarray.structures.triangles.coordinate_array.coordinate_array import (
26+
CoordinateArrayTriangles,
27+
)
2528

2629
MAX_CONTAINING_SIZE = None
2730

0 commit comments

Comments
 (0)