Except YouTube Error 429: Too Many Requests (#13167)
Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
This commit is contained in:
parent
25c9f77c8f
commit
15975e4702
1 changed files with 6 additions and 1 deletions
|
|
@ -1,6 +1,7 @@
|
||||||
# Ultralytics YOLO 🚀, AGPL-3.0 license
|
# Ultralytics YOLO 🚀, AGPL-3.0 license
|
||||||
|
|
||||||
import contextlib
|
import contextlib
|
||||||
|
import urllib
|
||||||
from copy import copy
|
from copy import copy
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
|
|
@ -19,6 +20,7 @@ from ultralytics.utils import (
|
||||||
ASSETS,
|
ASSETS,
|
||||||
DEFAULT_CFG,
|
DEFAULT_CFG,
|
||||||
DEFAULT_CFG_PATH,
|
DEFAULT_CFG_PATH,
|
||||||
|
LOGGER,
|
||||||
ONLINE,
|
ONLINE,
|
||||||
ROOT,
|
ROOT,
|
||||||
WEIGHTS_DIR,
|
WEIGHTS_DIR,
|
||||||
|
|
@ -136,7 +138,10 @@ def test_youtube():
|
||||||
Note: ConnectionError may occur during this test due to network instability or YouTube server availability.
|
Note: ConnectionError may occur during this test due to network instability or YouTube server availability.
|
||||||
"""
|
"""
|
||||||
model = YOLO(MODEL)
|
model = YOLO(MODEL)
|
||||||
|
try:
|
||||||
model.predict("https://youtu.be/G17sBkb38XQ", imgsz=96, save=True)
|
model.predict("https://youtu.be/G17sBkb38XQ", imgsz=96, save=True)
|
||||||
|
except urllib.error.HTTPError as e: # handle 'urllib.error.HTTPError: HTTP Error 429: Too Many Requests'
|
||||||
|
LOGGER.warning(f"WARNING: YouTube Test Error: {e}")
|
||||||
|
|
||||||
|
|
||||||
@pytest.mark.skipif(not ONLINE, reason="environment is offline")
|
@pytest.mark.skipif(not ONLINE, reason="environment is offline")
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue