HongChuang/ultralytics-ascend
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 3

Add windows-latest to CI Tests (#11233)

Browse source 
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Burhan <62214284+Burhan-Q@users.noreply.github.com>
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
Glenn Jocher 2024-05-04 18:16:20 +02:00 • committed by GitHub
parent 28cb2f2213
commit 03da2944d7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 17 additions and 16 deletions
Download patch file Download diff file Expand all files Collapse all files

2
.github/workflows/ci.yaml vendored
View file

@ -174,7 +174,7 @@ jobs:
strategy: strategy:
fail-fast: false fail-fast: false
matrix: matrix:
os: [ubuntu-latest, macos-14] os: [ubuntu-latest, windows-latest, macos-14]
python-version: ["3.11"] python-version: ["3.11"]
torch: [latest] torch: [latest]
include: include:

31
tests/test_python.py
View file

@ -235,38 +235,39 @@ def test_export_coreml():
YOLO(MODEL).export(format="coreml", nms=True) YOLO(MODEL).export(format="coreml", nms=True)
def test_export_tflite(enabled=False): @pytest.mark.skipif(not LINUX, reason="Test disabled as TF suffers from install conflicts on Windows and macOS")
def test_export_tflite():
""" """
Test exporting the YOLO model to TFLite format. Test exporting the YOLO model to TFLite format.
Note TF suffers from install conflicts on Windows and macOS. Note TF suffers from install conflicts on Windows and macOS.
""" """
if enabled and LINUX: model = YOLO(MODEL)
model = YOLO(MODEL) f = model.export(format="tflite")
f = model.export(format="tflite") YOLO(f)(SOURCE)
YOLO(f)(SOURCE)
def test_export_pb(enabled=False): @pytest.mark.skipif(True, reason="Test disabled")
@pytest.mark.skipif(not LINUX, reason="TF suffers from install conflicts on Windows and macOS")
def test_export_pb():
""" """
Test exporting the YOLO model to *.pb format. Test exporting the YOLO model to *.pb format.
Note TF suffers from install conflicts on Windows and macOS. Note TF suffers from install conflicts on Windows and macOS.
""" """
if enabled and LINUX: model = YOLO(MODEL)
model = YOLO(MODEL) f = model.export(format="pb")
f = model.export(format="pb") YOLO(f)(SOURCE)
YOLO(f)(SOURCE)
def test_export_paddle(enabled=False): @pytest.mark.skipif(True, reason="Test disabled as Paddle protobuf and ONNX protobuf requirementsk conflict.")
def test_export_paddle():
""" """
Test exporting the YOLO model to Paddle format. Test exporting the YOLO model to Paddle format.
Note Paddle protobuf requirements conflicting with onnx protobuf requirements. Note Paddle protobuf requirements conflicting with onnx protobuf requirements.
""" """
if enabled: YOLO(MODEL).export(format="paddle")
YOLO(MODEL).export(format="paddle")
@pytest.mark.slow @pytest.mark.slow
@ -436,12 +437,12 @@ def test_utils_checks():
checks.git_describe(ROOT) checks.git_describe(ROOT)
checks.check_requirements() # check requirements.txt checks.check_requirements() # check requirements.txt
checks.check_imgsz([600, 600], max_dim=1) checks.check_imgsz([600, 600], max_dim=1)
checks.check_imshow() checks.check_imshow(warn=True)
checks.check_version("ultralytics", "8.0.0") checks.check_version("ultralytics", "8.0.0")
checks.print_args() checks.print_args()
# checks.check_imshow(warn=True)
@pytest.mark.skipif(WINDOWS, reason="Windows profiling is extremely slow (cause unknown)")
def test_utils_benchmarks(): def test_utils_benchmarks():
"""Test model benchmarking.""" """Test model benchmarking."""
from ultralytics.utils.benchmarks import ProfileModels from ultralytics.utils.benchmarks import ProfileModels