Merge pull request #7 from NFFT/develop

fixed index_set helper function

Author: passscoed

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "pyGroupedTransforms"
-version = "1.0.0"
+version = "1.1.0"
 authors = [
   { name="Felix Wirth", email="fwi012001@gmail.com" },
 ]

src/pyGroupedTransforms/NFCTtools.py

@@ -77,8 +77,7 @@ def nfct_index_set(
     if d == 1:
         return np.array([i for i in range(0, bandwidths[0])], "int")
 
-    bandwidths = bandwidths[::-1]
-    tmp = tuple([list(range(0, bw)) for bw in bandwidths])
+    tmp = tuple([list(range(0, bw)) for bw in bandwidths[::-1]])
     tmp = itertools.product(*(tmp[::-1]))
 
     freq = np.empty((d, np.prod(bandwidths)), dtype=int)
@@ -127,7 +126,7 @@ def get_transform(
     if bandwidths.ndim > 1 or bandwidths.dtype != "int32":
         return "Please use an zero or one-dimensional numpy.array with dtype 'int32' as input"
 
-    (M, d) = X.shape
+    M, d = X.shape
 
     if len(bandwidths) == 0:
         return DeferredLinearOperator(

src/pyGroupedTransforms/NFFTtools.py

@@ -88,7 +88,7 @@ def nfft_index_set(
             list(range(int(-bandwidths[0] / 2), int(bandwidths[0] / 2))), dtype="int"
         )
 
-    tmp = [list(range(int(-bw / 2), int(bw / 2))) for bw in bandwidths]
+    tmp = [list(range(int(-bw / 2), int(bw / 2))) for bw in bandwidths[::-1]]
     tmp = itertools.product(*(tmp[::-1]))
 
     freq = np.empty((d, np.prod(bandwidths)), dtype=int)
@@ -134,11 +134,10 @@ def get_transform(
     # Output:
      * `F::LinearMap{ComplexF64}` ... Linear map of the Fourier-transform implemented by the NFFT
     """
-
     if bandwidths.ndim > 1 or bandwidths.dtype != "int32":
         return "Please use an zero or one-dimensional numpy.array with dtype 'int32' as input"
 
-    (M, d) = np.shape(X)
+    M, d = np.shape(X)
 
     if len(bandwidths) == 0:
         return DeferredLinearOperator(