From aa43eb668492e03a5d2c6cc5010479f6e0cba220 Mon Sep 17 00:00:00 2001 From: Simon Forman Date: Fri, 18 Dec 2020 16:10:45 -0800 Subject: [PATCH] Minor cleanup. --- joy/utils/stack.py | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/joy/utils/stack.py b/joy/utils/stack.py index 3582216..fc11151 100644 --- a/joy/utils/stack.py +++ b/joy/utils/stack.py @@ -206,17 +206,15 @@ def dnd(stack, from_index, to_index): h, stack = stack head.append(h) diff += 1 - stack = item, stack - while head: - stack = head.pop(), stack else: # from > to # so the destination is in the head list - while head: + while diff: stack = head.pop(), stack - from_index -= 1 - if from_index == to_index: - stack = item, stack + diff -= 1 + stack = item, stack + while head: + stack = head.pop(), stack return stack