refactor: more correct check not from PEP point of view

twirapp · Feb 15, 2024 · ac681d8 · ac681d8
1 parent 880a201
commit ac681d8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tts.py b/tts.py
@@ -33,9 +33,9 @@ def __init__(self):
 
     def generate(self, text: str, speaker: str, sample_rate: int) -> bytes:
         model = self.model_by_speaker.get(speaker)
-        if model is None:
+        if not model:
             raise NotFoundModelException(speaker)
-        if not sample_rate in self.VALID_SAMPLE_RATES:
+        if sample_rate not in self.VALID_SAMPLE_RATES:
             raise InvalidSampleRateException(sample_rate)
 
         return self._generate_audio(model, text, speaker, sample_rate)