diff options
Diffstat (limited to 'gen/generator.py')
-rw-r--r-- | gen/generator.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/gen/generator.py b/gen/generator.py index 6a07616..0999e84 100644 --- a/gen/generator.py +++ b/gen/generator.py | |||
@@ -6,7 +6,7 @@ import shutil as sh | |||
6 | 6 | ||
7 | from Cheetah.Template import Template | 7 | from Cheetah.Template import Template |
8 | from common import get_config_option,num_cpus,recordtype | 8 | from common import get_config_option,num_cpus,recordtype |
9 | from config.config import DEFAULTS,PARAMS | 9 | from config.config import FILES,PARAMS |
10 | from gen.dp import DesignPointGenerator | 10 | from gen.dp import DesignPointGenerator |
11 | from parse.col_map import ColMapBuilder | 11 | from parse.col_map import ColMapBuilder |
12 | 12 | ||
@@ -121,7 +121,7 @@ class Generator(object): | |||
121 | 121 | ||
122 | def _write_schedule(self, params): | 122 | def _write_schedule(self, params): |
123 | '''Write schedule file using current template for @params.''' | 123 | '''Write schedule file using current template for @params.''' |
124 | sched_file = self.out_dir + "/" + DEFAULTS['sched_file'] | 124 | sched_file = self.out_dir + "/" + FILES['sched_file'] |
125 | with open(sched_file, 'wa') as f: | 125 | with open(sched_file, 'wa') as f: |
126 | f.write(str(Template(self.template, searchList=[params]))) | 126 | f.write(str(Template(self.template, searchList=[params]))) |
127 | 127 | ||
@@ -135,7 +135,7 @@ class Generator(object): | |||
135 | else: | 135 | else: |
136 | tasks = 0 | 136 | tasks = 0 |
137 | 137 | ||
138 | exp_params_file = self.out_dir + "/" + DEFAULTS['params_file'] | 138 | exp_params_file = self.out_dir + "/" + FILES['params_file'] |
139 | with open(exp_params_file, 'wa') as f: | 139 | with open(exp_params_file, 'wa') as f: |
140 | params['scheduler'] = self.scheduler | 140 | params['scheduler'] = self.scheduler |
141 | pprint.pprint(params, f) | 141 | pprint.pprint(params, f) |