fix off by one for shift aug

Author: lucidrains

enformer_pytorch/data.py

@@ -83,9 +83,12 @@ def __getitem__(self, ind):
 
         if exists(self.shift_augs):
             min_shift, max_shift = self.shift_augs
+            max_shift += 1
+
             min_shift = max(start + min_shift, 0) - start
             max_shift = min(end + max_shift, chromosome_length) - end
-            rand_shift = randrange(min_shift, max_shift + 1)
+
+            rand_shift = randrange(min_shift, max_shift)
             start += rand_shift
             end += rand_shift
 

setup.py

@@ -4,7 +4,7 @@
   name = 'enformer-pytorch',
   packages = find_packages(exclude=[]),
   include_package_data = True,
-  version = '0.2.4',
+  version = '0.2.6',
   license='MIT',
   description = 'Enformer - Pytorch',
   author = 'Phil Wang',