From fa6f1e55b10f2c9cf3a6c217cdff2a5d3e36c17b Mon Sep 17 00:00:00 2001 From: Braelyn Boynton Date: Wed, 14 Aug 2024 12:16:08 -0700 Subject: [PATCH] test import fix --- tests/test_canary.py | 2 +- tests/test_events.py | 2 +- tests/test_record_action.py | 2 +- tests/test_record_tool.py | 2 +- tests/test_session.py | 3 +-- 5 files changed, 5 insertions(+), 6 deletions(-) diff --git a/tests/test_canary.py b/tests/test_canary.py index a89cf3447..8ff49a01b 100644 --- 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 index 5234f3fad..a2d70d756 100644 --- 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 index d32585835..bc433c551 100644 --- 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 index c9f444712..b290e131b 100644 --- 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 index f54620857..075aca84f 100644 --- 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)