diff --git a/sota-implementations/iql/iql_online.py b/sota-implementations/iql/iql_online.py index 499c2164b52..6e9c8a0d8ea 100644 --- a/sota-implementations/iql/iql_online.py +++ b/sota-implementations/iql/iql_online.py @@ -164,7 +164,7 @@ def update(sampled_tensordict): with timeit("rb - extend"): # add to replay buffer - tensordict = tensordict.rehsape(-1) + tensordict = tensordict.reshape(-1) replay_buffer.extend(tensordict.cpu()) collected_frames += current_frames