diff --git a/Go2Py/control/walk_these_ways.py b/Go2Py/control/walk_these_ways.py index 8c87963..75ad554 100755 --- a/Go2Py/control/walk_these_ways.py +++ b/Go2Py/control/walk_these_ways.py @@ -131,7 +131,8 @@ class CommandInterface: self.stance_width_cmd = 0.25 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): command = np.zeros((19,))