diff --git a/docs/dataset/core.md b/docs/dataset/core.md index 0bafd52a..0713c9dc 100644 --- a/docs/dataset/core.md +++ b/docs/dataset/core.md @@ -1,8 +1,8 @@ !!! warning - `Dataset` API is still fluid and may change. If you use Dataset in your project until further notice, freeze the - `supervision` version in your `requirements.txt`. + Dataset API is still fluid and may change. If you use Dataset API in your project until further notice, freeze the + `supervision` version in your `requirements.txt` or `setup.py`. -## Dataset +## DetectionDataset :::supervision.dataset.core.DetectionDataset \ No newline at end of file diff --git a/supervision/dataset/core.py b/supervision/dataset/core.py index f1ccb3d5..f5c85b24 100644 --- a/supervision/dataset/core.py +++ b/supervision/dataset/core.py @@ -29,7 +29,7 @@ class BaseDataset: @dataclass class DetectionDataset(BaseDataset): """ - Dataclass containing information about the dataset. + Dataclass containing information about object detection dataset. Attributes: classes (List[str]): List containing dataset class names.