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)