fix possibly unbound variables

noelleleigh · Sep 6, 2020 · 76d3d2c · 76d3d2c
1 parent 21cea80
commit 76d3d2c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/glitch_me/__main__.py b/glitch_me/__main__.py
@@ -94,8 +94,8 @@ def make_still(input_path: str, output_dir: str,
     im = ImageOps.exif_transpose(im)
 
     # Handle initial image scaling
+    original_size = im.size
     if line_count is not None:
-        original_size = im.size
         scale_factor = line_count / im.size[1]
         scaled_size = tuple(
             map(lambda val: int(val * scale_factor), im.size)
@@ -156,8 +156,8 @@ def make_gif(input_path: str, output_dir: str,
     im = ImageOps.exif_transpose(im)
 
     # Handle initial image scaling
+    original_size = im.size
     if line_count is not None:
-        original_size = im.size
         scale_factor = line_count / im.size[1]
         scaled_size = tuple(
             map(lambda val: int(val * scale_factor), im.size)