added inline comments

alexhernandezgarcia · Feb 20, 2024 · b14cc05 · b14cc05
1 parent f1ecd50
commit b14cc05
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/gflownet/envs/common.py b/tests/gflownet/envs/common.py
@@ -253,6 +253,7 @@ def test__trajectories_are_reversible(self, n_repeat=1):
             while not self.env.done:
                 state, action, valid = self.env.step_random(backward=False)
                 if valid:
+                    # Copy prevents mutation by next step.
                     states_trajectory_fw.append(copy(state))
                     actions_trajectory_fw.append(action)
 
@@ -265,6 +266,7 @@ def test__trajectories_are_reversible(self, n_repeat=1):
                     actions_trajectory_fw_copy.pop()
                 )
                 if valid:
+                    # Copy prevents mutation by next step.
                     states_trajectory_bw.append(copy(state))
                     actions_trajectory_bw.append(action)