alma/pyalma
1
0
Fork 0
Code Issues 5 Pull Requests Releases Wiki Activity

merged pass_worker_id_to_run_method into master

Browse Source
This commit is contained in:
Tom Krüger
2024-01-29 19:59:57 +01:00
parent 5ae3e11818
commit 00ad1e2a10
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
alma/experiment.py
View File

@@ -46,7 +46,7 @@ class Dispatcher (threading.Thread):
self.__exp_mod = exp_mod self.__exp_mod = exp_mod
for i in range(self.__num_workers): for i in range(self.__num_workers):
self.__workers.append(Worker(exp_mod, exp_plan)) self.__workers.append(Worker(exp_mod, exp_plan, i))
def run(self): def run(self):
for worker in self.__workers: for worker in self.__workers:
@@ -93,11 +93,12 @@ class Dispatcher (threading.Thread):
class Worker (multiprocessing.Process): class Worker (multiprocessing.Process):
def __init__(self, exp_mod, exp_plan): def __init__(self, exp_mod, exp_plan, id):
multiprocessing.Process.__init__(self) multiprocessing.Process.__init__(self)
self.__exp_mod = exp_mod self.__exp_mod = exp_mod
self.__exp_plan = exp_plan self.__exp_plan = exp_plan
self.__id = id
def run(self): def run(self):
instance = self.__exp_plan.next() instance = self.__exp_plan.next()
@@ -110,7 +111,8 @@ class Worker (multiprocessing.Process):
instance, instance,
data data
), ),
instance_state) instance_state,
worker_id=self.__id)
self.__exp_plan.done_with(instance) self.__exp_plan.done_with(instance)