(self, path_to_conf_file)
| 15 | self.setup(path_to_conf_file) |
| 16 | |
| 17 | def setup(self, path_to_conf_file): |
| 18 | cfg = OmegaConf.load(path_to_conf_file) |
| 19 | cfg = OmegaConf.to_container(cfg) |
| 20 | |
| 21 | self._obs_configs = cfg['obs_configs'] |
| 22 | self._train_cfg = cfg['training'] |
| 23 | |
| 24 | |
| 25 | self._policy_class = load_entry_point(cfg['policy']['entry_point']) |
| 26 | self._policy_kwargs = cfg['policy']['kwargs'] |
| 27 | if self._ckpt is None: |
| 28 | self._policy = None |
| 29 | else: |
| 30 | self._policy, self._train_cfg['kwargs'] = self._policy_class.load(self._ckpt) |
| 31 | self._policy = self._policy.eval() |
| 32 | |
| 33 | self._wrapper_class = load_entry_point(cfg['env_wrapper']['entry_point']) |
| 34 | self._wrapper_kwargs = cfg['env_wrapper']['kwargs'] |
| 35 | |
| 36 | def run_step(self, input_data, timestamp): |
| 37 | input_data = copy.deepcopy(input_data) |
no test coverage detected