Skip to content

Commit

Permalink
minor: removed type cast, now support tuples
Browse files Browse the repository at this point in the history
  • Loading branch information
julienroyd committed Apr 1, 2024
1 parent f92690a commit 15031a3
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/gflownet/data/replay_buffer.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,7 +27,7 @@ def push(self, *args):
assert self._input_size == len(args), "ReplayBuffer input size must be constant"
if len(self.buffer) < self.capacity:
self.buffer.append(None)
args = detach_and_cpu(list(args))
args = detach_and_cpu(args)
self.buffer[self.position] = args
self.position = (self.position + 1) % self.capacity

Expand Down

0 comments on commit 15031a3

Please sign in to comment.