(self)
| 130 | |
| 131 | class BaseExplorerCase(RayUnittestBase): |
| 132 | def setUp(self): |
| 133 | self.config = get_template_config() |
| 134 | self.config.mode = "explore" |
| 135 | self.config.buffer.total_epochs = 2 |
| 136 | self.config.buffer.batch_size = 4 |
| 137 | self.config.model.model_path = get_model_path() |
| 138 | self.config.explorer.rollout_model.engine_type = "vllm" |
| 139 | self.config.algorithm.repeat_times = 2 |
| 140 | self.config.monitor.monitor_type = "tensorboard" |
| 141 | self.config.project = "Trinity-unittest" |
| 142 | self.config.checkpoint_root_dir = get_checkpoint_path() |
| 143 | self.config.synchronizer.sync_interval = 2 |
| 144 | self.config.explorer.eval_interval = 4 |
| 145 | self.config.monitor.detailed_stats = False |
| 146 | |
| 147 | |
| 148 | class TestExplorerCountdownEval(BaseExplorerCase): |
nothing calls this directly
no test coverage detected