i ZSrSSKJr SSKJrJr SSKrSSKJr "SS5r \ "5r g)uLazy loader for ``pytorch_wavelets`` DWT1D modules used by DCW. Splitting this out keeps :mod:`acestep.models.common.dcw_correction` under the project's 200-LOC module cap. We import ``pytorch_wavelets`` lazily so the dependency stays optional — DCW is opt-in, and users who don't enable it never touch this code path. If the import fails when DCW *is* enabled, we log a single warning and return ``None`` so callers can short-circuit to a no-op. ) annotations)OptionalTupleN)loggercB\rSrSrSrSSjrSrS SjrSrg) _LazyWaveletzLazy loader for ``pytorch_wavelets`` DWT1D modules. We cache one ``DWT1DForward`` / ``DWT1DInverse`` pair per ``(device, dtype, wavelet)`` triple so repeated sampler steps don't keep rebuilding the filter banks. c 0UlSUlg)NF_cache_import_failed)selfs W/mnt/workspace/acestep.cpp/tests/../../ACE-Step-1.5/acestep/models/common/dcw_loader.py__init___LazyWavelet.__init__s #cUR(agSSKJnJn X4$![a SUl[ R "S5 gf=f)Nr) DWT1DForward DWT1DInverseTzDCW is enabled but 'pytorch_wavelets' is not installed. Install with `pip install pytorch_wavelets PyWavelets` to use Differential Correction in Wavelet domain. Falling back to no-op for this generation.)r pytorch_waveletsrr ImportErrorrwarning)rrrs r _try_import_LazyWavelet._try_import sP     C)) "&D  NN5   s'A A c $UR5nUcgUupV[U5[U5U4nURRU5nUbU$U"SSUS9R U[ R S9n U"SUS9R U[ R S9n X4URU'[U SS5n U b[U RS5OSn [R"S X<[U5[U55 X4$![a Sn NrDs. #" DDP r