Convert seed 'range' into a list

This fixes a bug later in in e.g., style_mixing_example where ranges are concatened together with the +-operator. Add .gitignore file.
NVlabs · Feb 5, 2020 · cec605e · cec605e
1 parent 4874628
commit cec605e
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,2 @@
+/.stylegan2-cache/
+__pycache__/
diff --git a/run_generator.py b/run_generator.py
@@ -93,7 +93,7 @@ def _parse_num_range(s):
     range_re = re.compile(r'^(\d+)-(\d+)$')
     m = range_re.match(s)
     if m:
-        return range(int(m.group(1)), int(m.group(2))+1)
+        return list(range(int(m.group(1)), int(m.group(2))+1))
     vals = s.split(',')
     return [int(x) for x in vals]
 

diff --git a/run_projector.py b/run_projector.py
@@ -76,7 +76,7 @@ def _parse_num_range(s):
     range_re = re.compile(r'^(\d+)-(\d+)$')
     m = range_re.match(s)
     if m:
-        return range(int(m.group(1)), int(m.group(2))+1)
+        return list(range(int(m.group(1)), int(m.group(2))+1))
     vals = s.split(',')
     return [int(x) for x in vals]