From 39f4302934455acb1888747b5f0e447e060295d5 Mon Sep 17 00:00:00 2001 From: Christian Hotz-Behofsits Date: Sat, 22 Jul 2023 10:17:52 +0200 Subject: [PATCH] Update nade.py --- nade/nade.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nade/nade.py b/nade/nade.py index 5c7fa5f..2c704f9 100644 --- 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)