From 2bf80e40762a3e7a62ebcc89640f9a6deda2d3ea Mon Sep 17 00:00:00 2001 From: Volpeon Date: Mon, 3 Apr 2023 07:19:06 +0200 Subject: Fix memory leak --- data/keywords.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'data/keywords.py') 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) -- cgit v1.2.3-54-g00ecf