(self, dirname, pipeline_name)
| 27 | self.prepare_dir(self.model_dir, 'custom-image') |
| 28 | |
| 29 | def prepare_dir(self, dirname, pipeline_name): |
| 30 | if not os.path.exists(dirname): |
| 31 | os.makedirs(dirname) |
| 32 | cfg_file = os.path.join(dirname, ModelFile.CONFIGURATION) |
| 33 | cfg = { |
| 34 | ConfigFields.framework: 'dummy', |
| 35 | ConfigFields.task: 'dummy-task', |
| 36 | ConfigFields.pipeline: { |
| 37 | 'type': pipeline_name, |
| 38 | } |
| 39 | } |
| 40 | io.dump(cfg, cfg_file) |
| 41 | |
| 42 | def test_abstract(self): |
| 43 |