Fix forgotten initialisation of variable

alexhernandezgarcia · Jun 6, 2024 · 190152c · 190152c
1 parent 2321aa4
commit 190152c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gflownet/evaluator/base.py b/gflownet/evaluator/base.py
@@ -557,7 +557,7 @@ def plot(self, x_sampled, kde_pred, kde_true, plot_kwargs, **kwargs):
             values are the figures.
         """
 
-        fig_kde_pred = fig_kde_true = fig_reward_samples = None
+        fig_kde_pred = fig_kde_true = fig_reward_samples = fig_samples_topk = None
 
         if hasattr(self.gfn.env, "plot_reward_samples") and x_sampled is not None:
             (sample_space_batch, rewards_sample_space) = (