| 206 | return action |
| 207 | |
| 208 | def get_init_kwargs(self) -> Dict[str, Any]: |
| 209 | init_kwargs = dict( |
| 210 | observation_space=self.observation_space, |
| 211 | action_space=self.action_space, |
| 212 | policy_head_arch=self.policy_head_arch, |
| 213 | value_head_arch=self.value_head_arch, |
| 214 | features_extractor_entry_point=self.features_extractor_entry_point, |
| 215 | features_extractor_kwargs=self.features_extractor_kwargs, |
| 216 | distribution_entry_point=self.distribution_entry_point, |
| 217 | distribution_kwargs=self.distribution_kwargs, |
| 218 | ) |
| 219 | return init_kwargs |
| 220 | |
| 221 | @classmethod |
| 222 | def load(cls, path): |