wassname/pyrobolearn
1
0
Fork 0
mirror of https://github.com/wassname/pyrobolearn.git synced 2026-09-09 11:31:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix roll in converter.py

Browse Source
This commit is contained in:
Brian Delhaisse
2019-11-20 04:22:04 +01:00
parent 7a88f62746
commit b1f4ac84f1
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pyrobolearn/utils/converter.py
+2 -2
View File
@@ -251,7 +251,7 @@ class QuaternionNumpyConverter(TypeConverter):
if isinstance(data, self.to_type):
return data
elif isinstance(data, self.from_type):
return np.quaternion(roll(data.ravel().tolist(), -self.convention))
return np.quaternion(*roll(data.ravel().tolist(), -self.convention))
else:
raise TypeError("Type not known: {}".format(type(data)))
@@ -301,7 +301,7 @@ class QuaternionPyTorchConverter(TypeConverter):
if isinstance(data, self.to_type):
return data
elif isinstance(data, self.from_type):
return np.quaternion(roll(data.view(-1).data.tolist(), -self.convention))
return np.quaternion(*roll(data.view(-1).data.tolist(), -self.convention))
else:
raise TypeError("Type not known: {}".format(type(data)))