(self, data)
| 26 | return self._save_dir |
| 27 | |
| 28 | def convert_data(self, data): |
| 29 | if data is None: |
| 30 | return None |
| 31 | elif isinstance(data, np.ndarray): |
| 32 | return data |
| 33 | elif isinstance(data, torch.Tensor): |
| 34 | return data.detach().cpu().numpy() |
| 35 | elif isinstance(data, list): |
| 36 | return [self.convert_data(d) for d in data] |
| 37 | elif isinstance(data, dict): |
| 38 | return {k: self.convert_data(v) for k, v in data.items()} |
| 39 | else: |
| 40 | raise TypeError( |
| 41 | "Data must be in type numpy.ndarray, torch.Tensor, list or dict, getting", |
| 42 | type(data), |
| 43 | ) |
| 44 | |
| 45 | def get_save_path(self, filename): |
| 46 | save_path = os.path.join(self.get_save_dir(), filename) |
no outgoing calls
no test coverage detected