iPddlmZddlZdZd dZd dZdZd dZ d d dZ y)) annotationsNz0.30.3cD ddl}t|ddS#t$rYywxYw)Nr __version__)mlx ImportErrorgetattr)rs k/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/demucs_mlx/mlx_backend.py_get_mlx_versionr s/ 3 t ,, s  c^t}|y|tkrtd|dtdyy)NzWarning: MLX version z is older than recommended .)r MIN_MLX_VERSIONprint)versions r ensure_mlx_versionrs: G  %gY.I/IZZ[\]!cN ddlm}|S#t$r}td|d}~wwxYw)Nrz$mlx is required for the MLX backend.)mlx.corecorer RuntimeError)mxexcs r _require_mlxrs4L I LABKLs $ $cZ ddlm}t||jS#t$rYywxYw)NrF)rr isinstancearrayr)xrs r is_mx_arrayrs-!RXX&& s  **cddl}t}||k(r|S|j}t|dk(rL|d}g}t |D]$} t || |||} |j | &|j|dSt|dk(rntd||j|j|d|||} |j| dS) a Resample MLX array using mlx-audio-io's native resampling. Args: x: Input MLX array with shape (channels, frames) or (batch, channels, frames) orig_freq: Original sample rate new_freq: Target sample rate quality: Resampling quality ("fastest", "low", "medium", "high", "best", "default") Returns: Resampled MLX array with same shape layout rN)axisz#Expected 2D or 3D array, got shape )r)quality) mlx_audio_iorshapelenrange resample_mxappendstack ValueErrorresample transpose) r orig_freqnew_freqr#macroriginal_shape batch_sizeoutputsi resampleds r r(r(#s$ BHWWN >a#A& z"A#AaD)XwGI NN9 %#xxax(( ^  ! >~>NOPP R\\!V4iSZ [I << 6 **r)returnztp.Optional[str])r6None)rztp.Anyr6bool)high)r.intr/r:r#str) __future__rtypingtpr r rrrr(rr r@sK"-^ *+*+*+ *+r