diff --git a/sam2/utils/__init__.py b/sam2/utils/__init__.py index d4d9f56d9..5277f4615 100644 --- 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 index d51fe43a4..2821bbbc0 100644 --- 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 index cde773189..71c9c9083 100644 --- 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