Made warning smarter

Author: x-tabdeveloping

turftopic/late.py

@@ -16,12 +16,15 @@
 
 
 class LateSentenceTransformer(SentenceTransformer):
+    has_used_token_level = False
+
     def encode(
         self, sentences: Union[str, list[str], np.ndarray], *args, **kwargs
     ):
-        warnings.warn(
-            "Encoder is contextual but topic model is not using contextual embeddings. Perhaps you wanted to use another topic model."
-        )
+        if not self.has_used_token_level:
+            warnings.warn(
+                "Encoder is contextual but topic model is not using contextual embeddings. Perhaps you wanted to use another topic model."
+            )
         return super().encode(sentences, *args, **kwargs)
 
     def _encode_tokens(
@@ -38,6 +41,7 @@ def _encode_tokens(
         offsets: list[list[tuple[int, int]]]
             Start and end character of each token in each document.
         """
+        self.has_used_token_level = True
         token_embeddings = []
         offsets = []
         for start_index in trange(