From e5151e1a328511671560d6b5e138640835b31298 Mon Sep 17 00:00:00 2001
From: Ruben van de Ven <git@rubenvandeven.com>
Date: Sun, 29 Dec 2024 20:37:27 +0100
Subject: [PATCH] Assure reset of _last_timestep

---
 trajectron/environment/node.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/trajectron/environment/node.py b/trajectron/environment/node.py
index d349b77..e17e785 100644
--- a/trajectron/environment/node.py
+++ b/trajectron/environment/node.py
@@ -31,6 +31,15 @@ class Node(object):
 
         self.forward_in_time_on_next_override = False
 
+    @property
+    def first_timestep(self):
+        return self._first_timestep
+
+    @first_timestep.setter
+    def first_timestep(self, value):
+        self._first_timestep = value
+        self._last_timestep = None # reset
+
     def __eq__(self, other):
         return ((isinstance(other, self.__class__)
                  or isinstance(self, other.__class__))