Register the var_name to the target (set of var names) Args: var_name: name (str) or list of names of the variable to register target: the set to register the var namee
(
self,
var_name: T.Union[str, T.List[str]],
target: T.Set[str],
)
| 1190 | self._register(var_name=var_name, target=self.losses_name) |
| 1191 | |
| 1192 | def _register( |
| 1193 | self, |
| 1194 | var_name: T.Union[str, T.List[str]], |
| 1195 | target: T.Set[str], |
| 1196 | ): |
| 1197 | """ |
| 1198 | Register the var_name to the target (set of var names) |
| 1199 | |
| 1200 | Args: |
| 1201 | var_name: |
| 1202 | name (str) or list of names of the variable to register |
| 1203 | target: |
| 1204 | the set to register the var namee |
| 1205 | """ |
| 1206 | if isinstance(var_name, str): |
| 1207 | var_name = [var_name] |
| 1208 | |
| 1209 | for name in var_name: |
| 1210 | target.add(name) |
| 1211 | |
| 1212 | def _gather_losses_by_losses_name(self, return_float=True) -> T.Dict[str, float]: |
| 1213 | """ |
no outgoing calls
no test coverage detected