remove scopre

landing-ai · Oct 15, 2024 · fc07ea0 · fc07ea0
1 parent c000a1b
commit fc07ea0
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/tests/models/florence2_ft/conftest.py b/tests/models/florence2_ft/conftest.py
@@ -13,18 +13,14 @@
 logging.basicConfig(level=logging.INFO)
 
 
-@pytest.fixture(scope="function")
+@pytest.fixture
 def large_model():
-    model = Florence2Ft(Florence2ModelName.FLORENCE_2_LARGE)
-    yield model
-    model.load_base()
+    return Florence2Ft(Florence2ModelName.FLORENCE_2_LARGE)
 
 
-@pytest.fixture(scope="function")
+@pytest.fixture
 def small_model():
-    model = Florence2Ft(Florence2ModelName.FLORENCE_2_BASE_FT)
-    yield model
-    model.load_base()
+    return Florence2Ft(Florence2ModelName.FLORENCE_2_BASE_FT)
 
 
 @pytest.fixture