| 4 | import os |
| 5 | |
| 6 | class DebugController: |
| 7 | def __init__(self, print=False, plot=False, epoch=0, batch=0, dataset='train', visualizations_dir = None): |
| 8 | self.print = print |
| 9 | self.plot = plot |
| 10 | self.epoch = epoch |
| 11 | self.batch = batch |
| 12 | self.dataset = dataset |
| 13 | self._visualizations_dir = visualizations_dir |
| 14 | self.action_qpos_normalizer = None |
| 15 | |
| 16 | @property |
| 17 | def visualizations_dir(self): |
| 18 | if self._visualizations_dir is None: |
| 19 | self._visualizations_dir = self.gen_visualizations_dir() |
| 20 | return self._visualizations_dir |
| 21 | |
| 22 | else: |
| 23 | return self._visualizations_dir |
| 24 | |
| 25 | @visualizations_dir.setter |
| 26 | def visualizations_dir(self, value): |
| 27 | if not os.path.exists(value): |
| 28 | os.makedirs(value) |
| 29 | self._visualizations_dir = value |
| 30 | |
| 31 | def gen_visualizations_dir(self) -> str: |
| 32 | n = 0 |
| 33 | while os.path.exists(f'visualizations/{n}'): |
| 34 | n += 1 |
| 35 | os.makedirs(f'visualizations/{n}') |
| 36 | return f'visualizations/{n}' |
| 37 | |
| 38 | |
| 39 | debug = DebugController() |
no outgoing calls
no test coverage detected