From fb1544df177aac8b210a429c2518988c86bf9967 Mon Sep 17 00:00:00 2001 From: jamescalam <james.briggs@hotmail.com> Date: Sun, 1 Dec 2024 15:11:57 +0100 Subject: [PATCH] fix: openai integration test --- tests/integration/encoders/test_openai_integration.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/integration/encoders/test_openai_integration.py b/tests/integration/encoders/test_openai_integration.py index c7bd8d9a..8c557d80 100644 --- a/tests/integration/encoders/test_openai_integration.py +++ b/tests/integration/encoders/test_openai_integration.py @@ -21,7 +21,7 @@ class TestOpenAIEncoder: os.environ.get("OPENAI_API_KEY") is None, reason="OpenAI API key required" ) def test_openai_encoder_init_success(self, openai_encoder): - assert openai_encoder.client is not None + assert openai_encoder._client is not None @pytest.mark.skipif( os.environ.get("OPENAI_API_KEY") is None, reason="OpenAI API key required" @@ -50,7 +50,7 @@ class TestOpenAIEncoder: ) def test_openai_encoder_call_uninitialized_client(self, openai_encoder): # Set the client to None to simulate an uninitialized client - openai_encoder.client = None + openai_encoder._client = None with pytest.raises(ValueError) as e: openai_encoder(["test document"]) assert "OpenAI client is not initialized." in str(e.value) -- GitLab