fixes #12

bmcfee · Jul 10, 2016 · de9c5b3 · de9c5b3
1 parent 7d4cac2
commit de9c5b3
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 1 deletion.
diff --git a/pumpp/feature/mel.py b/pumpp/feature/mel.py
@@ -10,7 +10,29 @@
 
 
 class Mel(FeatureExtractor):
+    '''Mel spectra feature extraction
 
+    Attributes
+    ----------
+    name : str or None
+        naming scope for this feature extractor
+
+    sr : number > 0
+        Sampling rate of the audio (in Hz)
+
+    hop_length : int > 0
+        Number of samples to advance between frames
+
+    n_fft :  int > 0
+        Number of samples per frame
+
+    n_mels : int > 0
+        Number of Mel frequency bins
+
+    fmax : number > 0
+        The maximum frequency bin.
+        Defaults to `0.5 * sr`
+    '''
     def __init__(self, name, sr, hop_length, n_fft, n_mels, fmax=None):
 
         super(Mel, self).__init__(name, sr, hop_length)
@@ -22,7 +44,6 @@ def __init__(self, name, sr, hop_length, n_fft, n_mels, fmax=None):
         self.register('mag', [None, n_mels], np.float32)
 
     def transform_audio(self, y):
-
         mel = np.sqrt(librosa.feature.melspectrogram(y=y, sr=self.sr, n_fft=self.n_fft,
                                                      hop_length=self.hop_length,
                                                      n_mels=self.n_mels,

diff --git a/pumpp/task/base.py b/pumpp/task/base.py
@@ -36,6 +36,9 @@ class BaseTaskTransformer(Scope):
     def __init__(self, name, namespace, sr, hop_length):
         super(BaseTaskTransformer, self).__init__(name)
 
+        # This will trigger an exception if the namespace is not found
+        jams.schema.is_dense(namespace)
+
         self.namespace = namespace
         self.sr = sr
         self.hop_length = hop_length