fix lint

jax-ml · May 3, 2024 · 96044ec · 96044ec
1 parent 63d94b1
commit 96044ec
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/coix/loss.py b/coix/loss.py
@@ -117,6 +117,7 @@ def elbo_loss(q_trace, p_trace, incoming_log_weight, incremental_log_weight):
 
 def fkl_loss(q_trace, p_trace, incoming_log_weight, incremental_log_weight):
   """Forward KL objective. Here we do not optimize p."""
+  del p_trace
   batch_ndims = incoming_log_weight.ndim
   q_log_probs = {
       name: util.get_site_log_prob(site) for name, site in q_trace.items()