1 files changed, 6 insertions, 4 deletions
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: