def prepath_to_info_space(prepath):
'''Create info_space from prepath such that it returns the same prepath with spec'''
from slm_lab.experiment.monitor import InfoSpace
experiment_ts = prepath_to_experiment_ts(prepath)
trial_index, session_index = prepath_to_idxs(prepath)
# create info_space for prepath
info_space = InfoSpace()
info_space.experiment_ts = experiment_ts
info_space.set('experiment', 0)
info_space.set('trial', trial_index)
info_space.set('session', session_index)
return info_space
def mock_spec_info_space(predir, trial_index=None, session_index=None):
'''Helper for retro analysis to build mock info_space and spec'''
from slm_lab.experiment.monitor import InfoSpace
_, _, _, spec_name, experiment_ts, _ = util.prepath_split(predir)
info_space = InfoSpace()
info_space.experiment_ts = experiment_ts
info_space.set('experiment', 0)
if trial_index is None:
filepath = f'{predir}/{spec_name}_spec.json'
else:
info_space.set('trial', trial_index)
filepath = f'{predir}/{spec_name}_t{trial_index}_spec.json'
if session_index is not None:
info_space.set('session', session_index)
spec = util.read(filepath)
return spec, info_space
def mock_info_space_spec(predir, trial_index=None, session_index=None):
'''Helper for retro analysis to build mock info_space and spec'''
from slm_lab.experiment.monitor import InfoSpace
spec_name = util.prepath_to_spec_name(predir)
experiment_ts = util.prepath_to_experiment_ts(predir)
info_space = InfoSpace()
info_space.experiment_ts = experiment_ts
info_space.set('experiment', 0)
if trial_index is None:
filepath = f'{predir}/{spec_name}_spec.json'
else:
info_space.set('trial', trial_index)
filepath = f'{predir}/{spec_name}_t{trial_index}_spec.json'
if session_index is not None:
info_space.set('session', session_index)
spec = util.read(filepath)
return spec, info_space