Minor Results.to_sql cleanup (#19081)

Co-authored-by: Muhammad Rizwan Munawar <muhammadrizwanmunawar123@gmail.com> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2025-02-07 11:19:58 +08:00 · 2025-02-07 11:19:58 +08:00 · 116813d3c0
commit 116813d3c0
parent c526a652ab
2 changed files with 9 additions and 20 deletions
--- a/ultralytics/utils/checks.py
+++ b/ultralytics/utils/checks.py
@ -131,7 +131,7 @@ def check_imgsz(imgsz, stride=32, min_dim=1, max_dim=2, floor=0):
        floor (int): Minimum allowed value for image size.

    Returns:
-        (List[int]): Updated image size.
+        (List[int] | int): Updated image size.
    """
    # Convert stride to integer if it is a tensor
    stride = int(stride.max() if isinstance(stride, torch.Tensor) else stride)