From f33c411bae5f02fb8712d5c79b848912e4545bc0 Mon Sep 17 00:00:00 2001
From: Fatih Beyhan <48058209+fatihbeyhan@users.noreply.github.com>
Date: Tue, 1 Mar 2022 22:08:34 +0300
Subject: [PATCH] Update to get_train_and_benchmark_data.py

Two lines of code were updated in order to download datasets from google drive without any problems.
---
 scripts/get_train_and_benchmark_data.sh | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/scripts/get_train_and_benchmark_data.sh b/scripts/get_train_and_benchmark_data.sh
index 972bef79..5c4eb26e 100755
--- a/scripts/get_train_and_benchmark_data.sh
+++ b/scripts/get_train_and_benchmark_data.sh
@@ -13,12 +13,11 @@ fi
 
 fileid="1IDjXFnNnHf__MO5j_onw4YwR97oS8lAy"
 filename="data/train_and_benchmark_data.zip"
-curl -c ./cookie -s -L "https://drive.google.com/uc?export=download&id=${fileid}" > /dev/null
-curl -Lb ./cookie "https://drive.google.com/uc?export=download&confirm=`awk '/download/ {print $NF}' ./cookie`&id=${fileid}" -o ${filename}
-rm cookie
+html=`curl -c ./cookie -s -L "https://drive.google.com/uc?export=download&id=${fileid}"`
+curl -Lb ./cookie "https://drive.google.com/uc?export=download&`echo ${html}|grep -Po '(confirm=[a-zA-Z0-9\-_]+)'`&id=${fileid}" -o ${filename}
 
 unzip -d $benchmark_data_folder $filename
 rm $filename
 
 mv $benchmark_data_folder/data/* $benchmark_data_folder/
-rm -r $benchmark_data_folder/data/
\ No newline at end of file
+rm -r $benchmark_data_folder/data/