assert bug removed

This commit is contained in:
Rooholla-Khorrambakht 2024-05-21 20:12:22 -04:00
parent 64fe38b843
commit 933f189952
1 changed files with 2 additions and 1 deletions

View File

@ -131,7 +131,8 @@ class CommandInterface:
self.stance_width_cmd = 0.25 self.stance_width_cmd = 0.25
def setGaitType(self, gait_type): def setGaitType(self, gait_type):
assert gait_type in [key for key in self.gaits.keys()], f'The gain type should be in {self.gait = torch.tensor(self.gaits[gait_type]) assert gait_type in [key for key in self.gaits.keys()], f'The gain type should be in {self.gaits.keys()}'
self.gait = torch.tensor(self.gaits[gait_type])
def get_command(self): def get_command(self):
command = np.zeros((19,)) command = np.zeros((19,))