Merge pull request #44 from mike9251/mike9251/fix_prep_filt_afb2d

Fix prep_filt_afb2d
fbcotter · Sep 20, 2022 · 9a0c507 · 9a0c507
2 parents 8d2e3b4 + 247be47
commit 9a0c507
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_wavelets/dwt/lowlevel.py b/pytorch_wavelets/dwt/lowlevel.py
@@ -942,7 +942,7 @@ def prep_filt_afb2d(h0_col, h1_col, h0_row=None, h1_row=None, device=None):
     """
     h0_col, h1_col = prep_filt_afb1d(h0_col, h1_col, device)
     if h0_row is None:
-        h0_row, h1_col = h0_col, h1_col
+        h0_row, h1_row = h0_col, h1_col
     else:
         h0_row, h1_row = prep_filt_afb1d(h0_row, h1_row, device)