From 982d0b5b450c8fff2cd2cd2885ca74b6a4375ac8 Mon Sep 17 00:00:00 2001
From: alexhernandezgarcia <alexhdezgarcia@gmail.com>
Date: Wed, 29 May 2024 23:24:10 -0400
Subject: [PATCH] Fix call of proxy.get_max_reward()

---
 gflownet/gflownet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/gflownet/gflownet.py b/gflownet/gflownet.py
index 42ce57b53..f0774410f 100644
--- a/gflownet/gflownet.py
+++ b/gflownet/gflownet.py
@@ -1538,7 +1538,7 @@ def sample_from_reward(
             format.
         """
         samples_final = []
-        max_reward = self.get_max_reward()
+        max_reward = self.proxy.get_max_reward()
         while len(samples_final) < n_samples:
             if proposal_distribution == "uniform":
                 # TODO: sample only the remaining number of samples