if smth is False -> if not smth

Co-authored-by: Alex <alexhg15@gmail.com>
alexhernandezgarcia · Nov 29, 2023 · 08ee25b · 08ee25b
1 parent f7e5ec3
commit 08ee25b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gflownet/envs/base.py b/gflownet/envs/base.py
@@ -786,7 +786,7 @@ def reward(self, state=None, done=None, do_non_terminating=False):
         """
         state = self._get_state(state)
         done = self._get_done(done)
-        if done is False and do_non_terminating is False:
+        if not done and not do_non_terminating:
             return tfloat(0.0, float_type=self.float, device=self.device)
         return self.proxy2reward(self.proxy(self.state2proxy(state))[0])