From 8c28e0c3fee2fdd00297b90222d39d4c4db89772 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 22 Feb 2024 22:16:52 +0100 Subject: [PATCH] Add missing single-line docstrings (#8362) Signed-off-by: Glenn Jocher Co-authored-by: Johannes Kaisinger --- docs/build_docs.py | 2 +- tests/test_python.py | 2 ++ ultralytics/models/yolo/obb/train.py | 2 +- 3 files changed, 4 insertions(+), 2 deletions(-) diff --git a/docs/build_docs.py b/docs/build_docs.py index 71556781..ddcbcc57 100644 --- a/docs/build_docs.py +++ b/docs/build_docs.py @@ -115,7 +115,7 @@ def update_subdir_edit_links(subdir="", docs_url=""): def main(): - # Build the docs + """Builds docs, updates titles and edit links, and prints local server command.""" build_docs() # Update titles diff --git a/tests/test_python.py b/tests/test_python.py index d3b816b2..adfb5b03 100644 --- a/tests/test_python.py +++ b/tests/test_python.py @@ -555,6 +555,7 @@ def test_hub(): @pytest.fixture def image(): + """Loads an image from a predefined source using OpenCV.""" return cv2.imread(str(SOURCE)) @@ -568,6 +569,7 @@ def image(): ], ) def test_classify_transforms_train(image, auto_augment, erasing, force_color_jitter): + """Tests classification transforms during training with various augmentation settings.""" import torchvision.transforms as T from ultralytics.data.augment import classify_augmentations diff --git a/ultralytics/models/yolo/obb/train.py b/ultralytics/models/yolo/obb/train.py index 43ebaecd..40a35a99 100644 --- a/ultralytics/models/yolo/obb/train.py +++ b/ultralytics/models/yolo/obb/train.py @@ -15,7 +15,7 @@ class OBBTrainer(yolo.detect.DetectionTrainer): ```python from ultralytics.models.yolo.obb import OBBTrainer - args = dict(model='yolov8n-seg.pt', data='coco8-seg.yaml', epochs=3) + args = dict(model='yolov8n-obb.pt', data='dota8.yaml', epochs=3) trainer = OBBTrainer(overrides=args) trainer.train() ```