From b40e0cdd699720b8999ae3d28cd1221dff37cf31 Mon Sep 17 00:00:00 2001 From: Kye Date: Thu, 27 Jul 2023 17:47:19 -0400 Subject: [PATCH] clean up --- tests/agents/models/hf.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/agents/models/hf.py b/tests/agents/models/hf.py index 68648904..292b94ba 100644 --- a/tests/agents/models/hf.py +++ b/tests/agents/models/hf.py @@ -2,7 +2,7 @@ import pytest import torch from unittest.mock import Mock, MagicMock from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig -from your_module import HuggingFaceLLM # replace with actual import +from swarms.agents.models.hf import HuggingFaceLLM # replace with actual import @pytest.fixture