fix a typo for shapes instead of shape in the argparse

Oslandia · Mar 27, 2018 · 05caabf · 05caabf
1 parent e4a7aab
commit 05caabf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/sources/test.py b/sources/test.py
@@ -20,7 +20,7 @@
                         help=("Number of images that must be contained "
                               "into a single batch"))
     parser.add_argument('-d', '--dataset', required=True, nargs='?',
-                        help="Dataset type (either mapillary or shape")
+                        help="Dataset type (either mapillary or shapes")
     parser.add_argument('-dp', '--datapath', required=False,
                         default="../data", nargs='?',
                         help="Relative path towards data directory")
@@ -80,7 +80,7 @@
     elif args.dataset == "shapes":
         testing_dataset = ShapeDataset(image_size, 3)
     else:
-        utils.logger.error("Unsupported dataset type. Please choose 'mapillary' or 'shape'")
+        utils.logger.error("Unsupported dataset type. Please choose 'mapillary' or 'shapes'")
         sys.exit(1)
 
     # Dataset populating/loading (depends on the existence of a specification file)

diff --git a/sources/train.py b/sources/train.py
@@ -24,7 +24,7 @@
                         help=("Measure the training phase execution time "
                               "and store it in a json file"))
     parser.add_argument('-d', '--dataset', required=True, nargs='?',
-                        help="Dataset type (either mapillary or shape")
+                        help="Dataset type (either mapillary or shapes")
     parser.add_argument('-dp', '--datapath', required=False,
                         default="../data", nargs='?',
                         help="Relative path towards data directory")
@@ -159,7 +159,7 @@
         train_dataset = ShapeDataset(args.image_size, 3)
         validation_dataset = ShapeDataset(args.image_size, 3)
     else:
-        utils.logger.error("Unsupported dataset type. Please choose 'mapillary' or 'shape'")
+        utils.logger.error("Unsupported dataset type. Please choose 'mapillary' or 'shapes'")
         sys.exit(1)
 
     # Dataset populating/loading (depends on the existence of a specification file)