occheung/artiq
1
0
Fork 0
forked from M-Labs/artiq
Code Pull Requests Activity

language/environment: disable processors by default

Browse Source
This commit is contained in:
Sebastien Bourdeauducq 2015-12-22 11:45:34 +08:00
parent a250b5da21
commit fc299ca918
2 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
artiq/language/environment.py
View File

@ -122,13 +122,14 @@ class HasEnvironment:
"""Provides methods to manage the environment of an experiment (devices, """Provides methods to manage the environment of an experiment (devices,
parameters, results, arguments).""" parameters, results, arguments)."""
def __init__(self, device_mgr=None, dataset_mgr=None, *, parent=None, def __init__(self, device_mgr=None, dataset_mgr=None, *, parent=None,
default_arg_none=False, **kwargs): default_arg_none=False, enable_processors=False, **kwargs):
self.requested_args = OrderedDict() self.requested_args = OrderedDict()
self.__device_mgr = device_mgr self.__device_mgr = device_mgr
self.__dataset_mgr = dataset_mgr self.__dataset_mgr = dataset_mgr
self.__parent = parent self.__parent = parent
self.__default_arg_none = default_arg_none self.__default_arg_none = default_arg_none
self.__enable_processors = enable_processors
self.__kwargs = kwargs self.__kwargs = kwargs
self.__in_build = True self.__in_build = True
@ -187,7 +188,10 @@ class HasEnvironment:
return None return None
else: else:
raise raise
if self.__enable_processors:
return processor.process(argval) return processor.process(argval)
else:
return argval
def setattr_argument(self, key, processor=None, group=None): def setattr_argument(self, key, processor=None, group=None):
"""Sets an argument as attribute. The names of the argument and of the """Sets an argument as attribute. The names of the argument and of the

7
artiq/master/worker_impl.py
View File

@ -153,7 +153,9 @@ def examine(device_mgr, dataset_mgr, file):
name = exp_class.__doc__.splitlines()[0].strip() name = exp_class.__doc__.splitlines()[0].strip()
if name[-1] == ".": if name[-1] == ".":
name = name[:-1] name = name[:-1]
exp_inst = exp_class(device_mgr, dataset_mgr, default_arg_none=True) exp_inst = exp_class(device_mgr, dataset_mgr,
default_arg_none=True,
enable_processors=True)
arginfo = OrderedDict( arginfo = OrderedDict(
(k, (proc.describe(), group)) (k, (proc.describe(), group))
for k, (proc, group) in exp_inst.requested_args.items()) for k, (proc, group) in exp_inst.requested_args.items())
@ -199,7 +201,8 @@ def main():
exp = get_exp(experiment_file, expid["class_name"]) exp = get_exp(experiment_file, expid["class_name"])
device_mgr.virtual_devices["scheduler"].set_run_info( device_mgr.virtual_devices["scheduler"].set_run_info(
rid, obj["pipeline_name"], expid, obj["priority"]) rid, obj["pipeline_name"], expid, obj["priority"])
exp_inst = exp(device_mgr, dataset_mgr, exp_inst = exp(
device_mgr, dataset_mgr, enable_processors=True,
**expid["arguments"]) **expid["arguments"])
put_object({"action": "completed"}) put_object({"action": "completed"})
elif action == "prepare": elif action == "prepare":