From 6a0cfd731bcfe5170e00ffaa65909ec2ae9fbcf0 Mon Sep 17 00:00:00 2001 From: kingbri Date: Sat, 3 Aug 2024 22:00:15 -0400 Subject: [PATCH] Main: Only import psutil when the experimental function is run Experimental options shouldn't be imported at the top level until the testing period is over. Signed-off-by: kingbri --- main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/main.py b/main.py index bae2f98..5ed20f3 100644 --- a/main.py +++ b/main.py @@ -9,8 +9,6 @@ import signal from loguru import logger from typing import Optional -import psutil - from common import config, gen_logging, sampling, model from common.args import convert_args_to_dict, init_argparser from common.auth import load_auth_keys @@ -162,6 +160,8 @@ def entrypoint(arguments: Optional[dict] = None): # Set the process priority if unwrap(developer_config.get("realtime_process_priority"), False): + import psutil + current_process = psutil.Process(os.getpid()) if platform.system() == "Windows": current_process.nice(psutil.REALTIME_PRIORITY_CLASS)