From 177fa55bce33353a26d98b55059343ca3bebdec1 Mon Sep 17 00:00:00 2001 From: skullY Date: Mon, 11 Nov 2019 13:03:39 -0800 Subject: [PATCH] fix pytest in qmkfm/base_container --- qmk_cli/script_qmk.py | 5 ++++- setup.py | 3 ++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/qmk_cli/script_qmk.py b/qmk_cli/script_qmk.py index 5234ac0..a64f98b 100644 --- a/qmk_cli/script_qmk.py +++ b/qmk_cli/script_qmk.py @@ -56,7 +56,10 @@ def find_qmk_firmware(): return Path(milc.cli.config.user.qmk_home).expanduser().resolve() if 'QMK_HOME' in os.environ: - return Path(os.environ['QMK_HOME']).expanduser().resolve() + path = Path(os.environ['QMK_HOME']).expanduser() + if path.exists(): + return path.resolve() + return path return Path.home() / 'qmk_firmware' diff --git a/setup.py b/setup.py index 4dcf1a3..96f5ac7 100644 --- a/setup.py +++ b/setup.py @@ -44,6 +44,7 @@ if __name__ == "__main__": "appdirs", "argcomplete", "colorama", - "hjson" + "hjson", + "yapf" ], )