Merge pull request #533 from roboflow/assets-fix

Assets fix
This commit is contained in:
Piotr Skalski 2023-10-24 16:16:30 +02:00 committed by GitHub
commit b9dbbbc0ec
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 13 additions and 6 deletions

View File

@ -3,7 +3,7 @@ that you can use in your demos.
## install extra
To install the Supervision assets utility, you can use pip. This utility is available
To install the Supervision assets utility, you can use `pip`. This utility is available
as an extra within the Supervision package.
!!! example "pip install"

View File

@ -1,6 +1,6 @@
[tool.poetry]
name = "supervision"
version = "0.17.0rc1"
version = "0.17.0rc2"
description = "A set of easy-to-use utils that will come in handy in any Computer Vision project"
authors = ["Piotr Skalski <piotr.skalski92@gmail.com>"]
maintainers = ["Piotr Skalski <piotr.skalski92@gmail.com>"]

View File

@ -1,2 +1,2 @@
from .downloader import download_assets
from .list import VideoAssets
from supervision.assets.downloader import download_assets
from supervision.assets.list import VideoAssets

View File

@ -4,8 +4,7 @@ from pathlib import Path
from shutil import copyfileobj
from typing import Union
from supervision.assets import VideoAssets
from supervision.assets.list import VIDEO_ASSETS
from supervision.assets.list import VIDEO_ASSETS, VideoAssets
try:
from requests import get
@ -51,6 +50,14 @@ def download_assets(asset_name: Union[VideoAssets, str]) -> str:
Returns:
str: The filename of the downloaded asset.
Example:
```python
>>> from supervision.assets import download_assets, VideoAssets
>>> download_assets(VideoAssets.VEHICLES)
"vehicles.mp4"
```
"""
filename = asset_name.value if isinstance(asset_name, VideoAssets) else asset_name