From 6da34c1f2beebb3b88465d37c86415dc55dc581b Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Tue, 3 Dec 2019 15:03:02 -0700
Subject: [PATCH] Only register tmux atexit when running HW Diags

* Prevents unintended killing of tmux panes when importing wk or wk.hw
---
 scripts/wk/hw/diags.py | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/scripts/wk/hw/diags.py b/scripts/wk/hw/diags.py
index 1a9758f9..316d2f8f 100644
--- a/scripts/wk/hw/diags.py
+++ b/scripts/wk/hw/diags.py
@@ -19,10 +19,6 @@ from wk.hw import obj as hw_obj
 from wk.hw import sensors as hw_sensors
 
 
-# atexit functions
-atexit.register(tmux.kill_all_panes)
-#TODO: Add state/dev data dump debug function
-
 # STATIC VARIABLES
 DOCSTRING = f'''{cfg.main.KIT_NAME_FULL}: Hardware Diagnostics
 
@@ -623,6 +619,8 @@ def main():
     raise RuntimeError('tmux session not found')
 
   # Init
+  atexit.register(tmux.kill_all_panes)
+  #TODO: Add state/dev data dump debug function
   menu = build_menu(cli_mode=args['--cli'], quick_mode=args['--quick'])
   state = State()