From 77da67caa1345cfd4e38fb01d7debca68250ce99 Mon Sep 17 00:00:00 2001 From: Du Phan Date: Wed, 8 May 2024 06:56:45 -0400 Subject: [PATCH] format --- coix/loss.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/coix/loss.py b/coix/loss.py index cbef1c5..1a10fec 100644 --- a/coix/loss.py +++ b/coix/loss.py @@ -29,7 +29,13 @@ ] -def apg_loss(q_trace, p_trace, incoming_log_weight, incremental_log_weight, aggregate=False): +def apg_loss( + q_trace, + p_trace, + incoming_log_weight, + incremental_log_weight, + aggregate=False, +): """RWS objective that exploits conditional dependency.""" del incoming_log_weight, incremental_log_weight p_log_probs = {