bug fix tests

Author: voorhs

tests/embedder/conftest.py

@@ -8,7 +8,7 @@
 # Check if OpenAI API key is available for testing
 openai_available = os.getenv("OPENAI_API_KEY") is not None
 
-pytest.importorskip("sentence-transformers")
+pytest.importorskip("sentence_transformers")
 
 
 @pytest.fixture

tests/embedder/test_basic.py

@@ -65,20 +65,3 @@ def test_similarity_symmetry(self, embedder: Embedder):
         sim2 = embedder.similarity(embeddings[1:], embeddings[:1])
 
         np.testing.assert_allclose(sim1, sim2.T, rtol=1e-5)
-
-    def test_return_tensors_functionality(self, embedder: Embedder):
-        """Test return_tensors parameter."""
-        utterances = ["Hello world", "Test sentence"]
-
-        # Test numpy return (default)
-        embeddings_np = embedder.embed(utterances, return_tensors=False)
-        assert isinstance(embeddings_np, np.ndarray)
-
-        # Test tensor return
-        embeddings_tensor = embedder.embed(utterances, return_tensors=True)
-        import torch
-
-        assert isinstance(embeddings_tensor, torch.Tensor)
-
-        # Values should be the same
-        np.testing.assert_allclose(embeddings_np, embeddings_tensor.cpu().numpy(), rtol=1e-5)