security_vision/Trajectron-plus-plus
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Assure reset of _last_timestep

Browse source
This commit is contained in:
Ruben van de Ven 2024-12-29 20:37:27 +01:00
parent 4e883511d3
commit e5151e1a32
1 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
trajectron/environment/node.py
View file

@ -31,6 +31,15 @@ class Node(object):
self.forward_in_time_on_next_override = False 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): def __eq__(self, other):
return ((isinstance(other, self.__class__) return ((isinstance(other, self.__class__)
or isinstance(self, other.__class__)) or isinstance(self, other.__class__))