From dbca0d328c5e23c7744f67839a6b5f00d22bcc1c Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Wed, 6 May 2020 23:16:35 -0600
Subject: [PATCH] Fix multi-lq bug

---
 codes/data/LQGT_dataset.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/codes/data/LQGT_dataset.py b/codes/data/LQGT_dataset.py
index 2d54cda7..02bfdd74 100644
--- a/codes/data/LQGT_dataset.py
+++ b/codes/data/LQGT_dataset.py
@@ -14,7 +14,7 @@ class LQGTDataset(data.Dataset):
     """
 
     def get_lq_path(self, i):
-        which_lq = random.randint(0, len(self.paths_LQ))
+        which_lq = random.randint(0, len(self.paths_LQ)-1)
         return self.paths_LQ[which_lq][i]
 
     def __init__(self, opt):