Update nade.py

inkrement · Jul 22, 2023 · 39f4302 · 39f4302
1 parent 66e3847
commit 39f4302
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nade/nade.py b/nade/nade.py
@@ -126,7 +126,7 @@ def predict_emojis(
     predict emotions based on emoji (stage 2)
     '''
     def predict(self, txts: List[str], dimensions: None | List[str]) -> List[str]:
-        dims_ = dimensions if dimensions not None else self.labels
+        dims_ = dimensions if dimensions is not None else self.labels
         ft_op = self.predict_emojis(txts, sort_by_key=True, k=151)
         X, _ = zip(*ft_op)