ultralytics 8.2.87 Ray shutdown() workers after tuning (#15976)

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="onnx")
+    model.export(format="torchscript")
 
 
 def test_predict_callback_and_setup():

ultralytics/__init__.py

@@ -1,6 +1,6 @@
 # Ultralytics YOLO 🚀, AGPL-3.0 license
 
-__version__ = "8.2.86"
+__version__ = "8.2.87"
 
 import os
 

ultralytics/utils/tuner.py

@@ -143,5 +143,10 @@ def run_ray_tune(
     # Run the hyperparameter search
     tuner.fit()
 
-    # Return the results of the hyperparameter search
-    return tuner.get_results()
+    # Get the results of the hyperparameter search
+    results = tuner.get_results()
+
+    # Shut down Ray to clean up workers
+    ray.shutdown()
+
+    return results