From 18de47cda6545ed30cdf5bfbf04b498752a92150 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 3 Sep 2024 11:10:09 +0200
Subject: [PATCH] Update `test_workflow` to ONNX (#15974)

---
 tests/test_python.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_python.py b/tests/test_python.py
index f15dd48eff..8fbab54cf2 100644
--- a/tests/test_python.py
+++ b/tests/test_python.py
@@ -202,7 +202,7 @@ def test_workflow():
     model.train(data="coco8.yaml", epochs=1, imgsz=32, optimizer="SGD")
     model.val(imgsz=32)
     model.predict(SOURCE, imgsz=32)
-    model.export(format="torchscript")
+    model.export(format="onnx")
 
 
 def test_predict_callback_and_setup():