From fd4f8dc227fcf1aaee3faf76437066cc871c3ff7 Mon Sep 17 00:00:00 2001
From: Jules Laplace <julescarbon@gmail.com>
Date: Tue, 15 May 2018 01:34:01 +0200
Subject: args

---
 options/dataset_options.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

(limited to 'options/dataset_options.py')

diff --git a/options/dataset_options.py b/options/dataset_options.py
index fcb9654..c302ec5 100644
--- a/options/dataset_options.py
+++ b/options/dataset_options.py
@@ -1,7 +1,7 @@
 from .base_options import BaseOptions
 
 class DatasetOptions(BaseOptions):
-    def initialize(self):
+    def initialize(self, args=None):
         # BaseOptions.initialize(self)
         # type = int, float, str OR action='store_true'
 
@@ -12,17 +12,19 @@ class DatasetOptions(BaseOptions):
         #   help=''
         # )
 
+        required = args is None
+
         self.parser.add_argument(
           '--in_dir',
           type=str,
-          required=True,
+          required=required,
           help='input directory'
         )
 
         self.parser.add_argument(
           '--out_dir',
           type=str,
-          required=True,
+          required=required,
           help='output directory'
         )
 
@@ -180,7 +182,7 @@ class DatasetOptions(BaseOptions):
 
     def parse(self, args=None):
         if not self.initialized:
-            self.initialize()
+            self.initialize(args)
         if args is not None:
           self.opt, unknown = self.parser.parse_known_args(args)
         else:
-- 
cgit v1.2.3-70-g09d2