diff --git a/flashbax/vault/vault.py b/flashbax/vault/vault.py index 09cb4a7..f366dd5 100644 --- a/flashbax/vault/vault.py +++ b/flashbax/vault/vault.py @@ -478,6 +478,6 @@ def read( # Return the read result as a fbx buffer state return TrajectoryBufferState( experience=read_result, - current_index=jnp.array(self.vault_index, dtype=int), + current_index=jnp.array(0, dtype=int), is_full=jnp.array(True, dtype=bool), )