Add debug print for integration test

comet-ml · Nov 28, 2023 · 0818ea5 · 0818ea5
1 parent 4865fc8
commit 0818ea5
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/lib_integration/openai/test_openai_autologger.py b/tests/lib_integration/openai/test_openai_autologger.py
@@ -4,10 +4,12 @@
 import pytest
 
 import comet_llm
+import logging
 import comet_llm.chains.state
 
 from ... import testlib
 
+LOGGER = logging.getLogger(__file__)
 
 @pytest.fixture
 def comet_setup():
@@ -18,7 +20,7 @@ def comet_setup():
 def test_openai_autologger__chain_exists__openai_call_was_made__openai_call_added_to_chain_as_node(comet_setup):
     import openai
 
-    print("!!!!!!!!!!!!!!!!!!!!!!!!!", os.environ["OPENAI_API_KEY"][:5], os.environ["OPENAI_ORG_ID"][:5])
+    LOGGER.error("ENV INFO:", os.environ["OPENAI_API_KEY"][:5] + " " + os.environ["OPENAI_ORG_ID"][:5])
 
     comet_llm.start_chain(
         inputs={"any-name": "any-input"},