Skip to content
Snippets Groups Projects
Commit f09a093b authored by Kenny Sharma's avatar Kenny Sharma
Browse files

[hotfix] Add reservation support for launching CLE.

This patch updates the distributed NEST launcher to be in line with
the normal CLELauncher, which now requires a reservation parameter
to be passed into the init method.

Change-Id: Id6d52ea7b35b70205e35450e770cd45e9ccfc8a1
parent ceaeade4
No related branches found
No related tags found
No related merge requests found
...@@ -20,7 +20,9 @@ class MUSICLauncher(object): ...@@ -20,7 +20,9 @@ class MUSICLauncher(object):
requested brain processes. requested brain processes.
""" """
def __init__(self, exd_file, bibi_file, env_file, exp_path, server_host, sim_id, timeout): # pylint: disable=too-many-arguments
def __init__(self, exd_file, bibi_file, env_file, exp_path, server_host, reservation,
sim_id, timeout):
""" """
Store all experiment configuration parameters so that they can be propagated Store all experiment configuration parameters so that they can be propagated
to the remote hosts. to the remote hosts.
...@@ -30,6 +32,7 @@ class MUSICLauncher(object): ...@@ -30,6 +32,7 @@ class MUSICLauncher(object):
:param env_file Absolute path to the simulation environment file. :param env_file Absolute path to the simulation environment file.
:param exp_path Absolute path to the base dir of the experiment files. :param exp_path Absolute path to the base dir of the experiment files.
:param server_host Target Gazebo/brain process host (e.g. local or lugano) :param server_host Target Gazebo/brain process host (e.g. local or lugano)
:param reservation Reservation string for cluster backend (None is a valid option)
:param sim_id The id of the simulation/experiment to be launched. :param sim_id The id of the simulation/experiment to be launched.
:param timeout The default simulation timeout (time initially allocated). :param timeout The default simulation timeout (time initially allocated).
""" """
...@@ -39,6 +42,7 @@ class MUSICLauncher(object): ...@@ -39,6 +42,7 @@ class MUSICLauncher(object):
self._env_file = env_file self._env_file = env_file
self._exp_path = exp_path self._exp_path = exp_path
self._server_host = server_host self._server_host = server_host
self._reservation = reservation if reservation else ''
self._sim_id = sim_id self._sim_id = sim_id
self._timeout = timeout self._timeout = timeout
...@@ -69,6 +73,7 @@ class MUSICLauncher(object): ...@@ -69,6 +73,7 @@ class MUSICLauncher(object):
'--environment={}'.format(self._env_file), '--environment={}'.format(self._env_file),
'--experiment-path={}'.format(self._exp_path), '--experiment-path={}'.format(self._exp_path),
'--gzserver-host={}'.format(self._server_host), '--gzserver-host={}'.format(self._server_host),
'--reservation={}'.format(self._reservation),
'--sim-id={}'.format(self._sim_id), '--sim-id={}'.format(self._sim_id),
'--timeout={}'.format(str(self._timeout).replace(' ', '_')), '--timeout={}'.format(str(self._timeout).replace(' ', '_')),
'--music'], '--music'],
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment