[CC] correction of nseg in case of a signal <= 200ms

Author: wantysal

mosqito/sq_metrics/roughness/roughness_dw/_roughness_dw_main_calc.py

@@ -11,7 +11,6 @@
     _ear_filter_coeff,
 )
 from mosqito.utils.conversion import freq2bark, db2amp, amp2db, bark2freq
-import matplotlib.pyplot as plt
 
 def _roughness_dw_main_calc(spec, freq_axis, fs, gzi, hWeight):
     """

mosqito/sq_metrics/roughness/roughness_dw/roughness_dw.py

@@ -67,7 +67,10 @@ def roughness_dw(signal, fs=None, overlap=0.5, is_sdt_output=False):
     sig, time = time_segmentation(
         signal, fs, nperseg=nperseg, noverlap=noverlap, is_ecma=False
     )
-    nseg = sig.shape[1]
+    if len(sig.shape) == 1:
+        nseg = 1
+    else:
+        nseg = sig.shape[1]
 
     spec, _ = spectrum(sig, fs, nfft="default", window="blackman", db=False)