Python 2 compatibility of preprocessing script.

data/download_and_preprocess_dataset.sh

 #!/usr/bin/env bash
 
+# exit if any command fails
+set -e
+
 if [ "$#" -ne 1 ]; then
   echo "Usage: $0 DATASET_NAME" >&2
   exit 1

video_prediction/datasets/kth_dataset.py

@@ -117,8 +117,7 @@ def main():
     for partition_name, partition_fnames in zip(partition_names, partition_fnames):
         partition_dir = os.path.join(args.output_dir, partition_name)
         if not os.path.exists(partition_dir):
-            os.makedirs(partition_dir, exist_ok=True)
-
+            os.makedirs(partition_dir)
         read_videos_and_save_tf_records(partition_dir, partition_fnames)