Merge pull request #310 from alexhernandezgarcia/fix-gfn-get-max-reward

[Fix bug] [One-line PR] Fix call of `proxy.get_max_reward()`
alexhernandezgarcia · May 31, 2024 · 72f961b · 72f961b
2 parents bcb1443 + 982d0b5
commit 72f961b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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