From bdd20c4eedec1cd180b3e33ea514eeec3c59082c Mon Sep 17 00:00:00 2001 From: Aravinda Rao Date: Thu, 7 Mar 2024 01:12:02 +0530 Subject: [PATCH] refactor(core/tree): handle setting root to None in caller instead of helper --- src/qtile_bonsai/core/tree.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/qtile_bonsai/core/tree.py b/src/qtile_bonsai/core/tree.py index 80fcab1..7b62c1f 100644 --- a/src/qtile_bonsai/core/tree.py +++ b/src/qtile_bonsai/core/tree.py @@ -385,6 +385,8 @@ def remove( br_rm, br_sib, br_rm_nodes = self._remove( node, consume_vacant_space=True, normalize=normalize ) + if br_rm is self._root: + self._root = None rm_nodes.extend(br_rm_nodes) if br_sib is not None: @@ -681,7 +683,6 @@ def _remove( """ br_rm, br_rm_nodes = self._find_removal_branch(node) if br_rm is self._root: - self._root = None return (br_rm, None, br_rm_nodes) container = br_rm.parent