From acf5df27ce65ce5cf643ed22879fddcaaa018c39 Mon Sep 17 00:00:00 2001 From: Chaoying <32626585+Chaoyingz@users.noreply.github.com> Date: Mon, 8 May 2023 16:17:15 +0800 Subject: [PATCH] Add support for redis password (#1508) --- qlib/config.py | 1 + qlib/utils/__init__.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/qlib/config.py b/qlib/config.py index 11ccc52ec..7b726c658 100644 --- a/qlib/config.py +++ b/qlib/config.py @@ -147,6 +147,7 @@ _default_config = { "redis_host": "127.0.0.1", "redis_port": 6379, "redis_task_db": 1, + "redis_password": None, # This value can be reset via qlib.init "logging_level": logging.INFO, # Global configuration of qlib log diff --git a/qlib/utils/__init__.py b/qlib/utils/__init__.py index 51363f2f7..910a4c08b 100644 --- a/qlib/utils/__init__.py +++ b/qlib/utils/__init__.py @@ -44,7 +44,7 @@ is_deprecated_lexsorted_pandas = version.parse(pd.__version__) > version.parse(" #################### Server #################### def get_redis_connection(): """get redis connection instance.""" - return redis.StrictRedis(host=C.redis_host, port=C.redis_port, db=C.redis_task_db) + return redis.StrictRedis(host=C.redis_host, port=C.redis_port, db=C.redis_task_db, password=C.redis_password) #################### Data ####################