(self, outputs: Any)
| 231 | yield torch.cat(patch_batch[:idx_in_batch]), location_batch, idx_in_batch |
| 232 | |
| 233 | def _ensure_tuple_outputs(self, outputs: Any) -> tuple: |
| 234 | if isinstance(outputs, dict): |
| 235 | if self.output_keys is None: |
| 236 | self.output_keys = list(outputs.keys()) # model's output keys |
| 237 | return tuple(outputs[k] for k in self.output_keys) |
| 238 | return ensure_tuple(outputs, wrap_array=True) |
| 239 | |
| 240 | def _run_inference(self, network: Callable, patch: torch.Tensor, *args: Any, **kwargs: Any) -> tuple: |
| 241 | # pre-process |
no test coverage detected