commit to support the docker detatched mode

igb/download.py

@@ -58,14 +58,14 @@ def check_md5sum(dataset_type, dataset_size, filename):
         raise Exception(" md5sum verification failed!.")
 
 
-def download_dataset(path, dataset_type, dataset_size):
+def download_dataset(path, dataset_type, dataset_size, confirm_download):
     output_directory = path
     url = dataset_urls[dataset_type][dataset_size]
     filename = path + "/igb_" + dataset_type + "_" + dataset_size + ".tar.gz"
     # check if the dataset is already downloaded
     if os.path.exists(filename):
         print("Dataset already downloaded.")
-    elif decide_download(url):
+    elif confirm_download or decide_download(url):
         data = ur.urlopen(url)
         size = int(data.info()["Content-Length"])
         chunk_size = 1024*1024
@@ -103,6 +103,10 @@ def download_dataset(path, dataset_type, dataset_size):
     parser.add_argument('--dataset_size', type=str, default='tiny',
         choices=['tiny', 'small', 'medium'], 
         help='size of the datasets')
+    parser.add_argument(
+        "--confirm-download",
+        action="store_true",
+        help="To skip the user prompt for confirming the download, which is useful for Docker detached mode.")
     args = parser.parse_args()    
-    download_dataset(args.path, args.dataset_type, args.dataset_size)
+    download_dataset(args.path, args.dataset_type, args.dataset_size, args.confirm_download)