diff --git a/qlib/backtest/__init__.py b/qlib/backtest/__init__.py index 9daba911533..e016ef5c41c 100644 --- a/qlib/backtest/__init__.py +++ b/qlib/backtest/__init__.py @@ -163,7 +163,7 @@ def create_account_instance( position_dict=position_dict, pos_type=pos_type, benchmark_config=( - {} + {"benchmark": None} if benchmark is None else { "benchmark": benchmark, diff --git a/tests/backtest/test_benchmark_config.py b/tests/backtest/test_benchmark_config.py new file mode 100644 index 00000000000..fe20162b4f6 --- /dev/null +++ b/tests/backtest/test_benchmark_config.py @@ -0,0 +1,15 @@ +from qlib.backtest import create_account_instance + + +def test_create_account_instance_disables_benchmark_when_none(): + account = create_account_instance( + start_time="2020-01-01", + end_time="2020-01-02", + benchmark=None, + account=100000, + ) + + assert account.benchmark_config["benchmark"] is None + assert "start_time" not in account.benchmark_config + assert "end_time" not in account.benchmark_config + assert account.portfolio_metrics.bench is None