From 6114a8685586fe9c5a3412c960032d9b5d01975b Mon Sep 17 00:00:00 2001 From: Kye Date: Thu, 27 Jul 2023 17:47:19 -0400 Subject: [PATCH] clean up Former-commit-id: 8edb73116b3be54a5a278dd32bbd6819cafeec92 --- 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