Remove print statements

MaartenGr · Sep 29, 2023 · 3a76899 · 3a76899
1 parent 38f036d
commit 3a76899
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 8 deletions.
diff --git a/docs/changelog.md b/docs/changelog.md
@@ -3,8 +3,13 @@ hide:
  - navigation
 ---
 
+## **Version 0.8.1**
+*Release date: 29 September, 2023*
+
+* Remove unnecessary print statements
+
 ## **Version 0.8.0**
-*Release date: 27 September, 2023*
+*Release date: 29 September, 2023*
 
 **Highlights**:
 

diff --git a/keybert/__init__.py b/keybert/__init__.py
@@ -1,4 +1,4 @@
 from keybert._llm import KeyLLM
 from keybert._model import KeyBERT
 
-__version__ = "0.8.0"
+__version__ = "0.8.1"
diff --git a/keybert/_llm.py b/keybert/_llm.py
@@ -87,12 +87,10 @@ def extract_keywords(
  # Extract keywords for all documents not in a cluster
  if out_cluster:
  selected_docs = [docs[index] for index in out_cluster]
- print(out_cluster, selected_docs)
  if candidate_keywords is not None:
  selected_keywords = [candidate_keywords[index] for index in out_cluster]
  else:
  selected_keywords = None
- print(f"Call LLM with {len(selected_docs)} docs; out-cluster")
  out_cluster_keywords = self.llm.extract_keywords(
  selected_docs,
  selected_keywords,
@@ -102,12 +100,10 @@ def extract_keywords(
  # Extract keywords for only the first document in a cluster 
  if in_cluster:
  selected_docs = [docs[cluster[0]] for cluster in clusters]
- print(in_cluster, selected_docs)
  if candidate_keywords is not None:
  selected_keywords = [candidate_keywords[cluster[0]] for cluster in in_cluster]
  else:
  selected_keywords = None
- print(f"Call LLM with {len(selected_docs)} docs; in-cluster")
  in_cluster_keywords = self.llm.extract_keywords(
  selected_docs,
  selected_keywords
@@ -122,7 +118,6 @@ def extract_keywords(
  if out_cluster:
  if in_cluster:
  out_cluster_keywords.update(in_cluster_keywords)
- print(out_cluster_keywords)
  keywords = [out_cluster_keywords[index] for index in range(len(docs))]
  else:
  keywords = [in_cluster_keywords[index] for index in range(len(docs))]

diff --git a/setup.py b/setup.py
@@ -37,7 +37,7 @@
 setup(
  name="keybert",
  packages=find_packages(exclude=["notebooks", "docs"]),
- version="0.8.0",
+ version="0.8.1",
  author="Maarten Grootendorst",
  author_email="[email protected]",
  description="KeyBERT performs keyword extraction with state-of-the-art transformer models.",