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