diff options
| author | Jules Laplace <julescarbon@gmail.com> | 2018-05-15 01:32:55 +0200 |
|---|---|---|
| committer | Jules Laplace <julescarbon@gmail.com> | 2018-05-15 01:32:55 +0200 |
| commit | f997993eb9ccdabaf9ab1b86ad5f668368120982 (patch) | |
| tree | 73b0cc3ec050040de857fba52828410095b628d5 /options/dataset_options.py | |
| parent | f2edb4afa85a35a18b4eebdd83fb6045af9cfb5a (diff) | |
args
Diffstat (limited to 'options/dataset_options.py')
| -rw-r--r-- | options/dataset_options.py | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/options/dataset_options.py b/options/dataset_options.py index 5eca374..fcb9654 100644 --- a/options/dataset_options.py +++ b/options/dataset_options.py @@ -178,8 +178,11 @@ class DatasetOptions(BaseOptions): help='canny high threshold' ) - def parse(self): + def parse(self, args=None): if not self.initialized: self.initialize() - self.opt = self.parser.parse_known_args() + if args is not None: + self.opt, unknown = self.parser.parse_known_args(args) + else: + self.opt, unknown = self.parser.parse_known_args() return self.opt
\ No newline at end of file |