Skip to content

Commit 22bebf0

Browse files
committed
Remove benchmark exports from __init__
Stop exporting benchmark_videos and download_benchmarking_data from dlclive.__init__.py and update tests to import them directly from dlclive.benchmark. Keeps the package __init__ smaller and avoids importing the benchmark module at package import time.
1 parent 398eae2 commit 22bebf0

2 files changed

Lines changed: 1 addition & 2 deletions

File tree

dlclive/__init__.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111

1212
_AVAILABLE_BACKENDS = get_available_backends()
1313

14-
from dlclive.benchmark import benchmark_videos, download_benchmarking_data
1514
from dlclive.display import Display
1615
from dlclive.dlclive import DLCLive
1716
from dlclive.engine import Engine

tests/test_benchmark_script.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
import pytest
44

5-
from dlclive import benchmark_videos, download_benchmarking_data
5+
from dlclive.benchmark import benchmark_videos, download_benchmarking_data
66
from dlclive.engine import Engine
77

88

0 commit comments

Comments
 (0)