(self, z_loga, type_name, parameter_per_dim, size, shape)
| 127 | |
| 128 | # ! init the z_logas |
| 129 | def add_one_module(self, z_loga, type_name, parameter_per_dim, size, shape): |
| 130 | self.types.append(type_name) |
| 131 | self.z_logas[type_name] = z_loga |
| 132 | self.parameters_per_dim[type_name] = parameter_per_dim |
| 133 | self.sizes[type_name] = size |
| 134 | self.shapes[type_name] = shape |
| 135 | |
| 136 | def constrain_parameters(self): |
| 137 | for key in self.z_logas: |
no outgoing calls
no test coverage detected