From 05caabfe1c755e5a3c1e64b17ef58505d6bc96e5 Mon Sep 17 00:00:00 2001
From: Damien Garaud <damien.garaud@oslandia.com>
Date: Tue, 27 Mar 2018 18:08:39 +0200
Subject: [PATCH] fix a typo for shapes instead of shape in the argparse

---
 sources/test.py  | 4 ++--
 sources/train.py | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/sources/test.py b/sources/test.py
index 890f2fcd..e683adca 100644
--- 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
index 2f58cb37..99f0e2ef 100644
--- 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)