Skip to content

Commit 6a06099

Browse files
Fix remark and add docstrings
1 parent 4c4b34f commit 6a06099

File tree

1 file changed

+17
-9
lines changed

1 file changed

+17
-9
lines changed

dpctl/_device_selection.py

Lines changed: 17 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -13,15 +13,23 @@ def select_device_with_aspects(aspect_list, deny_list=[]):
1313
selected_dev = None
1414

1515
for dev in devs:
16-
aspect_status = True
17-
for asp in aspect_list:
18-
has_aspect = "dev.has_aspect_" + asp
19-
if not eval(has_aspect):
20-
aspect_status = False
21-
for deny in deny_list:
22-
has_aspect = "dev.has_aspect_" + deny
23-
if eval(has_aspect):
24-
aspect_status = False
16+
# aspect_status = True
17+
# for asp in aspect_list:
18+
# has_aspect = "dev.has_aspect_" + asp
19+
# if not eval(has_aspect):
20+
# aspect_status = False
21+
# for deny in deny_list:
22+
# has_aspect = "dev.has_aspect_" + deny
23+
# if eval(has_aspect):
24+
# aspect_status = False
25+
aspect_status = all(
26+
(getattr(dev, "has_aspect_" + asp) is True for asp in aspect_list)
27+
)
28+
aspect_status = aspect_status and not (
29+
any(
30+
(getattr(dev, "has_aspect_" + asp) is True for asp in deny_list)
31+
)
32+
)
2533
if aspect_status and dev.default_selector_score > max_score:
2634
max_score = dev.default_selector_score
2735
selected_dev = dev

0 commit comments

Comments
 (0)