From 04fbfb841103af1d84cbac7b38ca50449f10f3ad Mon Sep 17 00:00:00 2001 From: ThibaultLSDC Date: Mon, 31 Mar 2025 11:42:36 -0400 Subject: [PATCH 1/2] fixing stale test --- tests/llm/test_tracking.py | 11 +++-------- 1 file changed, 3 insertions(+), 8 deletions(-) diff --git a/tests/llm/test_tracking.py b/tests/llm/test_tracking.py index 01ebcc06..229c2ef1 100644 --- a/tests/llm/test_tracking.py +++ b/tests/llm/test_tracking.py @@ -5,13 +5,9 @@ import pytest import agentlab.llm.tracking as tracking -from agentlab.llm.chat_api import ( - AzureChatModel, - OpenAIChatModel, - OpenRouterChatModel, - make_system_message, - make_user_message, -) +from agentlab.llm.chat_api import (AzureChatModel, OpenAIChatModel, + OpenRouterChatModel, make_system_message, + make_user_message) def test_get_action_decorator(): @@ -37,7 +33,6 @@ def test_get_action_decorator(): "meta-llama/llama-3.1-70b-instruct", "meta-llama/llama-3.1-8b-instruct", "google/gemini-pro-1.5", - "qwen/qwen-2-vl-72b-instruct", ) From 725fc860092181673d4f487ae49dd6487c8d9500 Mon Sep 17 00:00:00 2001 From: ThibaultLSDC Date: Mon, 31 Mar 2025 11:48:40 -0400 Subject: [PATCH 2/2] format --- tests/llm/test_tracking.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/tests/llm/test_tracking.py b/tests/llm/test_tracking.py index 229c2ef1..a38872e4 100644 --- a/tests/llm/test_tracking.py +++ b/tests/llm/test_tracking.py @@ -5,9 +5,13 @@ import pytest import agentlab.llm.tracking as tracking -from agentlab.llm.chat_api import (AzureChatModel, OpenAIChatModel, - OpenRouterChatModel, make_system_message, - make_user_message) +from agentlab.llm.chat_api import ( + AzureChatModel, + OpenAIChatModel, + OpenRouterChatModel, + make_system_message, + make_user_message, +) def test_get_action_decorator():