diff --git a/tests/conftest.py b/tests/conftest.py index 58c07b79..ba3d8346 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -3,7 +3,7 @@ import shutil from pathlib import Path -TMP = Path(__file__).resolve().parent / "tmp" # temp directory for test files +from tests import TMP def pytest_addoption(parser): diff --git a/tests/test_cli.py b/tests/test_cli.py index 71bc2e48..24eab756 100644 --- a/tests/test_cli.py +++ b/tests/test_cli.py @@ -7,7 +7,7 @@ import pytest from ultralytics.cfg import TASK2DATA, TASK2MODEL, TASKS from ultralytics.utils import ASSETS, WEIGHTS_DIR, checks -from . import CUDA_DEVICE_COUNT, CUDA_IS_AVAILABLE +from tests import CUDA_DEVICE_COUNT, CUDA_IS_AVAILABLE # Constants TASK_MODEL_DATA = [(task, WEIGHTS_DIR / TASK2MODEL[task], TASK2DATA[task]) for task in TASKS] diff --git a/tests/test_cuda.py b/tests/test_cuda.py index 4e8d4c68..e3de070d 100644 --- a/tests/test_cuda.py +++ b/tests/test_cuda.py @@ -10,7 +10,7 @@ from ultralytics import YOLO from ultralytics.utils import ASSETS, WEIGHTS_DIR from ultralytics.cfg import TASK2DATA, TASK2MODEL, TASKS -from . import CUDA_DEVICE_COUNT, CUDA_IS_AVAILABLE, MODEL, SOURCE +from tests import CUDA_DEVICE_COUNT, CUDA_IS_AVAILABLE, MODEL, SOURCE def test_checks(): diff --git a/tests/test_engine.py b/tests/test_engine.py index 73a3b744..cc21a4fe 100644 --- a/tests/test_engine.py +++ b/tests/test_engine.py @@ -9,7 +9,7 @@ from ultralytics.engine.exporter import Exporter from ultralytics.models.yolo import classify, detect, segment from ultralytics.utils import ASSETS, DEFAULT_CFG, WEIGHTS_DIR -from . import MODEL +from tests import MODEL def test_func(*args): # noqa diff --git a/tests/test_exports.py b/tests/test_exports.py index a9b5d24a..788861e6 100644 --- a/tests/test_exports.py +++ b/tests/test_exports.py @@ -18,8 +18,7 @@ from ultralytics.utils import ( checks, ) from ultralytics.utils.torch_utils import TORCH_1_9, TORCH_1_13 - -from . import MODEL, SOURCE +from tests import MODEL, SOURCE def test_export_torchscript(): diff --git a/tests/test_integrations.py b/tests/test_integrations.py index d0126a0f..74ebc690 100644 --- a/tests/test_integrations.py +++ b/tests/test_integrations.py @@ -12,7 +12,7 @@ from ultralytics import YOLO, download from ultralytics.utils import DATASETS_DIR, SETTINGS from ultralytics.utils.checks import check_requirements -from . import MODEL, SOURCE, TMP +from tests import MODEL, SOURCE, TMP @pytest.mark.skipif(not check_requirements("ray", install=False), reason="ray[tune] not installed") diff --git a/tests/test_python.py b/tests/test_python.py index 0f1ba251..fb6f36ef 100644 --- a/tests/test_python.py +++ b/tests/test_python.py @@ -27,7 +27,7 @@ from ultralytics.utils import ( ) from ultralytics.utils.downloads import download, is_url from ultralytics.utils.torch_utils import TORCH_1_9 -from . import CFG, IS_TMP_WRITEABLE, MODEL, SOURCE, TMP +from tests import CFG, IS_TMP_WRITEABLE, MODEL, SOURCE, TMP def test_model_forward(): diff --git a/ultralytics/__init__.py b/ultralytics/__init__.py index b7beda50..8d1a571d 100644 --- a/ultralytics/__init__.py +++ b/ultralytics/__init__.py @@ -1,6 +1,6 @@ # Ultralytics YOLO 🚀, AGPL-3.0 license -__version__ = "8.2.18" +__version__ = "8.2.19" from ultralytics.data.explorer.explorer import Explorer from ultralytics.models import RTDETR, SAM, YOLO, YOLOWorld