From 81d02a610366f0305428d2b300c4b059f34b3875 Mon Sep 17 00:00:00 2001 From: Kye Date: Fri, 3 Nov 2023 15:44:06 -0400 Subject: [PATCH] tests --- swarms/models/bioclip.py | 1 - tests/models/huggingface.py | 10 +++++++--- 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/swarms/models/bioclip.py b/swarms/models/bioclip.py index 937634e3..318de290 100644 --- a/swarms/models/bioclip.py +++ b/swarms/models/bioclip.py @@ -42,7 +42,6 @@ Please refer to the corresponding paper, "Large-Scale Domain-Specific Pretrainin """ import open_clip -import glob import torch from PIL import Image import matplotlib.pyplot as plt diff --git a/tests/models/huggingface.py b/tests/models/huggingface.py index 1bb44bed..847ced06 100644 --- a/tests/models/huggingface.py +++ b/tests/models/huggingface.py @@ -1,7 +1,11 @@ -import torch +from unittest.mock import MagicMock, patch + import pytest -from unittest.mock import patch, MagicMock -from swarms.models.huggingface import HuggingfaceLLM # Replace with the actual import path +import torch + +from swarms.models.huggingface import ( + HuggingfaceLLM, # Replace with the actual import path +) # Fixture for the class instance