diff options
Diffstat (limited to 'src/ee/doit.py')
| -rw-r--r-- | src/ee/doit.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/src/ee/doit.py b/src/ee/doit.py index 014af05..dc89fae 100644 --- a/src/ee/doit.py +++ b/src/ee/doit.py @@ -35,5 +35,11 @@ class DoItConfig(object): def data_sets_for(self, task): return self._data_sets[task] + def out_data_set_for(self, task): + return self.data_sets_for(task)[0] + + def input_data_sets_for(self, task): + return self.data_sets_for(task)[1] + def set_data_sets_for(self, task, out_dataset: str, *in_datasets: str): self._data_sets[task] = [out_dataset, in_datasets] |