(self, obs_dim_dict, module_config_dict)
| 4 | |
| 5 | class BaseModule(nn.Module): |
| 6 | def __init__(self, obs_dim_dict, module_config_dict): |
| 7 | super(BaseModule, self).__init__() |
| 8 | self.obs_dim_dict = obs_dim_dict |
| 9 | self.module_config_dict = module_config_dict |
| 10 | self.history_length = module_config_dict.get('history_length', {}) |
| 11 | |
| 12 | self._calculate_input_dim() |
| 13 | self._calculate_output_dim() |
| 14 | self._build_network_layer(self.module_config_dict.layer_config) |
| 15 | |
| 16 | def _calculate_input_dim(self): |
| 17 | # calculate input dimension based on the input specifications |
nothing calls this directly
no test coverage detected