Skip to content

Commit 41e6ac1

Browse files
committed
improved docs; added new methods to init
1 parent be4227b commit 41e6ac1

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

dlclibrary/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,8 @@
1111

1212
from dlclibrary.dlcmodelzoo.modelzoo_download import (
1313
download_huggingface_model,
14+
get_available_detectors,
15+
get_available_models,
1416
parse_available_supermodels,
1517
)
1618
from dlclibrary.version import __version__, VERSION

dlclibrary/dlcmodelzoo/modelzoo_download.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -86,15 +86,17 @@ def parse_available_supermodels():
8686

8787

8888
def get_available_detectors(dataset: str) -> list[str]:
89-
"""
89+
""" Only for PyTorch models.
90+
9091
Returns:
9192
The detectors available for the dataset.
9293
"""
9394
return list(_load_pytorch_dataset_models(dataset)["detectors"].keys())
9495

9596

9697
def get_available_models(dataset: str) -> list[str]:
97-
"""
98+
""" Only for PyTorch models.
99+
98100
Returns:
99101
The pose models available for the dataset.
100102
"""

0 commit comments

Comments
 (0)