Merge pull request #31 from dleemiller/fix-deduplicate

shape check before equality comparison

wordllama/inference.py

@@ -192,7 +192,7 @@ def cosine_similarity(a: np.ndarray, b: np.ndarray) -> np.ndarray:
             np.ndarray: A 2D array containing the cosine similarity scores between vectors in `a` and `b`.
         """
         # Normalize the vectors
-        if (a == b).all():
+        if a.shape == b.shape and (a == b).all():
             a = WordLlamaInference.normalize_embeddings(a)
             b = a
         else: