test: Add unit test setup for GradioUI class

huggingface · Feb 18, 2025 · 8809bd9 · 8809bd9
1 parent 6f6d7cb
commit 8809bd9
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/test_gradio_ui.py b/tests/test_gradio_ui.py
@@ -16,13 +16,12 @@
 import os
 import shutil
 import tempfile
-import unittest
 from unittest.mock import Mock, patch
 
 from smolagents.gradio_ui import GradioUI
 
 
-class GradioUITester(unittest.TestCase):
+class GradioUITester:
     def setUp(self):
         """Initialize test environment"""
         self.temp_dir = tempfile.mkdtemp()