Fix regex matching being used in PartOfSpeech representation model (#…

…2138)
MaartenGr · Sep 17, 2024 · eba1d34 · eba1d34
1 parent 0b4265a
commit eba1d34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bertopic/representation/_pos.py b/bertopic/representation/_pos.py
@@ -120,7 +120,7 @@ def extract_topics(
             candidate_documents = []
             for keyword in keywords:
                 selection = documents.loc[documents.Topic == topic, :]
-                selection = selection.loc[selection.Document.str.contains(keyword), "Document"]
+                selection = selection.loc[selection.Document.str.contains(keyword, regex=False), "Document"]
                 if len(selection) > 0:
                     for document in selection[:2]:
                         candidate_documents.append(document)