Add explicit exception for remove discourse

MontrealCorpusTools · Jul 17, 2019 · 4959c32 · 4959c32
1 parent 70ed648
commit 4959c32
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/polyglotdb/corpus/base.py b/polyglotdb/corpus/base.py
@@ -448,6 +448,8 @@ def remove_discourse(self, name):
         name : str
             Name of the discourse to remove
         """
+        if name not in self.discourses:
+            raise GraphQueryError('{} is not a discourse in this corpus.'.format(name))
         d = self.discourse_sound_file(name)
         if d['consonant_file_path'] is not None and os.path.exists(d['consonant_file_path']):
             directory = self.discourse_audio_directory(name)