Skip to content

Commit 67872c2

Browse files
committed
fix some tests
1 parent 489c182 commit 67872c2

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

tests/embedder/test_caching.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,7 @@ def test_caching_disabled_consistency(self, embedder_config: EmbedderConfig):
5050
embeddings2 = embedder.embed(test_utterances)
5151

5252
# Should still be the same since same model/input (deterministic)
53-
np.testing.assert_allclose(embeddings1, embeddings2, rtol=1e-5)
53+
np.testing.assert_allclose(embeddings1, embeddings2, atol=1e-3)
5454

5555

5656
class TestSentenceTransformerCachingSpecific:

tests/embedder/test_dump_load.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
import os
12
import tempfile
23
from pathlib import Path
34

@@ -74,7 +75,7 @@ def test_load_with_config_override(self, embedder: Embedder, on_windows):
7475
# For OpenAI, we can override batch_size too
7576
from autointent.configs import OpenaiEmbeddingConfig
7677

77-
override_config = OpenaiEmbeddingConfig(batch_size=16)
78+
override_config = OpenaiEmbeddingConfig(batch_size=16, api_key=os.getenv("OPENAI_API_KEY"))
7879

7980
# Load with override
8081
embedder_loaded = Embedder.load(temp_path, override_config)
@@ -125,5 +126,5 @@ def test_multiple_dump_load_cycles(self, embedder: Embedder, on_windows):
125126
embeddings_2 = embedder_2.embed(test_utterances)
126127

127128
# All embeddings should be consistent
128-
np.testing.assert_allclose(original_embeddings, embeddings_1, rtol=1e-5)
129-
np.testing.assert_allclose(embeddings_1, embeddings_2, rtol=1e-5)
129+
np.testing.assert_allclose(original_embeddings, embeddings_1, atol=1e-3)
130+
np.testing.assert_allclose(embeddings_1, embeddings_2, atol=1e-3)

0 commit comments

Comments
 (0)