format

jax-ml · May 8, 2024 · 77da67c · 77da67c
1 parent 5458e14
commit 77da67c
Showing 1 changed file with 7 additions and 1 deletion.
diff --git 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 = {