trying to run in debug mode

Querent-ai · Dec 15, 2023 · 3bd7d61 · 3bd7d61
1 parent 267464e
commit 3bd7d61
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.github/workflows/workflow.yml b/.github/workflows/workflow.yml
@@ -122,7 +122,7 @@ jobs:
             gdown "${{ secrets.GOOGLE_DRIVE_LINK_LLAMA2 }}" -O ./tests/llama-2-7b-chat.Q5_K_S.gguf
           
       - name: Run Pytest
-        run: python -B -m pytest -v --disable-warnings .
+        run: python -B -m pytest -s --disable-warnings .
         env:
               PYTHONDONTWRITEBYTECODE: 1
 

diff --git a/querent/core/transformers/bert_llm.py b/querent/core/transformers/bert_llm.py
@@ -173,8 +173,10 @@ async def process_tokens(self, data: IngestedTokens):
                     cluster_persistence = cluster_output['cluster_persistence']
                     final_clustered_triples = self.triple_filter.filter_by_cluster_persistence(pairs_with_predicates, cluster_persistence, cluster_labels)
                     if final_clustered_triples is not None:
+                        print("Final cluster triple..........", final_clustered_triples)
                         filtered_triples, _ = self.triple_filter.filter_triples(final_clustered_triples)
                     else:
+                        print("Clustered triples..........", clustered_triples)
                         filtered_triples, _ = self.triple_filter.filter_triples(clustered_triples)
                         self.logger.log(f"Filtering in {self.__class__.__name__} producing 0 entity pairs. Filtering Disabled. ")
                 else: