From 3a76899704beaf3740c43c54e9d04cddf6a3f5b5 Mon Sep 17 00:00:00 2001 From: MaartenGr Date: Fri, 29 Sep 2023 11:01:00 +0200 Subject: [PATCH] Remove print statements --- docs/changelog.md | 7 ++++++- keybert/__init__.py | 2 +- keybert/_llm.py | 5 ----- setup.py | 2 +- 4 files changed, 8 insertions(+), 8 deletions(-) diff --git a/docs/changelog.md b/docs/changelog.md index 5a7c8ad9..b1735336 100644 --- 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 index fece795c..45afb744 100644 --- 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 index ca8ff22e..bc2b4d52 100644 --- 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 index c1478967..9400b596 100644 --- 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="maartengrootendorst@gmail.com", description="KeyBERT performs keyword extraction with state-of-the-art transformer models.",