From 05d024d2944759e34be754a2a6cf1aea11cf0b79 Mon Sep 17 00:00:00 2001 From: Kye Date: Sat, 8 Jul 2023 15:58:10 -0400 Subject: [PATCH] llm class --- tests/LLM.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/LLM.py b/tests/LLM.py index 578723ce..57360294 100644 --- a/tests/LLM.py +++ b/tests/LLM.py @@ -3,7 +3,7 @@ import os from unittest.mock import patch, MagicMock from langchain import PromptTemplate, HuggingFaceHub, ChatOpenAI, LLMChain -from your_module import LLM # import the module where you defined the LLM class +from swarms.utils.llm import LLM class TestLLM(unittest.TestCase): @patch.object(HuggingFaceHub, '__init__', return_value=None)