From db736a8cb4e7b97f8b0c44a286782b2767bbc09c Mon Sep 17 00:00:00 2001
From: Victor Schmidt <vsch@pm.me>
Date: Wed, 21 Feb 2024 13:04:08 -0500
Subject: [PATCH] remove unused

---
 gflownet/evaluator/base.py | 1 -
 gflownet/gflownet.py       | 1 -
 2 files changed, 2 deletions(-)

diff --git a/gflownet/evaluator/base.py b/gflownet/evaluator/base.py
index ae7d44a38..cae5dd888 100644
--- a/gflownet/evaluator/base.py
+++ b/gflownet/evaluator/base.py
@@ -513,7 +513,6 @@ def compute_log_prob_metrics(self, x_tt, metrics=None):
     def compute_density_metrics(self, x_tt, dict_tt, metrics=None):
         gfn = self.gfn_agent
         metrics = self.make_metrics(metrics)
-        reqs = self.make_requirements(metrics=metrics)  # TODO-V: unused for now, TBD
 
         density_metrics = {}
         x_sampled = density_true = density_pred = None
diff --git a/gflownet/gflownet.py b/gflownet/gflownet.py
index 1b6399f73..4c5b85a6c 100644
--- a/gflownet/gflownet.py
+++ b/gflownet/gflownet.py
@@ -1101,7 +1101,6 @@ def train(self):
             # Log
             if self.logger.lightweight:
                 all_losses = all_losses[-100:]
-                all_visited = states_term  # TODO-V: unused
             else:
                 all_visited.extend(states_term)
             # Progress bar