From b638c4ed9a24270a6875cdd47d9eeda99204ef5a Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Thu, 24 Aug 2023 15:15:42 +0200
Subject: [PATCH] Update HUB repo to `main` branch (#4549)

Co-authored-by: Yonghye Kwon <developer.0hye@gmail.com>
Co-authored-by: andresinsitu <andres.rodriguez@ingenieriainsitu.com>
---
 docs/hub/datasets.md      | 2 +-
 tests/test_python.py      | 2 +-
 ultralytics/data/utils.py | 1 +
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/docs/hub/datasets.md b/docs/hub/datasets.md
index e2ac4654..d5e23443 100644
--- a/docs/hub/datasets.md
+++ b/docs/hub/datasets.md
@@ -23,7 +23,7 @@ For example, if your dataset is called "coco8", as our [COCO8](https://docs.ultr
 zip -r coco8.zip coco8
 ```
 
-You can download our [COCO8](https://github.com/ultralytics/hub/blob/master/example_datasets/coco8.zip) example dataset and unzip it to see exactly how to structure your dataset.
+You can download our [COCO8](https://github.com/ultralytics/hub/blob/main/example_datasets/coco8.zip) example dataset and unzip it to see exactly how to structure your dataset.
 
 <p align="center">
   <img  src="https://raw.githubusercontent.com/ultralytics/assets/main/docs/hub/datasets/hub_upload_dataset_1.jpg" alt="COCO8 Dataset Structure" width="80%" />
diff --git a/tests/test_python.py b/tests/test_python.py
index 2a0d556f..1fc5fc81 100644
--- a/tests/test_python.py
+++ b/tests/test_python.py
@@ -275,7 +275,7 @@ def test_data_utils():
     # from ultralytics.utils.files import WorkingDirectory
     # with WorkingDirectory(ROOT.parent / 'tests'):
 
-    download('https://github.com/ultralytics/hub/raw/master/example_datasets/coco8.zip', unzip=False)
+    download('https://github.com/ultralytics/hub/raw/main/example_datasets/coco8.zip', unzip=False)
     shutil.move('coco8.zip', TMP)
     stats = HUBDatasetStats(TMP / 'coco8.zip', task='detect')
     stats.get_json(save=True)
diff --git a/ultralytics/data/utils.py b/ultralytics/data/utils.py
index 6bc2e30d..05b291ba 100644
--- a/ultralytics/data/utils.py
+++ b/ultralytics/data/utils.py
@@ -373,6 +373,7 @@ class HUBDatasetStats:
         autodownload (bool): Attempt to download dataset if not found locally. Default is False.
 
     Example:
+        Download *.zip files from i.e. https://github.com/ultralytics/hub/raw/main/example_datasets/coco8.zip.
         ```python
         from ultralytics.data.utils import HUBDatasetStats