diff --git a/simpeg_drivers/components/meshes.py b/simpeg_drivers/components/meshes.py index ce4aaf0c..bf2934a9 100644 --- a/simpeg_drivers/components/meshes.py +++ b/simpeg_drivers/components/meshes.py @@ -251,7 +251,7 @@ def ensure_cell_convention(mesh: Octree) -> TreeMesh | None: indices = treemesh.get_containing_cells(mesh.centroids) ind = np.argsort(indices) for child in mesh.children: - if child.values is None or isinstance(child.values, np.ndarray): + if child.values is None or not isinstance(child.values, np.ndarray): continue child.values = child.values[ind] diff --git a/tests/driver_test.py b/tests/driver_test.py index b428ec2c..388566cc 100644 --- a/tests/driver_test.py +++ b/tests/driver_test.py @@ -129,4 +129,5 @@ def test_mesh_visual_parameters_copy(tmp_path: Path): gz_uncertainty=2e-3, ) driver = GravityInversionDriver(params) + assert driver.inversion_mesh.ensure_cell_convention(mesh) is not None assert driver.inversion_mesh.entity.visual_parameters.colour == [255, 0, 0]