test import fix

AgentOps-AI · Aug 14, 2024 · fa6f1e5 · fa6f1e5
1 parent 4bf8ba1
commit fa6f1e5
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 6 deletions.
diff --git a/tests/test_canary.py b/tests/test_canary.py
@@ -3,7 +3,7 @@
 import time
 import agentops
 from agentops import ActionEvent
-from agentops.helpers import clear_singletons
+from agentops.singleton import clear_singletons
 
 
 @pytest.fixture(autouse=True)

diff --git a/tests/test_events.py b/tests/test_events.py
@@ -3,7 +3,7 @@
 import pytest
 import agentops
 from agentops import ActionEvent, ErrorEvent
-from agentops.helpers import clear_singletons
+from agentops.singleton import clear_singletons
 
 
 @pytest.fixture(autouse=True)

diff --git a/tests/test_record_action.py b/tests/test_record_action.py
@@ -4,7 +4,7 @@
 import agentops
 from agentops import record_action
 from datetime import datetime
-from agentops.helpers import clear_singletons
+from agentops.singleton import clear_singletons
 import contextlib
 
 jwts = ["some_jwt", "some_jwt2", "some_jwt3"]

diff --git a/tests/test_record_tool.py b/tests/test_record_tool.py
@@ -5,7 +5,7 @@
 from agentops import record_tool
 from datetime import datetime
 
-from agentops.helpers import clear_singletons
+from agentops.singleton import clear_singletons
 import contextlib
 
 jwts = ["some_jwt", "some_jwt2", "some_jwt3"]

diff --git a/tests/test_session.py b/tests/test_session.py
@@ -3,8 +3,7 @@
 import time
 import agentops
 from agentops import ActionEvent, Client
-from agentops.exceptions import NoSessionException, MultiSessionException
-from agentops.helpers import clear_singletons
+from agentops.singleton import clear_singletons
 
 
 @pytest.fixture(autouse=True)