From 7603b93338e5391813efd02544b3ab9b5a3450e1 Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Sun, 13 Aug 2023 20:47:55 -0700
Subject: [PATCH] Remove temp variable from set_mode()

---
 scripts/wk/clone/state.py | 21 ++++++---------------
 1 file changed, 6 insertions(+), 15 deletions(-)

diff --git a/scripts/wk/clone/state.py b/scripts/wk/clone/state.py
index e9913d72..a5ae6325 100644
--- a/scripts/wk/clone/state.py
+++ b/scripts/wk/clone/state.py
@@ -1047,24 +1047,15 @@ def select_disk_obj(label:str, disk_menu: cli.Menu, disk_path: str) -> hw_disk.D
 
 def set_mode(docopt_args) -> str:
   """Set mode from docopt_args or user selection, returns str."""
-  mode = '?'
-
-  # Check docopt_args
   if docopt_args['clone']:
-    mode = 'Clone'
-  elif docopt_args['image']:
-    mode = 'Image'
+    return 'Clone'
+
+  if docopt_args['image']:
+    return 'Image'
 
   # Ask user if necessary
-  if not mode:
-    answer = cli.choice('Are we cloning or imaging?', ['C', 'I'])
-    if answer == 'C':
-      mode = 'Clone'
-    else:
-      mode = 'Image'
-
-  # Done
-  return mode
+  answer = cli.choice('Are we cloning or imaging?', ['C', 'I'])
+  return 'Clone' if answer == 'C' else 'Image'
 
 
 if __name__ == '__main__':