fix: remove accidental pytest dependency

SauravMaheshkar · Dec 29, 2024 · 0c6f3b2 · 0c6f3b2
1 parent e1549c1
commit 0c6f3b2
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/sam2/utils/__init__.py b/sam2/utils/__init__.py
@@ -3,5 +3,3 @@
 
 # This source code is licensed under the license found in the
 # LICENSE file in the root directory of this source tree.
-
-from .download import download_weights
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -5,7 +5,7 @@
 from sam2.automatic_mask_generator import SAM2AutomaticMaskGenerator
 from sam2.build_sam import build_sam2, build_sam2_video_predictor
 from sam2.sam2_image_predictor import SAM2ImagePredictor
-from sam2.utils import download_weights
+from sam2.utils.download import download_weights
 from sam2.utils.misc import variant_to_config_mapping
 
 

diff --git a/tests/test_build_model.py b/tests/test_build_model.py
@@ -2,7 +2,7 @@
 import torch
 
 from sam2.build_sam import build_sam2
-from sam2.utils import download_weights
+from sam2.utils.download import download_weights
 from sam2.utils.misc import variant_to_config_mapping
Original file line number	Diff line number	Diff line change
Expand Up		@@ -3,5 +3,3 @@

		# This source code is licensed under the license found in the
		# LICENSE file in the root directory of this source tree.

		from .download import download_weights