Skip to content

Commit a185715

Browse files
committed
use sorter module
Signed-off-by: sirutBuasai <sirutbuasai27@outlook.com>
1 parent dd21fcb commit a185715

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

docs/src/generate.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,6 @@
3535
sort_by_version,
3636
)
3737
from jinja2 import Template
38-
from sorter import accelerator_sorter, platform_sorter, repository_sorter
3938
from utils import (
4039
get_framework_order,
4140
load_jinja2,
@@ -45,7 +44,7 @@
4544
write_output,
4645
)
4746

48-
DEFAULT_TIEBREAKERS = [platform_sorter, accelerator_sorter]
47+
DEFAULT_TIEBREAKERS = [sorter_module.platform_sorter, sorter_module.accelerator_sorter]
4948

5049

5150
LOGGER = logging.getLogger(__name__)
@@ -104,7 +103,11 @@ def _generate_framework_index(
104103
for version in sorted_versions:
105104
sorted_images = sort_by_version(
106105
images_by_version[version],
107-
tiebreakers=[repository_sorter, platform_sorter, accelerator_sorter],
106+
tiebreakers=[
107+
sorter_module.repository_sorter,
108+
sorter_module.platform_sorter,
109+
sorter_module.accelerator_sorter,
110+
],
108111
)
109112

110113
supported = [img for img in sorted_images if img.is_supported]

0 commit comments

Comments
 (0)