From 190152cc315a7829731f749f0e0847e52c7d7051 Mon Sep 17 00:00:00 2001 From: alexhernandezgarcia Date: Wed, 5 Jun 2024 21:47:45 -0400 Subject: [PATCH] Fix forgotten initialisation of variable --- gflownet/evaluator/base.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/gflownet/evaluator/base.py b/gflownet/evaluator/base.py index 9085d9f84..adc633c73 100644 --- 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) = (