@@ -627,7 +627,7 @@ def forward(self, tensordict: TensorDictBase) -> TensorDictBase:
                 "At least one dimension of the tensordict must be named 'time' in offline mode"
             )
         time_dim = time_dim[0]
-        for in_key, out_key in _zip_strict(self.in_keys, self.out_keys):
+        for in_key, out_key in zip(self.in_keys, self.out_keys, strict=True):
             reward = tensordict[in_key]
             cumsum = reward.cumsum(time_dim)
             tensordict.set(out_key, cumsum)
Original file line number	Diff line number	Diff line change
`@@ -627,7 +627,7 @@ def forward(self, tensordict: TensorDictBase) -> TensorDictBase:`
`627`	`627`	`"At least one dimension of the tensordict must be named 'time' in offline mode"`
`628`	`628`	`)`
`629`	`629`	`time_dim = time_dim[0]`
`630`		`- for in_key, out_key in _zip_strict(self.in_keys, self.out_keys):`
	`630`	`+ for in_key, out_key in zip(self.in_keys, self.out_keys, strict=True):`
`631`	`631`	`reward = tensordict[in_key]`
`632`	`632`	`cumsum = reward.cumsum(time_dim)`
`633`	`633`	`tensordict.set(out_key, cumsum)`