Skip to content

Commit

Permalink
Merge branch 'MaartenGr:master' into master
Browse files Browse the repository at this point in the history
  • Loading branch information
robechun authored Sep 24, 2024
2 parents 28c159b + eba1d34 commit 242001f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion bertopic/representation/_pos.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down

0 comments on commit 242001f

Please sign in to comment.