vessim/storage.py

@@ -114,11 +114,11 @@ def update(self, soc: float, duration: int) -> float:
                     self.T_celsius,
                 )
                 self._q = self.model.outputs["q"][-1]
+                self.samples = 0
 
-            sample = self.samples % self.sample_size
-            self.t_secs[sample] = self.t_secs[sample - 1] + duration
-            self.soc[sample] = soc
-            self.T_celsius[sample] = self.temp.now(self.t)
+            self.t_secs[self.samples] = self.t_secs[self.samples - 1] + duration
+            self.soc[self.samples] = soc
+            self.T_celsius[self.samples] = self.temp.now(self.t)
             self.samples += 1
             return self._q