Fix memory leak

1 files changed, 4 insertions, 1 deletions

diff --git a/data/keywords.py b/data/keywords.py
index 7385809..629006d 100644
--- a/data/keywords.py
+++ b/data/keywords.py
@@ -1,9 +1,12 @@
 import numpy as np
 
 
-def keywords_to_str(keywords: list[str], dropout: float = 0, shuffle: bool = False) -> str:
+def keywords_to_str(keywords: list[str], undroppable_keywords: list[str] = [], dropout: float = 0, shuffle: bool = False) -> str:
     if dropout != 0:
         keywords = [keyword for keyword in keywords if np.random.random() > dropout]
+    else:
+        keywords = keywords.copy()
+    keywords += undroppable_keywords
     if shuffle:
         np.random.shuffle(keywords)
     return ", ".join(keywords)