From 6b05e6a299f1d2d1dfec2cb25d3ab314d2906eb5 Mon Sep 17 00:00:00 2001
From: Edward Shen <code@eddie.sh>
Date: Tue, 21 Apr 2020 13:49:37 -0400
Subject: [PATCH] edit test script

---
 qlearning-results/test | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/qlearning-results/test b/qlearning-results/test
index 4ebb3df..c423dfe 100755
--- a/qlearning-results/test
+++ b/qlearning-results/test
@@ -13,7 +13,11 @@ for a in 0.1 0.5 0.7 0.9; do
 	if [ -f $file ]; then
 	  echo "skipping $file, exists"
 	else
-          ./tetris train $agent -a $a -g $g -e $e -n 200000 >> "$file"
+	  if [ "$agent" = approximateqlearning ]; then
+    	    ./tetris train $agent -a $a -g $g -e $e -n 20000 >> "$file"
+	  else
+    	    ./tetris train $agent -a $a -g $g -e $e -n 200000 >> "$file"
+	  fi
 	fi
       done
     done