def _agent_env(self, env=None):
return job.agent_env(
env=(env or PKDict()).pksetdefault(
PYKERN_PKDEBUG_WANT_PID_TIME='1',
SIREPO_PKCLI_JOB_AGENT_SUPERVISOR_SIM_DB_FILE_URI=job.
supervisor_file_uri(
self.cfg.supervisor_uri,
job.SIM_DB_FILE_URI,
sirepo.simulation_db.USER_ROOT_DIR,
self.uid,
),
SIREPO_PKCLI_JOB_AGENT_SUPERVISOR_SIM_DB_FILE_TOKEN=self.
_sim_db_file_token,
SIREPO_PKCLI_JOB_AGENT_AGENT_ID=self._agentId,
SIREPO_PKCLI_JOB_AGENT_START_DELAY=self.get(
'_agent_start_delay', 0),
SIREPO_PKCLI_JOB_AGENT_SUPERVISOR_URI=self.cfg.supervisor_uri.
replace(
#TODO(robnagler) figure out why we need ws (wss, implicit)
'http',
'ws',
1,
) + job.AGENT_URI),
uid=self.uid,
)
def job_cmd_env(self, env=None):
return job.agent_env(
env=(env or PKDict()).pksetdefault(
SIREPO_MPI_CORES=self.msg.get('mpiCores', 1),
SIREPO_SIM_DATA_LIB_FILE_URI=self.msg.get('libFileUri', ''),
),
)
def job_cmd_env(self, env=None):
return job.agent_env(env=(env or PKDict()).pksetdefault(
SIREPO_MPI_CORES=self.msg.get('mpiCores', 1),
SIREPO_SIM_DATA_LIB_FILE_URI=self._lib_file_uri,
SIREPO_SIM_DATA_LIB_FILE_LIST=self._lib_file_list_f,
SIREPO_SIM_DATA_SUPERVISOR_SIM_DB_FILE_URI=cfg.
supervisor_sim_db_file_uri,
SIREPO_SIM_DATA_SUPERVISOR_SIM_DB_FILE_TOKEN=cfg.
supervisor_sim_db_file_token,
), )
def _agent_env(self, env=None):
return job.agent_env(
env=(env or PKDict()).pksetdefault(
PYKERN_PKDEBUG_WANT_PID_TIME='1',
SIREPO_PKCLI_JOB_AGENT_AGENT_ID=self._agentId,
SIREPO_PKCLI_JOB_AGENT_SUPERVISOR_URI=self.cfg.supervisor_uri.replace(
#TODO(robnagler) figure out why we need ws (wss, implicit)
'http',
'ws',
1,
) + job.AGENT_URI
),
uid=self.uid,
)
def job_cmd_env(self, env=None):
return job.agent_env(env=(env or PKDict()).pksetdefault(
SIREPO_MPI_CORES=self.msg.get('mpiCores', 1),
SIREPO_SIM_DATA_LIB_FILE_URI=self._lib_file_uri,
SIREPO_SIM_DATA_LIB_FILE_LIST=self._lib_file_list_f,
), )