o si @sddlZdddZdS)NcsPzddlWn tytdwt\dur&dddn"dkr5dvs4Jfnd krDdksCJfntjjksVJjjfjd krtjd D]}j|d kstJ|jjfqbt fd dtj jddD jddSjd krjdSt j)aWrapper to allow independent axis for STOI. Args: reference: Shape [..., num_samples] estimation: Shape [..., num_samples] sample_rate: Returns: >>> np.random.seed(0) >>> a = np.random.normal(size=16_000) >>> b = a + np.random.normal(size=16_000) >>> pesq(a, b, sample_rate=16000) 2.2297563552856445 >>> pesq(a, b, sample_rate=8000) 1.0334522724151611 >>> pesq(a, b, sample_rate=16000, mode='nb') 3.200247049331665 >>> pesq([a, a], [b, b], sample_rate=16000) array([2.22975636, 2.22975636]) >>> pesq([[a, a], [b, a]], [[b, b], [b, b]], sample_rate=16000) array([[2.22975636, 2.22975636], [4.64388847, 2.22975636]]) >>> pesq(a, b, sample_rate=8000, mode='wb') Traceback (most recent call last): ... AssertionError: ('wb', 8000) rNz~To use this pesq implementation, install pesq from https://github.com/ludlows/python-pesq or istall it with `pip install pesq`nbwb)@>r)rrrcs&g|]}j||dqS)refdegfsmode)pesq).0i estimationr r reference sample_rateV/home/ubuntu/.local/lib/python3.10/site-packages/pb_bss_eval/evaluation/module_pesq.py Aszpesq..r ) r ImportErrorAssertionErrornpbroadcast_arrays ValueErrorshapendimrangearrayndindexreshapeNotImplementedError)rrrr rrrrrs:"   "  r)N)numpyrrrrrrs