Skip to content

Commit a41fee0

Browse files
committed
use a set for task2name
1 parent a90447e commit a41fee0

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

custom_components/pyscript/function.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -166,13 +166,11 @@ async def task_unique(name, kill_me=False):
166166
if name in cls.unique_name2task:
167167
task = cls.unique_name2task[name]
168168
if task in cls.unique_task2name:
169-
if name in cls.unique_task2name[task]:
170-
cls.unique_task2name[task].remove(name)
169+
cls.unique_task2name[task].discard(name)
171170
cls.unique_name2task[name] = curr_task
172171
if curr_task not in cls.unique_task2name:
173-
cls.unique_task2name[curr_task] = []
174-
if name not in cls.unique_task2name[curr_task]:
175-
cls.unique_task2name[curr_task].append(name)
172+
cls.unique_task2name[curr_task] = set()
173+
cls.unique_task2name[curr_task].add(name)
176174

177175
return task_unique
178176

0 commit comments

Comments
 (0)