diff --git a/python/PiFinder/utils.py b/python/PiFinder/utils.py index 951bc25b9..e864bff2e 100644 --- a/python/PiFinder/utils.py +++ b/python/PiFinder/utils.py @@ -45,11 +45,21 @@ def serialize_solution(solution: dict) -> str: def get_sys_utils(): - try: - # Attempt to import the real sys_utils - sys_utils = importlib.import_module("PiFinder.sys_utils") - except ImportError: + # Check if we should use fake sys_utils for local development + use_fake = os.environ.get("PIFINDER_USE_FAKE_SYS_UTILS", "").lower() in ( + "1", + "true", + "yes", + ) + + if use_fake: sys_utils = importlib.import_module("PiFinder.sys_utils_fake") + else: + try: + # Attempt to import the real sys_utils + sys_utils = importlib.import_module("PiFinder.sys_utils") + except ImportError: + sys_utils = importlib.import_module("PiFinder.sys_utils_fake") return sys_utils