fix: @fs_cache fails on Python3 when index_col is set by numeric index

CMUSTRUDEL · Mar 15, 2019 · 4762049 · 4762049
1 parent b72cf04
commit 4762049
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/stutils/decorators.py b/stutils/decorators.py
@@ -42,7 +42,8 @@ def __init__(self, func, cache_dir='', app_name='', cache_type='', idx=1,
         self.cache_path = mkdir(cache_dir or DEFAULT_PATH, app_name, cache_type)
 
         if isinstance(idx, int):
-            idx = range(idx)
+            # Python3 range objects won't work, so explicitly convert to list
+            idx = list(range(idx))
         self.idx = idx
 
     def get_cache_fname(self, *args):