From 9305fb653a05e91069cd1111a0598852f8aed9b2 Mon Sep 17 00:00:00 2001 From: rohanbabbar04 Date: Tue, 15 Jul 2025 11:27:59 +0530 Subject: [PATCH] Use get_normalize_axis_index --- pylops_mpi/basicoperators/Laplacian.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pylops_mpi/basicoperators/Laplacian.py b/pylops_mpi/basicoperators/Laplacian.py index 7cc482e3..5395182c 100644 --- a/pylops_mpi/basicoperators/Laplacian.py +++ b/pylops_mpi/basicoperators/Laplacian.py @@ -1,10 +1,10 @@ from typing import Tuple import numpy as np -from numpy.core.multiarray import normalize_axis_index from mpi4py import MPI from pylops.utils.typing import DTypeLike, InputDimsLike from pylops.basicoperators import SecondDerivative +from pylops.utils.backend import get_normalize_axis_index from pylops_mpi import DistributedArray, MPILinearOperator, Partition from pylops_mpi.DistributedArray import local_split @@ -75,7 +75,7 @@ def __init__(self, dims: InputDimsLike, base_comm: MPI.Comm = MPI.COMM_WORLD, dtype: DTypeLike = np.float64): self.dims = dims - axes = tuple(normalize_axis_index(ax, len(dims)) for ax in axes) + axes = tuple(get_normalize_axis_index()(ax, len(dims)) for ax in axes) if not (len(axes) == len(weights) == len(sampling)): raise ValueError("axes, weights, and sampling have different size") self.axes = axes